pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-21 06:59:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

KeyExists keyspace function now removes the key if the key expiry is in the past.

Browse Source 
KeyData type moved to utils package to allow sharing between multiple packages.
Updated all commands and command tests to pass context object to KeyExists, KeyLock, keyUnlock, KeyRLock, and KeyRUnlock.
Create context object for each test in all test suites instead of just passing context.Background() to all functions that accept a context.
This commit is contained in:
Kelvin Mwinuka
2024-03-10 23:19:05 +08:00
parent 10f1aeab9e
commit c414da16b4
19 changed files with 1148 additions and 871 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/server/server.go
View File

@@ -21,17 +21,12 @@ import (
"time"
)
type KeyData struct {
value interface{}
expireAt time.Time
}
type Server struct {
Config utils.Config
ConnID atomic.Uint64
store map[string]KeyData
store map[string]utils.KeyData
keyLocks map[string]*sync.RWMutex
keyCreationLock *sync.Mutex
lfuCache struct {
@@ -77,7 +72,7 @@ func NewServer(opts Opts) *Server {
PubSub: opts.PubSub,
CancelCh: opts.CancelCh,
Commands: opts.Commands,
store: make(map[string]KeyData),
store: make(map[string]utils.KeyData),
keyLocks: make(map[string]*sync.RWMutex),
keyCreationLock: &sync.Mutex{},
}
@@ -105,13 +100,14 @@ func NewServer(opts Opts) *Server {
SetLatestSnapshotMilliseconds: server.SetLatestSnapshot,
GetLatestSnapshotMilliseconds: server.GetLatestSnapshot,
SetValue: func(key string, value interface{}) error {
if _, err := server.CreateKeyAndLock(context.Background(), key); err != nil {
ctx := context.Background()
if _, err := server.CreateKeyAndLock(ctx, key); err != nil {
return err
}
if err := server.SetValue(context.Background(), key, value); err != nil {
if err := server.SetValue(ctx, key, value); err != nil {
return err
}
server.KeyUnlock(key)
server.KeyUnlock(ctx, key)
return nil
},
})
@@ -123,13 +119,14 @@ func NewServer(opts Opts) *Server {
aof.WithFinishRewriteFunc(server.FinishRewriteAOF),
aof.WithGetStateFunc(server.GetState),
aof.WithSetValueFunc(func(key string, value interface{}) error {
if _, err := server.CreateKeyAndLock(context.Background(), key); err != nil {
ctx := context.Background()
if _, err := server.CreateKeyAndLock(ctx, key); err != nil {
return err
}
if err := server.SetValue(context.Background(), key, value); err != nil {
if err := server.SetValue(ctx, key, value); err != nil {
return err
}
server.KeyUnlock(key)
server.KeyUnlock(ctx, key)
return nil
}),
aof.WithHandleCommandFunc(func(command []byte) {