From e1a85c507dbbd143ddeeb5e32c448e49fbeb3fec Mon Sep 17 00:00:00 2001
From: brettlangdon <brett@blangdon.com>
Date: Thu, 28 Jan 2016 09:36:49 -0500
Subject: [PATCH] add --workers to main

---
 cmd/realm/main.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/cmd/realm/main.go b/cmd/realm/main.go
index 1ea5270..a21db3e 100644
--- a/cmd/realm/main.go
+++ b/cmd/realm/main.go
@@ -3,21 +3,28 @@ package main
 import (
 	"log"
 	"os"
+	"runtime"
 
 	"github.com/alexflint/go-arg"
 	"github.com/brettlangdon/realm"
 )
 
 var args struct {
-	Zones  []string `arg:"--zone,positional,help:DNS zone files to serve from this server"`
-	Bind   string   `arg:"help:[<host>]:<port> to bind too"`
-	StatsD string   `arg:"--statsd,help:<host>:<port> to send StatsD metrics to"`
+	Zones   []string `arg:"--zone,positional,help:DNS zone files to serve from this server"`
+	Bind    string   `arg:"help:[<host>]:<port> to bind too"`
+	StatsD  string   `arg:"--statsd,help:<host>:<port> to send StatsD metrics to"`
+	Workers int      `arg:"--workers,help:number of workers to start [default: $GOMAXPROCS]`
 }
 
 func main() {
 	args.Bind = ":53"
 	argParser := arg.MustParse(&args)
 
+	// Control number of workers via GOMAXPROCS
+	if args.Workers > 0 {
+		runtime.GOMAXPROCS(args.Workers)
+	}
+
 	if len(args.Zones) == 0 {
 		log.Println("must supply at least 1 zone file to serve")
 		argParser.WriteUsage(os.Stderr)