mirror of
https://github.com/EchoVault/SugarDB.git
synced 2025-10-06 08:27:04 +08:00
Implemented LoadModules method to load external modules at runtime.
Implemented UnloadModules method to remove modules at runtime. Implemented ListModules method to list the current loaded modules. Implemented "MODULE LOAD", "MODULE UNLOAD", and "MODULE LIST" commands.
This commit is contained in:
68
volumes/modules/module_set/module_set.go
Normal file
68
volumes/modules/module_set/module_set.go
Normal file
@@ -0,0 +1,68 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
var Command string = "Module.Set"
|
||||
|
||||
var Categories []string = []string{"write", "fast"}
|
||||
|
||||
var Description string = `(Module.Set key value) This module stores the given value at the specified key.
|
||||
The value must be an integer`
|
||||
|
||||
var Sync bool = true
|
||||
|
||||
func KeyExtractionFunc(cmd []string, args ...string) ([]string, []string, error) {
|
||||
if len(cmd) != 3 {
|
||||
return nil, nil, fmt.Errorf("wrong no of args for %s command", Command)
|
||||
}
|
||||
return []string{}, cmd[1:2], nil
|
||||
}
|
||||
|
||||
func HandlerFunc(
|
||||
ctx context.Context,
|
||||
command []string,
|
||||
keyExists func(ctx context.Context, key string) bool,
|
||||
keyLock func(ctx context.Context, key string) (bool, error),
|
||||
keyUnlock func(ctx context.Context, key string),
|
||||
keyRLock func(ctx context.Context, key string) (bool, error),
|
||||
keyRUnlock func(ctx context.Context, key string),
|
||||
createKeyAndLock func(ctx context.Context, key string) (bool, error),
|
||||
getValue func(ctx context.Context, key string) interface{},
|
||||
setValue func(ctx context.Context, key string, value interface{}) error,
|
||||
args ...string) ([]byte, error) {
|
||||
|
||||
_, writeKeys, err := KeyExtractionFunc(command, args...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
key := writeKeys[0]
|
||||
|
||||
if !keyExists(ctx, key) {
|
||||
_, err := createKeyAndLock(ctx, key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
} else {
|
||||
_, err := keyLock(ctx, key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
defer keyUnlock(ctx, key)
|
||||
|
||||
value, err := strconv.ParseInt(command[2], 10, 64)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = setValue(ctx, key, value)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return []byte("+OK\r\n"), nil
|
||||
}
|
Reference in New Issue
Block a user