Print bandwidth limit information

Author: mohanson

cmd/daze/main.go

@@ -4,7 +4,6 @@ import (
 	"flag"
 	"fmt"
 	"log"
-	"math"
 	"net"
 	"net/http"
 	"net/http/pprof"
@@ -90,13 +89,15 @@ func main() {
 			}
 			log.Println("main: domain server is", *flDnserv)
 		}
+		rateLimits := func() *rate.Limits {
+			log.Println("main: bandwidth is set", *flLimits)
+			return rate.NewLimits(daze.SizeParser(*flLimits), time.Second)
+		}
 		switch *flProtoc {
 		case "ashe":
 			server := ashe.NewServer(*flListen, *flCipher)
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				server.Limits = rate.NewLimits(n, time.Second)
+				server.Limits = rateLimits()
 			}
 			defer server.Close()
 			doa.Nil(server.Run())
@@ -106,27 +107,21 @@ func main() {
 				server.Masker = *flExtend
 			}
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				server.Limits = rate.NewLimits(n, time.Second)
+				server.Limits = rateLimits()
 			}
 			defer server.Close()
 			doa.Nil(server.Run())
 		case "czar":
 			server := czar.NewServer(*flListen, *flCipher)
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				server.Limits = rate.NewLimits(n, time.Second)
+				server.Limits = rateLimits()
 			}
 			defer server.Close()
 			doa.Nil(server.Run())
 		case "dahlia":
 			server := dahlia.NewServer(*flListen, *flExtend, *flCipher)
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				server.Limits = rate.NewLimits(n, time.Second)
+				server.Limits = rateLimits()
 			}
 			defer server.Close()
 			doa.Nil(server.Run())
@@ -167,6 +162,10 @@ func main() {
 			}
 			log.Println("main: domain server is", *flDnserv)
 		}
+		rateLimits := func() *rate.Limits {
+			log.Println("main: bandwidth is set", *flLimits)
+			return rate.NewLimits(daze.SizeParser(*flLimits), time.Second)
+		}
 		switch *flProtoc {
 		case "ashe":
 			client := ashe.NewClient(*flServer, *flCipher)
@@ -176,9 +175,7 @@ func main() {
 				Cidr: *flCidrls,
 			}))
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				locale.Limits = rate.NewLimits(n, time.Second)
+				locale.Limits = rateLimits()
 			}
 			defer locale.Close()
 			doa.Nil(locale.Run())
@@ -190,9 +187,7 @@ func main() {
 				Cidr: *flCidrls,
 			}))
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				locale.Limits = rate.NewLimits(n, time.Second)
+				locale.Limits = rateLimits()
 			}
 			defer locale.Close()
 			doa.Nil(locale.Run())
@@ -205,18 +200,14 @@ func main() {
 				Cidr: *flCidrls,
 			}))
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				locale.Limits = rate.NewLimits(n, time.Second)
+				locale.Limits = rateLimits()
 			}
 			defer locale.Close()
 			doa.Nil(locale.Run())
 		case "dahlia":
 			client := dahlia.NewClient(*flListen, *flServer, *flCipher)
 			if *flLimits != "" {
-				n := daze.SizeParser(*flLimits)
-				doa.Doa(n <= math.MaxInt)
-				client.Limits = rate.NewLimits(n, time.Second)
+				client.Limits = rateLimits()
 			}
 			defer client.Close()
 			doa.Nil(client.Run())