Skip to content

Commit

Permalink
refactor: replace if-else with switch (#777)
Browse files Browse the repository at this point in the history
  • Loading branch information
blight19 committed Mar 6, 2024
1 parent 256dd64 commit d27254c
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions cmd/redis-shake/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,8 @@ func main() {

// create reader
var theReader reader.Reader
if v.IsSet("sync_reader") {
switch {
case v.IsSet("sync_reader"):
opts := new(reader.SyncReaderOptions)
defaults.SetDefaults(opts)
err := v.UnmarshalKey("sync_reader", opts)
Expand All @@ -48,7 +49,7 @@ func main() {
theReader = reader.NewSyncStandaloneReader(ctx, opts)
log.Infof("create SyncStandaloneReader: %v", opts.Address)
}
} else if v.IsSet("scan_reader") {
case v.IsSet("scan_reader"):
opts := new(reader.ScanReaderOptions)
defaults.SetDefaults(opts)
err := v.UnmarshalKey("scan_reader", opts)
Expand All @@ -62,7 +63,7 @@ func main() {
theReader = reader.NewScanStandaloneReader(ctx, opts)
log.Infof("create ScanStandaloneReader: %v", opts.Address)
}
} else if v.IsSet("rdb_reader") {
case v.IsSet("rdb_reader"):
opts := new(reader.RdbReaderOptions)
defaults.SetDefaults(opts)
err := v.UnmarshalKey("rdb_reader", opts)
Expand All @@ -71,7 +72,7 @@ func main() {
}
theReader = reader.NewRDBReader(opts)
log.Infof("create RdbReader: %v", opts.Filepath)
} else if v.IsSet("aof_reader") {
case v.IsSet("aof_reader"):
opts := new(reader.AOFReaderOptions)
defaults.SetDefaults(opts)
err := v.UnmarshalKey("aof_reader", opts)
Expand All @@ -80,13 +81,13 @@ func main() {
}
theReader = reader.NewAOFReader(opts)
log.Infof("create AOFReader: %v", opts.Filepath)
} else {
default:
log.Panicf("no reader config entry found")
}

// create writer
var theWriter writer.Writer
if v.IsSet("redis_writer") {
switch {
case v.IsSet("redis_writer"):
opts := new(writer.RedisWriterOptions)
defaults.SetDefaults(opts)
err := v.UnmarshalKey("redis_writer", opts)
Expand All @@ -109,10 +110,9 @@ func main() {
entry.Argv = []string{"FLUSHALL"}
theWriter.Write(entry)
}
} else {
default:
log.Panicf("no writer config entry found")
}

// create status
status.Init(theReader, theWriter)

Expand Down

0 comments on commit d27254c

Please sign in to comment.