konstantinmds/ussd
1
0
Fork 0
forked from urdt/ussd
Code Pull Requests Activity

make packKey accessible from tests

Browse Source
This commit is contained in:
Carlosokumu 2024-09-07 16:25:29 +03:00
parent dd97531861
commit 01c13ec581
Signed by untrusted user: carlos
GPG Key ID: 7BD6BC8160A5C953
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/utils/db.go
View File

@ -33,15 +33,16 @@ func typToBytes(typ DataTyp) []byte {
return b[:]
}
func packKey(typ DataTyp, data []byte) []byte {
func PackKey(typ DataTyp, data []byte) []byte {
v := typToBytes(typ)
return append(v, data...)
}
func ReadEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp) ([]byte, error) {
store.SetPrefix(db.DATATYPE_USERDATA)
store.SetSession(sessionId)
k := packKey(typ, []byte(sessionId))
k := PackKey(typ, []byte(sessionId))
b, err := store.Get(ctx, k)
if err != nil {
return nil, err
@ -52,6 +53,6 @@ func ReadEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp)
func WriteEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp, value []byte) error {
store.SetPrefix(db.DATATYPE_USERDATA)
store.SetSession(sessionId)
k := packKey(typ, []byte(sessionId))
k := PackKey(typ, []byte(sessionId))
return store.Put(ctx, k, value)
}