From b85496922f98b31cbc938c0e5d2f79401fe6c5a8 Mon Sep 17 00:00:00 2001
From: Jonathan Boulle <jonathanboulle@gmail.com>
Date: Wed, 5 Nov 2014 11:31:13 -0800
Subject: [PATCH] etcdmain: simplify proxy start logic

---
 etcdmain/etcd.go               | 28 ++++++++++++++--------------
 hack/insta-discovery/discovery |  3 ++-
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/etcdmain/etcd.go b/etcdmain/etcd.go
index 4a3fc6c8a..8899333c8 100644
--- a/etcdmain/etcd.go
+++ b/etcdmain/etcd.go
@@ -141,8 +141,8 @@ func init() {
 
 func Main() {
 	fs.Usage = flags.UsageWithIgnoredFlagsFunc(fs, ignored)
-	err := fs.Parse(os.Args[1:])
-	switch err {
+	perr := fs.Parse(os.Args[1:])
+	switch perr {
 	case nil:
 	case flag.ErrHelp:
 		os.Exit(0)
@@ -157,20 +157,20 @@ func Main() {
 
 	flags.SetFlagsFromEnv(fs)
 
-	if proxyFlag.String() == proxyFlagOff {
-		if err := startEtcd(); err == nil {
-			// Block indefinitely
-			<-make(chan struct{})
-		} else {
-			if err == discovery.ErrFullCluster && fallbackFlag.String() == fallbackFlagProxy {
-				fmt.Printf("etcd: discovery cluster full, falling back to %s", fallbackFlagProxy)
-			} else {
-				log.Fatalf("etcd: %v", err)
-			}
+	var err error
+	shouldProxy := proxyFlag.String() != proxyFlagOff
+	if !shouldProxy {
+		err = startEtcd()
+		if err == discovery.ErrFullCluster && fallbackFlag.String() == fallbackFlagProxy {
+			log.Printf("etcd: discovery cluster full, falling back to %s", fallbackFlagProxy)
+			shouldProxy = true
 		}
 	}
-	if err = startProxy(); err != nil {
-		log.Fatalf("proxy: %v", err)
+	if shouldProxy {
+		err = startProxy()
+	}
+	if err != nil {
+		log.Fatalf("etcd: %v", err)
 	}
 	// Block indefinitely
 	<-make(chan struct{})
diff --git a/hack/insta-discovery/discovery b/hack/insta-discovery/discovery
index c80d5c758..b33b1512b 100755
--- a/hack/insta-discovery/discovery
+++ b/hack/insta-discovery/discovery
@@ -1,6 +1,7 @@
 #!/bin/sh
 
-disc=$(curl https://discovery.etcd.io/new?size=3)
+rm -rf infra*.etcd
+disc=$(curl -s https://discovery.etcd.io/new?size=3)
 echo ETCD_DISCOVERY=${disc} > .env
 echo "setup discovery start your cluster"
 cat .env