In registry, make consul datacenter a flag (#29)

* In registry, make consul datacenter a flag * Change datacenter flag default to Consul's default
RackHD · Sep 23, 2016 · 4b31f10 · 4b31f10
1 parent 0ad0bf5
commit 4b31f10
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/registry/cmd/registry/main.go b/registry/cmd/registry/main.go
@@ -13,10 +13,11 @@ import (
 )
 
 var binaryName, buildDate, buildUser, commitHash, goVersion, osArch, releaseVersion string
-var backendAddr string
+var backendAddr, datacenter string
 
 func init() {
 	flag.StringVar(&backendAddr, "address", "127.0.0.1:8500", "Address:port of the backend store")
+	flag.StringVar(&datacenter, "datacenter", "dc1", "Datacenter name of the backend storage")
 }
 
 func main() {
@@ -31,7 +32,7 @@ func main() {
 	flag.Parse()
 
 	consul.Register()
-	registry, err := registry.NewRegistry(libreg.CONSUL, "dc-docker", backendAddr)
+	registry, err := registry.NewRegistry(libreg.CONSUL, datacenter, backendAddr)
 	if err != nil {
 		log.Fatalf("Error creating new Service Registry: %s", err)
 	}

diff --git a/registry/docker-compose-registry.yaml b/registry/docker-compose-registry.yaml
@@ -40,7 +40,7 @@ services:
       - "8500/udp"
       - "8600"
       - "8600/udp"
-    command: "-address=consulclient:8500"
+    command: "-address=consulclient:8500 -datacenter=dc-docker"
     depends_on:
       - consulclient
   ssdpspoofer: