urdt/ussd
5
0
Fork 0
Code Issues 34 Pull Requests Actions Packages Projects Releases Wiki Activity

Fix missing byte allocation for typ serialize

Browse Source
This commit is contained in:
lash 2024-09-05 15:41:27 +01:00
parent 6fa5f49bc0
commit 62b7adb967
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/utils/db.go
View File

@ -18,10 +18,9 @@ const (
) )
func typToBytes(typ DataTyp) []byte { func typToBytes(typ DataTyp) []byte {
//var b []byte var b [2]byte
b := make([]byte, 2) binary.BigEndian.PutUint16(b[:], uint16(typ))
binary.BigEndian.PutUint16(b, uint16(typ)) return b[:]
return b
} }
func packKey(typ DataTyp, data []byte) []byte { func packKey(typ DataTyp, data []byte) []byte {
@ -31,9 +30,8 @@ func packKey(typ DataTyp, data []byte) []byte {
func ReadEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp) ([]byte, error) { func ReadEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp) ([]byte, error) {
store.SetPrefix(db.DATATYPE_USERDATA) store.SetPrefix(db.DATATYPE_USERDATA)
store.SetSession(sessionId) store.SetSession(sessionId)
k := packKey(typ, []byte(sessionId)) k := packKey(typ, []byte(sessionId))
//k := []byte(sessionId)
b, err := store.Get(ctx, k) b, err := store.Get(ctx, k)
if err != nil { if err != nil {
return nil, err return nil, err
@ -45,7 +43,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 { func WriteEntry(ctx context.Context, store db.Db, sessionId string, typ DataTyp, value []byte) error {
store.SetPrefix(db.DATATYPE_USERDATA) store.SetPrefix(db.DATATYPE_USERDATA)
store.SetSession(sessionId) store.SetSession(sessionId)
//k := packKey(typ, []byte(sessionId)) k := packKey(typ, []byte(sessionId))
k := []byte(sessionId)
return store.Put(ctx, k, value) return store.Put(ctx, k, value)
} }