wip
This commit is contained in:
@ -1,13 +1,3 @@
|
||||
module gitea.codeblob.work/pk/gut/settingsdb
|
||||
|
||||
go 1.21.6
|
||||
|
||||
require github.com/syndtr/goleveldb v1.0.0
|
||||
|
||||
require (
|
||||
github.com/golang/snappy v0.0.1 // indirect
|
||||
golang.org/x/net v0.17.0 // indirect
|
||||
golang.org/x/sys v0.16.0 // indirect
|
||||
golang.org/x/text v0.14.0 // indirect
|
||||
gopkg.in/yaml.v2 v2.2.2 // indirect
|
||||
)
|
||||
|
@ -1,27 +0,0 @@
|
||||
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
||||
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||
github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||
github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
|
||||
github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=
|
||||
github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
|
||||
github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
||||
github.com/onsi/ginkgo v1.7.0 h1:WSHQ+IS43OoUrWtD1/bbclrwK8TTH5hzp+umCiuxHgs=
|
||||
github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
||||
github.com/onsi/gomega v1.4.3 h1:RE1xgDvH7imwFD45h+u2SgIfERHlS2yNG4DObb5BSKU=
|
||||
github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
|
||||
github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
|
||||
github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
|
||||
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
|
||||
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=
|
||||
gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
|
||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
|
||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
||||
gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
|
||||
|
@ -2,67 +2,130 @@ package settingsdb
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"gitea.codeblob.work/pk/gut/generics"
|
||||
|
||||
"github.com/syndtr/goleveldb/leveldb"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type SettingsDB struct {
|
||||
db *leveldb.DB
|
||||
l sync.Mutex
|
||||
m map[string]any
|
||||
path string
|
||||
}
|
||||
|
||||
func NewOrLoadSettingsDB(path string) (*SettingsDB, error) {
|
||||
db, err := leveldb.OpenFile(path, nil)
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
data = []byte("{}")
|
||||
err := os.WriteFile(path, data, 0755)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
m := map[string]any{}
|
||||
err = json.Unmarshal(data, &m)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &SettingsDB{
|
||||
db: db,
|
||||
m: m,
|
||||
path: path,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *SettingsDB) Set(key string, value any) error {
|
||||
var data []byte
|
||||
s.l.Lock()
|
||||
defer s.l.Unlock()
|
||||
|
||||
switch v := value.(type) {
|
||||
case string:
|
||||
data = []byte(v)
|
||||
default:
|
||||
var err error
|
||||
data, err = json.Marshal(v)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
s.m[key] = value
|
||||
data, err := json.Marshal(s.m)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return s.db.Put([]byte(key), data, nil)
|
||||
return os.WriteFile(s.path, data, 0755)
|
||||
}
|
||||
|
||||
func (s *SettingsDB) GetString(key string) string {
|
||||
data, err := s.db.Get([]byte(key), nil)
|
||||
if err != nil {
|
||||
func (s *SettingsDB) Get(key string) any {
|
||||
s.l.Lock()
|
||||
defer s.l.Unlock()
|
||||
|
||||
v, ok := s.m[key]
|
||||
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
|
||||
return v
|
||||
}
|
||||
|
||||
func (s *SettingsDB) GetString(key string, defValue ...string) string {
|
||||
v := s.Get(key)
|
||||
if v == nil {
|
||||
if len(defValue) > 0 {
|
||||
return defValue[0]
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
return string(data)
|
||||
val, ok := v.(string)
|
||||
if !ok {
|
||||
if len(defValue) > 0 {
|
||||
return defValue[0]
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
return val
|
||||
}
|
||||
|
||||
func (s *SettingsDB) GetStrings(key string) []string {
|
||||
data, err := s.db.Get([]byte(key), nil)
|
||||
if err != nil {
|
||||
v := s.Get(key)
|
||||
|
||||
if v == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
value, err := generics.UnmarshalTo[[]string](data)
|
||||
if err != nil {
|
||||
val, ok := v.([]string)
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
|
||||
return *value
|
||||
return val
|
||||
}
|
||||
|
||||
func (s *SettingsDB) Close() {
|
||||
s.db.Close()
|
||||
func (s *SettingsDB) GetBool(key string) bool {
|
||||
v := s.Get(key)
|
||||
if v == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
val, ok := v.(bool)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
return val
|
||||
}
|
||||
|
||||
func (s *SettingsDB) Dump() (map[string]any, error) {
|
||||
data, err := os.ReadFile(s.path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
m := map[string]any{}
|
||||
err = json.Unmarshal(data, &m)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user