forked from urdt/ussd
52 lines
945 B
Go
52 lines
945 B
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.defalsify.org/vise.git/db"
|
|
)
|
|
|
|
const (
|
|
DATATYPE_USERSUB = 64
|
|
)
|
|
|
|
const (
|
|
SUBPREFIX_TIME = uint16(1)
|
|
)
|
|
|
|
type SubPrefixDb struct {
|
|
store db.Db
|
|
pfx []byte
|
|
}
|
|
|
|
func NewSubPrefixDb(store db.Db, pfx []byte) *SubPrefixDb {
|
|
return &SubPrefixDb{
|
|
store: store,
|
|
pfx: pfx,
|
|
}
|
|
}
|
|
|
|
func(s *SubPrefixDb) SetSession(sessionId string) {
|
|
s.store.SetSession(sessionId)
|
|
}
|
|
|
|
func(s *SubPrefixDb) toKey(k []byte) []byte {
|
|
return append(s.pfx, k...)
|
|
}
|
|
|
|
func(s *SubPrefixDb) Get(ctx context.Context, key []byte) ([]byte, error) {
|
|
s.store.SetPrefix(DATATYPE_USERSUB)
|
|
key = s.toKey(key)
|
|
v, err := s.store.Get(ctx, key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func(s *SubPrefixDb) Put(ctx context.Context, key []byte, val []byte) error {
|
|
s.store.SetPrefix(DATATYPE_USERSUB)
|
|
key = s.toKey(key)
|
|
return s.store.Put(ctx, key, val)
|
|
}
|