forked from urdt/ussd
121 lines
2.2 KiB
Go
121 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.defalsify.org/vise.git/db"
|
|
"git.defalsify.org/vise.git/lang"
|
|
gdbmdb "git.defalsify.org/vise.git/db/gdbm"
|
|
)
|
|
|
|
var (
|
|
dbC map[string]chan db.Db
|
|
)
|
|
|
|
type ThreadGdbmDb struct {
|
|
db db.Db
|
|
connStr string
|
|
}
|
|
|
|
func NewThreadGdbmDb() *ThreadGdbmDb {
|
|
if dbC == nil {
|
|
dbC = make(map[string]chan db.Db)
|
|
}
|
|
return &ThreadGdbmDb{}
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Connect(ctx context.Context, connStr string) error {
|
|
var ok bool
|
|
_, ok = dbC[connStr]
|
|
if ok {
|
|
logg.WarnCtxf(ctx, "already registered thread gdbm, skipping", "connStr", connStr)
|
|
return nil
|
|
}
|
|
gdb := gdbmdb.NewGdbmDb()
|
|
err := gdb.Connect(ctx, connStr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dbC[connStr] = make(chan db.Db, 1)
|
|
dbC[connStr]<- gdb
|
|
tdb.connStr = connStr
|
|
return nil
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) reserve() {
|
|
if tdb.db == nil {
|
|
tdb.db = <-dbC[tdb.connStr]
|
|
}
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) release() {
|
|
if tdb.db == nil {
|
|
return
|
|
}
|
|
dbC[tdb.connStr] <- tdb.db
|
|
tdb.db = nil
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) SetPrefix(pfx uint8) {
|
|
tdb.reserve()
|
|
tdb.db.SetPrefix(pfx)
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) SetSession(sessionId string) {
|
|
tdb.reserve()
|
|
tdb.db.SetSession(sessionId)
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) SetLanguage(lng *lang.Language) {
|
|
tdb.reserve()
|
|
tdb.db.SetLanguage(lng)
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Safe() bool {
|
|
tdb.reserve()
|
|
v := tdb.db.Safe()
|
|
tdb.release()
|
|
return v
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Prefix() uint8 {
|
|
tdb.reserve()
|
|
v := tdb.db.Prefix()
|
|
tdb.release()
|
|
return v
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) SetLock(typ uint8, locked bool) error {
|
|
tdb.reserve()
|
|
err := tdb.db.SetLock(typ, locked)
|
|
tdb.release()
|
|
return err
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Put(ctx context.Context, key []byte, val []byte) error {
|
|
tdb.reserve()
|
|
err := tdb.db.Put(ctx, key, val)
|
|
tdb.release()
|
|
return err
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Get(ctx context.Context, key []byte) ([]byte, error) {
|
|
tdb.reserve()
|
|
v, err := tdb.db.Get(ctx, key)
|
|
tdb.release()
|
|
return v, err
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Close() error {
|
|
tdb.reserve()
|
|
close(dbC[tdb.connStr])
|
|
delete(dbC, tdb.connStr)
|
|
err := tdb.db.Close()
|
|
tdb.db = nil
|
|
return err
|
|
}
|
|
|
|
func(tdb *ThreadGdbmDb) Dump(ctx context.Context, key []byte) (*db.Dumper, error) {
|
|
return tdb.db.Dump(ctx, key)
|
|
}
|