forked from p93542168/wheat-cache
feat(dao-channelx): add dao channelx
This commit is contained in:
parent
a57eab1184
commit
bf0c03a77f
|
@ -2,32 +2,93 @@ package dao
|
|||
|
||||
import (
|
||||
"gitee.com/wheat-os/wheatCache/pkg/errorx"
|
||||
"gitee.com/wheat-os/wheatCache/pkg/event2"
|
||||
"gitee.com/wheat-os/wheatCache/pkg/proto"
|
||||
"gitee.com/wheat-os/wheatCache/pkg/structure"
|
||||
"gitee.com/wheat-os/wheatCache/pkg/structure/channelx"
|
||||
)
|
||||
|
||||
|
||||
|
||||
func (d *Dao) CPush(key *proto.BaseKey, Value []string) (interface{}, error) {
|
||||
val, ok := d.lru.Get(key)
|
||||
if !ok {
|
||||
return nil, errorx.NotKeyErr(key.Key)
|
||||
}
|
||||
chanVal, ok := val.()
|
||||
chanVal, ok := val.(structure.ChannelXInterface)
|
||||
if !ok {
|
||||
return nil, errorx.DaoTypeErr("channelx")
|
||||
}
|
||||
|
||||
return event2.EventAwaitFunc(func() (interface{}, error) {
|
||||
var sumUp structure.UpdateLength
|
||||
for _, v := range Value {
|
||||
sumUp += chanVal.Push(v)
|
||||
}
|
||||
|
||||
d.lru.UpdateLruSize(sumUp)
|
||||
|
||||
return &proto.CPushResponse{}, nil
|
||||
}), nil
|
||||
|
||||
}
|
||||
|
||||
func (d *Dao) CPop(_ *proto.BaseKey, _ int32) (interface{}, error) {
|
||||
panic("not implemented") // TODO: Implement
|
||||
func (d *Dao) CPop(key *proto.BaseKey, count int32) (interface{}, error) {
|
||||
val, ok := d.lru.Get(key)
|
||||
if !ok {
|
||||
return nil, errorx.NotKeyErr(key.Key)
|
||||
}
|
||||
chanVal, ok := val.(structure.ChannelXInterface)
|
||||
if !ok {
|
||||
return nil, errorx.DaoTypeErr("channelx")
|
||||
}
|
||||
|
||||
return event2.EventAwaitFunc(func() (interface{}, error) {
|
||||
var sumUp structure.UpdateLength
|
||||
result := make([]string, 0, count)
|
||||
for i := int32(0); i < count; i++ {
|
||||
v, up := chanVal.Pop()
|
||||
sumUp += up
|
||||
result = append(result, v)
|
||||
}
|
||||
|
||||
d.lru.UpdateLruSize(sumUp)
|
||||
return &proto.CPopResponse{Result: result}, nil
|
||||
}), nil
|
||||
}
|
||||
|
||||
func (d *Dao) CMake(key *proto.BaseKey, length int32) (*proto.CMakeResponse, error) {
|
||||
panic("not implemented") // TODO: Implement
|
||||
chanVal := channelx.MakeChannelX(int(length))
|
||||
err := d.lru.Add(key, chanVal)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &proto.CMakeResponse{}, nil
|
||||
}
|
||||
|
||||
func (d *Dao) CLen(_ *proto.BaseKey) (*proto.CLenResponse, error) {
|
||||
panic("not implemented") // TODO: Implement
|
||||
func (d *Dao) CLen(key *proto.BaseKey) (*proto.CLenResponse, error) {
|
||||
val, ok := d.lru.Get(key)
|
||||
if !ok {
|
||||
return nil, errorx.NotKeyErr(key.Key)
|
||||
}
|
||||
chanVal, ok := val.(structure.ChannelXInterface)
|
||||
if !ok {
|
||||
return nil, errorx.DaoTypeErr("channelx")
|
||||
}
|
||||
|
||||
return &proto.CLenResponse{Length: int32(chanVal.Length())}, nil
|
||||
}
|
||||
|
||||
func (d *Dao) CClean(_ *proto.BaseKey) (*proto.CCleanResponse, error) {
|
||||
panic("not implemented") // TODO: Implement
|
||||
func (d *Dao) CClean(key *proto.BaseKey) (*proto.CCleanResponse, error) {
|
||||
val, ok := d.lru.Get(key)
|
||||
if !ok {
|
||||
return nil, errorx.NotKeyErr(key.Key)
|
||||
}
|
||||
chanVal, ok := val.(structure.ChannelXInterface)
|
||||
if !ok {
|
||||
return nil, errorx.DaoTypeErr("channelx")
|
||||
}
|
||||
up := chanVal.Clean()
|
||||
d.lru.UpdateLruSize(up)
|
||||
|
||||
return &proto.CCleanResponse{}, nil
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue