add in pollEndpoint dep

Author: whyrusleeping

sharness/Makefile

@@ -20,6 +20,7 @@ BINS += bin/ipfs-3-to-4
 BINS += bin/ipfs-update
 BINS += bin/random-files
 BINS += bin/go-sleep
+BINS += bin/pollEndpoint
 
 # Source files location
 FS_REPO_SRC = ../
@@ -28,6 +29,7 @@ IPFS_1_TO_2_SRC = ../ipfs-1-to-2
 IPFS_2_TO_3_SRC = ../ipfs-2-to-3
 IPFS_3_TO_4_SRC = ../ipfs-3-to-4
 IPFS_UPDATE_SRC = ../Godeps/_workspace/src/github.com/ipfs/ipfs-update
+POLL_ENDPOINT_SRC = ./dependencies/pollEndpoint
 
 # User might want to override those on the command line
 GOFLAGS =
@@ -93,6 +95,10 @@ bin/ipfs-3-to-4: $(call find_go_files, $(IPFS_3_TO_4_SRC)) BUILD-OPTIONS
 	@echo "*** installing $@ ***"
 	go build $(GOFLAGS) -o $@ $(IPFS_3_TO_4_SRC)
 
+bin/pollEndpoint: $(call find_go_files, $(POLL_ENDPOINT_SRC)) BUILD-OPTIONS
+	@echo "*** installing $@ ***"
+	go build $(GOFLAGS) -o $@ $(POLL_ENDPOINT_SRC)
+
 BUILD-OPTIONS: FORCE
 	@bin/checkflags '$@' '$(GOFLAGS)' '*** new Go flags ***'
 

sharness/dependencies/pollEndpoint/main.go

@@ -0,0 +1,88 @@
+// pollEndpoint is a helper utility that waits for a http endpoint to be reachable and return with http.StatusOK
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+	"time"
+
+	logging "gx/ipfs/QmNQynaz7qfriSUJkiEZUrm2Wen1u3Kj9goZzWtrPyu7XR/go-log"
+	manet "gx/ipfs/QmPpRcbNUXauP3zWZ1NJMLWpe4QnmEHrd2ba2D3yqWznw7/go-multiaddr-net"
+	ma "gx/ipfs/QmYzDkkgAEmrcNzFCiYo6L1dTX4EAG1gZkbtdbd9trL4vd/go-multiaddr"
+)
+
+var (
+	host     = flag.String("host", "/ip4/127.0.0.1/tcp/5001", "the multiaddr host to dial on")
+	endpoint = flag.String("ep", "/version", "which http endpoint path to hit")
+	tries    = flag.Int("tries", 10, "how many tries to make before failing")
+	timeout  = flag.Duration("tout", time.Second, "how long to wait between attempts")
+	verbose  = flag.Bool("v", false, "verbose logging")
+)
+
+var log = logging.Logger("pollEndpoint")
+
+func main() {
+	flag.Parse()
+
+	// extract address from host flag
+	addr, err := ma.NewMultiaddr(*host)
+	if err != nil {
+		log.Fatal("NewMultiaddr() failed: ", err)
+	}
+	p := addr.Protocols()
+	if len(p) < 2 {
+		log.Fatal("need two protocols in host flag (/ip/tcp): ", addr)
+	}
+	_, host, err := manet.DialArgs(addr)
+	if err != nil {
+		log.Fatal("manet.DialArgs() failed: ", err)
+	}
+
+	if *verbose { // lower log level
+		logging.SetDebugLogging()
+	}
+
+	// construct url to dial
+	var u url.URL
+	u.Scheme = "http"
+	u.Host = host
+	u.Path = *endpoint
+
+	// show what we got
+	start := time.Now()
+	log.Debug("starting at %s, tries: %d, timeout: %s, url: %s", start, *tries, *timeout, u)
+
+	for *tries > 0 {
+
+		err := checkOK(http.Get(u.String()))
+		if err == nil {
+			log.Debugf("ok -  endpoint reachable with %d tries remaining, took %s", *tries, time.Since(start))
+			os.Exit(0)
+		}
+		log.Debug("get failed: ", err)
+		time.Sleep(*timeout)
+		*tries--
+	}
+
+	log.Error("failed.")
+	os.Exit(1)
+}
+
+func checkOK(resp *http.Response, err error) error {
+	if err == nil { // request worked
+		defer resp.Body.Close()
+		if resp.StatusCode == http.StatusOK {
+			return nil
+		}
+		body, err := ioutil.ReadAll(resp.Body)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "pollEndpoint: ioutil.ReadAll() Error: %s", err)
+		}
+		return fmt.Errorf("Response not OK. %d %s %q", resp.StatusCode, resp.Status, string(body))
+	}
+	return err
+}