mirror of
https://gitea.phreedom.club/localhost_frssoft/bloat.git
synced 2024-11-05 06:33:01 +00:00
Update kv store
This commit is contained in:
parent
9cd9e8bc7c
commit
a4a1ce9677
41
kv/kv.go
41
kv/kv.go
@ -15,7 +15,7 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Database struct {
|
type Database struct {
|
||||||
data map[string][]byte
|
cache map[string][]byte
|
||||||
basedir string
|
basedir string
|
||||||
m sync.RWMutex
|
m sync.RWMutex
|
||||||
}
|
}
|
||||||
@ -27,40 +27,37 @@ func NewDatabse(basedir string) (db *Database, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return &Database{
|
return &Database{
|
||||||
data: make(map[string][]byte),
|
cache: make(map[string][]byte),
|
||||||
basedir: basedir,
|
basedir: basedir,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *Database) Set(key string, val []byte) (err error) {
|
func (db *Database) Set(key string, val []byte) (err error) {
|
||||||
if len(key) < 1 {
|
if len(key) < 1 || strings.ContainsRune(key, os.PathSeparator) {
|
||||||
return errInvalidKey
|
return errInvalidKey
|
||||||
}
|
}
|
||||||
|
|
||||||
db.m.Lock()
|
|
||||||
defer func() {
|
|
||||||
if err != nil {
|
|
||||||
delete(db.data, key)
|
|
||||||
}
|
|
||||||
db.m.Unlock()
|
|
||||||
}()
|
|
||||||
|
|
||||||
db.data[key] = val
|
|
||||||
|
|
||||||
err = ioutil.WriteFile(filepath.Join(db.basedir, key), val, 0644)
|
err = ioutil.WriteFile(filepath.Join(db.basedir, key), val, 0644)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
db.m.Lock()
|
||||||
|
db.cache[key] = val
|
||||||
|
db.m.Unlock()
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *Database) Get(key string) (val []byte, err error) {
|
func (db *Database) Get(key string) (val []byte, err error) {
|
||||||
if len(key) < 1 {
|
if len(key) < 1 || strings.ContainsRune(key, os.PathSeparator) {
|
||||||
return nil, errInvalidKey
|
return nil, errInvalidKey
|
||||||
}
|
}
|
||||||
|
|
||||||
db.m.RLock()
|
db.m.RLock()
|
||||||
defer db.m.RUnlock()
|
data, ok := db.cache[key]
|
||||||
|
db.m.RUnlock()
|
||||||
|
|
||||||
data, ok := db.data[key]
|
|
||||||
if !ok {
|
if !ok {
|
||||||
data, err = ioutil.ReadFile(filepath.Join(db.basedir, key))
|
data, err = ioutil.ReadFile(filepath.Join(db.basedir, key))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -68,7 +65,9 @@ func (db *Database) Get(key string) (val []byte, err error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
db.data[key] = data
|
db.m.Lock()
|
||||||
|
db.cache[key] = data
|
||||||
|
db.m.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
val = make([]byte, len(data))
|
val = make([]byte, len(data))
|
||||||
@ -82,11 +81,11 @@ func (db *Database) Remove(key string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
db.m.Lock()
|
|
||||||
defer db.m.Unlock()
|
|
||||||
|
|
||||||
delete(db.data, key)
|
|
||||||
os.Remove(filepath.Join(db.basedir, key))
|
os.Remove(filepath.Join(db.basedir, key))
|
||||||
|
|
||||||
|
db.m.Lock()
|
||||||
|
delete(db.cache, key)
|
||||||
|
db.m.Unlock()
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user