Initial commit

Author: sgreben

.gitignore

@@ -0,0 +1,3 @@
+/*.json
+/binaries/
+/release/

Gopkg.lock

@@ -0,0 +1,30 @@
+
+# Gopkg.toml example
+#
+# Refer to https://github.com/golang/dep/blob/master/docs/Gopkg.toml.md
+# for detailed Gopkg.toml documentation.
+#
+# required = ["github.com/user/thing/cmd/thing"]
+# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
+#
+# [[constraint]]
+#   name = "github.com/user/project"
+#   version = "1.0.0"
+#
+# [[constraint]]
+#   name = "github.com/user/project2"
+#   branch = "dev"
+#   source = "github.com/myfork/project2"
+#
+# [[override]]
+#  name = "github.com/x/y"
+#  version = "2.4.0"
+
+
+[[constraint]]
+  name = "github.com/gonum/stat"
+  branch = "master"
+
+[[constraint]]
+  name = "github.com/schollz/progressbar"
+  branch = "master"

Gopkg.toml

@@ -0,0 +1,51 @@
+VERSION = 1.0.0
+
+APP      := tcp-time
+PACKAGES := $(shell go list -f {{.Dir}} ./...)
+GOFILES  := $(addsuffix /*.go,$(PACKAGES))
+GOFILES  := $(wildcard $(GOFILES))
+
+.PHONY: clean release docker docker-latest
+
+# go get -u github.com/github/hub
+release: zip
+	git push
+	hub release delete $(VERSION) || true
+	hub release create $(VERSION) -m "$(VERSION)" -a release/$(APP)_$(VERSION)_osx_x86_64.zip -a release/$(APP)_$(VERSION)_windows_x86_64.zip -a release/$(APP)_$(VERSION)_linux_x86_64.zip
+
+docker: binaries/linux_x86_64/$(APP)
+	docker build -t quay.io/sergey_grebenshchikov/$(APP):v$(VERSION) .
+	docker push quay.io/sergey_grebenshchikov/$(APP):v$(VERSION)
+
+docker-latest: docker
+	docker tag quay.io/sergey_grebenshchikov/$(APP):v$(VERSION) quay.io/sergey_grebenshchikov/$(APP):latest
+	docker push quay.io/sergey_grebenshchikov/$(APP):latest
+
+zip: release/$(APP)_$(VERSION)_osx_x86_64.zip release/$(APP)_$(VERSION)_windows_x86_64.zip release/$(APP)_$(VERSION)_linux_x86_64.zip
+
+binaries: binaries/osx_x86_64/$(APP) binaries/windows_x86_64/$(APP).exe binaries/linux_x86_64/$(APP)
+
+clean: 
+	rm -rf binaries/
+	rm -rf release/
+
+release/$(APP)_$(VERSION)_osx_x86_64.zip: binaries/osx_x86_64/$(APP)
+	mkdir -p release
+	cd ./binaries/osx_x86_64 && zip -r -D ../../release/$(APP)_$(VERSION)_osx_x86_64.zip $(APP)
+	
+binaries/osx_x86_64/$(APP): $(GOFILES)
+	GOOS=darwin GOARCH=amd64 go build -ldflags "-X main.version=$(VERSION)" -o binaries/osx_x86_64/$(APP) ./cmd/$(APP)
+
+release/$(APP)_$(VERSION)_windows_x86_64.zip: binaries/windows_x86_64/$(APP).exe
+	mkdir -p release
+	cd ./binaries/windows_x86_64 && zip -r -D ../../release/$(APP)_$(VERSION)_windows_x86_64.zip $(APP).exe
+	
+binaries/windows_x86_64/$(APP).exe: $(GOFILES)
+	GOOS=windows GOARCH=amd64 go build -ldflags "-X main.version=$(VERSION)" -o binaries/windows_x86_64/$(APP).exe ./cmd/$(APP)
+
+release/$(APP)_$(VERSION)_linux_x86_64.zip: binaries/linux_x86_64/$(APP)
+	mkdir -p release
+	cd ./binaries/linux_x86_64 && zip -r -D ../../release/$(APP)_$(VERSION)_linux_x86_64.zip $(APP)
+	
+binaries/linux_x86_64/$(APP): $(GOFILES)
+	GOOS=linux GOARCH=amd64 go build -ldflags "-X main.version=$(VERSION)" -o binaries/linux_x86_64/$(APP) ./cmd/$(APP)

Makefile

@@ -0,0 +1,99 @@
+# tcp-time
+
+Repeatedly measures TCP connection durations to a given target. Prints the measurements and a summary (mean, standard deviation, quantiles, histogram) as JSON to stdout.
+
+<!-- TOC -->
+
+- [Get it](#get-it)
+- [Use it](#use-it)
+- [Example](#example)
+
+<!-- /TOC -->
+
+## Get it
+
+```bash
+go get -u github.com/sgreben/tcp-time/cmd/tcp-time
+```
+
+Or [download the binary](https://github.com/sgreben/tcp-time/releases/latest) from the releases page. 
+
+```bash
+# Linux
+curl -LO https://github.com/sgreben/tcp-time/releases/download/1.0.0/tcp-time_1.0.0_linux_x86_64.zip
+unzip tcp-time_1.0.0_linux_x86_64.zip
+
+# OS X
+curl -LO https://github.com/sgreben/tcp-time/releases/download/1.0.0/tcp-time_1.0.0_osx_x86_64.zip
+unzip tcp-time_1.0.0_osx_x86_64.zip
+
+# Windows
+curl -LO https://github.com/sgreben/tcp-time/releases/download/1.0.0/tcp-time_1.0.0_windows_x86_64.zip
+unzip tcp-time_1.0.0_windows_x86_64.zip
+```
+
+## Use it
+
+```text
+Usage of tcp-time:
+  -target string
+    	host:port to ping. (default "duckduckgo.com:443")
+  -n int
+    	Number of connections to make. (default 10)
+  -p int
+    	Number of connections to make in parallel. (default 3)
+  -b int
+    	Number of histogram bins. (default 5)
+  -progress
+    	Print a progress bar to stderr.
+  -debug
+    	Print debug logs to stderr.
+```
+
+## Example
+
+```bash
+$ tcp-time -n 1000 | jq .
+{
+  "Measurements": [
+      {
+      "Valid": true,
+      "Duration": 40605393
+    },
+    {
+      "Valid": true,
+      "Duration": 35469927
+    },
+    # ...
+  ],
+  "Summary": {
+    "All": {
+      "Mean": 36694356.154,
+      "StdDev": 31847099.84842516,
+      "Quantiles": [ 32873052, 34962729, 36098973, 37387095, 89838530 ],
+      "Histogram": [
+        {
+          "Label": "32.873052ms",
+          "Value": 32873052,
+          "Count": 990
+        },
+        {
+          "Label": "47.114421ms",
+          "Value": 47114421,
+          "Count": 2
+        },
+        {
+          "Label": "61.355791ms",
+          "Value": 61355791,
+          "Count": 2
+        },
+        {
+          "Label": "75.597161ms",
+          "Value": 75597161,
+          "Count": 6
+        }
+      ]
+    }
+  }
+}
+```

README.md

@@ -0,0 +1,80 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"io/ioutil"
+	"log"
+	"os"
+	"sync"
+
+	"github.com/schollz/progressbar"
+	"golang.org/x/sync/semaphore"
+)
+
+type configuration struct {
+	Target        string
+	N             int
+	Parallel      int
+	HistogramBins int
+	Debug         bool
+	Progress      bool
+}
+
+type output struct {
+	Measurements measurements
+	Summary      summary
+}
+
+var config configuration
+
+func init() {
+	flag.StringVar(&config.Target, "target", "duckduckgo.com:443", "host:port to ping.")
+	flag.IntVar(&config.N, "n", 10, "Number of pings to make.")
+	flag.IntVar(&config.Parallel, "p", 3, "Number of pings to make in parallel.")
+	flag.IntVar(&config.HistogramBins, "b", 5, "Number of histogram bins.")
+	flag.BoolVar(&config.Debug, "debug", false, "Print debug logs to stderr.")
+	flag.BoolVar(&config.Progress, "progress", false, "Print a progress bar to stderr.")
+	flag.Parse()
+	log.SetOutput(os.Stderr)
+	if !config.Debug {
+		log.SetOutput(ioutil.Discard)
+	}
+}
+
+func main() {
+	var output output
+	var mu sync.Mutex
+	var wg sync.WaitGroup
+	ctx := context.Background()
+	limit := semaphore.NewWeighted(int64(config.Parallel))
+	var bar *progressbar.ProgressBar
+	if config.Progress {
+		bar = progressbar.New(config.N)
+		bar.SetWriter(os.Stderr)
+	}
+
+	for i := 0; i < config.N; i++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			if config.Progress {
+				defer bar.Add(1)
+			}
+			limit.Acquire(ctx, 1)
+			defer limit.Release(1)
+			d, err := connectDuration(config.Target)
+			mu.Lock()
+			defer mu.Unlock()
+			output.Measurements.append(sample{
+				Valid:    err == nil,
+				Duration: d,
+			})
+		}()
+	}
+	wg.Wait()
+	output.Summary = output.Measurements.summary()
+	enc := json.NewEncoder(os.Stdout)
+	enc.Encode(output)
+}

cmd/tcp-time/main.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"sort"
+	"time"
+)
+
+type sample struct {
+	Valid    bool
+	Duration time.Duration
+}
+
+type measurements []sample
+
+func (m *measurements) append(s sample) {
+	*m = append(*m, s)
+}
+
+func (m *measurements) invalidCount() (out int) {
+	for _, s := range *m {
+		if !s.Valid {
+			out++
+		}
+	}
+	return
+}
+func (m *measurements) allSeconds() []float64 {
+	out := make([]float64, len(*m))
+	for i, s := range *m {
+		out[i] = float64(s.Duration)
+	}
+	sort.Float64s(out)
+	return out
+}
+
+func (m *measurements) validSeconds() []float64 {
+	out := make([]float64, 0, len(*m))
+	for _, s := range *m {
+		if s.Valid {
+			out = append(out, float64(s.Duration))
+		}
+	}
+	sort.Float64s(out)
+	return out
+}