Merge pull request #867 from adamweiner/to_nsq_divide_by_zero_866

apps/to_nsq: fix --rate divide by zero bug
nsqio · Mar 24, 2017 · 8ad2491 · 8ad2491
2 parents 62c3858 + eaeb38c
commit 8ad2491
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/apps/to_nsq/to_nsq.go b/apps/to_nsq/to_nsq.go
@@ -68,7 +68,11 @@ func main() {
 
 	throttleEnabled := *rate >= 1
 	balance := int64(1)
-	interval := time.Second / time.Duration(*rate)
+	// avoid divide by 0 if !throttleEnabled
+	var interval time.Duration
+	if throttleEnabled {
+		interval = time.Second / time.Duration(*rate)
+	}
 	go func() {
 		if !throttleEnabled {
 			return