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

Added config flags to choose between aof restore and snapshot restore in standalone mode.

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2024-02-01 04:42:44 +08:00
parent 04ddd46d2b
commit 15d289c33d
2 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/server/server.go
View File

@@ -256,9 +256,12 @@ func (server *Server) Start(ctx context.Context) {
StartRewriteAOF: server.StartRewriteAOF,
FinishRewriteAOF: server.FinishRewriteAOF,
})
if err := server.AOFEngine.Restore(ctx); err != nil {
if conf.RestoreAOF && !conf.RestoreSnapshot {
err := server.AOFEngine.Restore(ctx)
if err != nil {
log.Println(err)
}
}
server.AOFEngine.Start(ctx)
// Initialize and start standalone snapshot engine
server.SnapshotEngine = snapshot.NewSnapshotEngine(snapshot.Opts{
@@ -272,9 +275,12 @@ func (server *Server) Start(ctx context.Context) {
KeyUnlock: server.KeyUnlock,
SetValue: server.SetValue,
})
if err := server.SnapshotEngine.Restore(ctx); err != nil {
if conf.RestoreSnapshot {
err := server.SnapshotEngine.Restore(ctx)
if err != nil {
log.Println(err)
}
}
server.SnapshotEngine.Start(ctx)
}

6
src/utils/config.go
View File

@@ -30,6 +30,8 @@ type Config struct {
Password string `json:"password" yaml:"password"`
SnapShotThreshold uint64 `json:"snapshotThreshold" yaml:"snapshotThreshold"`
SnapshotInterval uint `json:"snapshotInterval" yaml:"snapshotInterval"`
RestoreSnapshot bool `json:"restoreSnapshot" yaml:"restoreSnapshot"`
RestoreAOF bool `json:"restoreAOF" yaml:"restoreAOF"`
}
func GetConfig() (Config, error) {
@@ -49,6 +51,8 @@ func GetConfig() (Config, error) {
aclConfig := flag.String("aclConfig", "", "ACL config file path.")
snapshotThreshold := flag.Uint64("snapshotThreshold", 1000, "The number of entries that trigger a snapshot. Default is 1000.")
snapshotInterval := flag.Uint("snapshotInterval", 60*5, "The time interval between snapshots (in seconds). Default is 5 minutes.")
restoreSnapshot := flag.Bool("restoreSnapshot", false, "This flag prompts the server to restore state from snapshot when set to true. Only works in standalone mode. Higher priority than restoreAOF.")
restoreAOF := flag.Bool("restoreAOF", false, "This flag prompts the server to restore state from append-only logs. Only works in standalone mode. Lower priority than restoreSnapshot.")
forwardCommand := flag.Bool(
"forwardCommand",
false,
@@ -93,6 +97,8 @@ It is a plain text value by default but you can provide a SHA256 hash by adding
Password: *password,
SnapShotThreshold: *snapshotThreshold,
SnapshotInterval: *snapshotInterval,
RestoreSnapshot: *restoreSnapshot,
RestoreAOF: *restoreAOF,
}
if len(*config) > 0 {