add --redis flag (#52)

add --redis flag
flashbots · Sep 11, 2024 · d9fd104 · d9fd104
1 parent 53a6055
commit d9fd104
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/cmd/service/bidcollect.go b/cmd/service/bidcollect.go
@@ -128,6 +128,7 @@ var bidCollectCmd = &cobra.Command{
 			OutDir:                  outDir,
 			OutputTSV:               outputTSV,
 			RedisAddr:               redisAddr,
+			UseRedis:                useRedis,
 		}
 
 		bidCollector, err := bidcollect.NewBidCollector(&opts)

diff --git a/services/bidcollect/bid-processor.go b/services/bidcollect/bid-processor.go
@@ -27,6 +27,7 @@ type BidProcessorOpts struct {
 	OutDir    string
 	OutputTSV bool
 	RedisAddr string
+	UseRedis  bool
 }
 
 type OutFiles struct {
@@ -68,7 +69,7 @@ func NewBidProcessor(opts *BidProcessorOpts) (*BidProcessor, error) {
 		c.csvFileEnding = "csv"
 	}
 
-	if opts.RedisAddr != "" {
+	if opts.UseRedis && opts.RedisAddr != "" {
 		c.redisClient = redis.NewClient(&redis.Options{
 			Addr:     opts.RedisAddr,
 			Password: "", // no password set

diff --git a/services/bidcollect/bidcollector.go b/services/bidcollect/bidcollector.go
@@ -22,6 +22,7 @@ type BidCollectorOpts struct {
 	OutputTSV bool
 
 	RedisAddr string
+	UseRedis  bool
 }
 
 type BidCollector struct {
@@ -57,6 +58,7 @@ func NewBidCollector(opts *BidCollectorOpts) (c *BidCollector, err error) {
 		OutDir:    opts.OutDir,
 		OutputTSV: opts.OutputTSV,
 		RedisAddr: opts.RedisAddr,
+		UseRedis:  opts.UseRedis,
 	})
 	return c, err
 }