Made maxBulkStringSize a command line option 'max-bulk-string-size'. It defaults to 32MB.

cmd/mtredisalize/main.go

@@ -16,9 +16,10 @@ import (
 )
 
 const (
-	Version        = "0.3"
-	GCDuration     = "24h"
-	ChangeDuration = "30s"
+	defaultMaxBulkStringSize = 32 * 1024 * 1024
+	Version                  = "0.3"
+	GCDuration               = "24h"
+	ChangeDuration           = "30s"
 )
 
 func usage() {
@@ -31,15 +32,16 @@ func usage() {
 func main() {
 
 	var (
-		port           int
-		host           string
-		driver         string
-		cacheSize      int
-		version        bool
-		interleaved    bool
-		changeUrl      string
-		gcDuration     string
-		changeDuration string
+		port              int
+		host              string
+		driver            string
+		cacheSize         int
+		version           bool
+		interleaved       bool
+		changeUrl         string
+		gcDuration        string
+		changeDuration    string
+		maxBulkStringSize int64
 	)
 
 	flag.Usage = usage
@@ -56,6 +58,8 @@ func main() {
 	flag.StringVar(&changeDuration,
 		"change-duration", ChangeDuration, "Duration to aggregate changes.")
 	flag.StringVar(&changeUrl, "change-url", "", "URL to send changes to.")
+	flag.Int64Var(&maxBulkStringSize, "max-bulk-string-size", defaultMaxBulkStringSize,
+		"max size of a bulk string to be accepted as input (in bytes).")
 	flag.Parse()
 
 	if version {
@@ -147,7 +151,7 @@ func main() {
 				log.Printf("Cannot create session: %s", err)
 				conn.Close()
 			} else {
-				go NewConnection(conn, session).Run()
+				go NewConnection(conn, session, maxBulkStringSize).Run()
 			}
 		case <-sigChan:
 			log.Println("Shutting down")