cdle
122 lines
2.1 KiB
Go
122 lines
2.1 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/boltdb/bolt"
|
|
)
|
|
|
|
var sillyGirl Bucket = NewBucket("sillyGirl")
|
|
var db *bolt.DB
|
|
|
|
type Bucket string
|
|
|
|
var Buckets = []Bucket{}
|
|
|
|
func NewBucket(name string) Bucket {
|
|
b := Bucket(name)
|
|
Buckets = append(Buckets, b)
|
|
return b
|
|
}
|
|
|
|
func initStore() {
|
|
if _, err := os.Stat(ExecPath + "/sillyGirl.cache"); err == nil {
|
|
os.Rename(ExecPath+"/sillyGirl.cache", "/etc/sillyGirl/sillyGirl.cache")
|
|
}
|
|
var err error
|
|
db, err = bolt.Open("/etc/sillyGirl/sillyGirl.cache", 0600, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (bucket Bucket) Set(key interface{}, value interface{}) {
|
|
db.Update(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket([]byte(bucket))
|
|
if b == nil {
|
|
b, _ = tx.CreateBucket([]byte(bucket))
|
|
}
|
|
b.Put([]byte(fmt.Sprint(key)), []byte(fmt.Sprint(value)))
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (bucket Bucket) Get(kv ...interface{}) string {
|
|
var key, value string
|
|
for i := range kv {
|
|
if i == 0 {
|
|
key = fmt.Sprint(kv[0])
|
|
} else {
|
|
value = fmt.Sprint(kv[1])
|
|
}
|
|
}
|
|
db.View(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket([]byte(bucket))
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
if v := string(b.Get([]byte(key))); v != "" {
|
|
value = v
|
|
}
|
|
return nil
|
|
})
|
|
return value
|
|
}
|
|
|
|
func (bucket Bucket) GetInt(key interface{}, vs ...int) int {
|
|
var value int
|
|
if len(vs) != 0 {
|
|
value = vs[0]
|
|
}
|
|
db.View(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket([]byte(bucket))
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
v := Int(string(b.Get([]byte(fmt.Sprint(key)))))
|
|
if v != 0 {
|
|
value = v
|
|
}
|
|
return nil
|
|
})
|
|
return value
|
|
}
|
|
|
|
func (bucket Bucket) GetBool(key interface{}, vs ...bool) bool {
|
|
var value bool
|
|
if len(vs) != 0 {
|
|
value = vs[0]
|
|
}
|
|
db.View(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket([]byte(bucket))
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
v := string(b.Get([]byte(fmt.Sprint(key))))
|
|
if v == "true" {
|
|
value = true
|
|
} else if v == "false" {
|
|
value = false
|
|
}
|
|
return nil
|
|
})
|
|
return value
|
|
}
|
|
|
|
func (bucket Bucket) Foreach(f func(k, v []byte) error) {
|
|
db.View(func(tx *bolt.Tx) error {
|
|
b := tx.Bucket([]byte(bucket))
|
|
if b != nil {
|
|
b.ForEach(f)
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
var Int = func(s string) int {
|
|
i, _ := strconv.Atoi(s)
|
|
return i
|
|
}
|