60 lines
1.0 KiB
Go
60 lines
1.0 KiB
Go
package core
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/cdle/sillyGirl/im"
|
|
)
|
|
|
|
var Buckets []Bucket
|
|
|
|
func init() {
|
|
AddCommand("", []Function{
|
|
{
|
|
Rules: []string{"set ? ? ?"},
|
|
Handle: func(s im.Sender) interface{} {
|
|
b := Bucket(s.Get(0))
|
|
if !IsBucket(b) {
|
|
return errors.New("不存在的存储桶")
|
|
}
|
|
b.Set(s.Get(1), s.Get(2))
|
|
return "设置成功"
|
|
},
|
|
},
|
|
{
|
|
Rules: []string{"delete ? ?"},
|
|
Handle: func(s im.Sender) interface{} {
|
|
b := Bucket(s.Get(0))
|
|
if !IsBucket(b) {
|
|
return errors.New("不存在的存储桶")
|
|
}
|
|
b.Set(s.Get(1), "")
|
|
return "删除成功"
|
|
},
|
|
},
|
|
{
|
|
Rules: []string{"get ? ? ?"},
|
|
Handle: func(s im.Sender) interface{} {
|
|
b := Bucket(s.Get(0))
|
|
if !IsBucket(b) {
|
|
return errors.New("不存在的存储桶")
|
|
}
|
|
v := b.Get(s.Get(1))
|
|
if v == "" {
|
|
return errors.New("空值")
|
|
}
|
|
return v
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func IsBucket(b Bucket) bool {
|
|
for i := range Buckets {
|
|
if Buckets[i] == b {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|