Initial Prometheus metrics support

bakins · Brian Akins · commit 1ea73371b3b9 · 2018-07-30T11:48:18.000-04:00
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/Gopkg.toml b/Gopkg.toml
@@ -21,3 +21,11 @@
 [[constraint]]
   name = "k8s.io/client-go"
   version = "~7.0.0"
+
+[[constraint]]
+  name = "github.com/prometheus/client_golang"
+  version = "0.8.0"
+
+[[override]]
+  name = "github.com/golang/protobuf"
+  version = "1.1.0"
diff --git a/main.go b/main.go
@@ -10,6 +10,7 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 	log "github.com/sirupsen/logrus"
 	"gopkg.in/alecthomas/kingpin.v2"
 
@@ -161,10 +162,21 @@ func main() {
 	)
 
 	if metricsAddress != "" {
+		http.Handle("/metrics", promhttp.Handler())
 		http.HandleFunc("/healthz",
 			func(w http.ResponseWriter, r *http.Request) {
 				fmt.Fprintln(w, "OK")
 			})
+		http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+			w.Write([]byte(`<html>
+					<head><title>chaoskube</title></head>
+					<body>
+					<h1>chaoskube</h1>
+					<p><a href="/metrics">Metrics</a></p>
+					<p><a href="/healthz">Health Check</a></p>
+					</body>
+					</html>`))
+		})
 		go func() {
 			if err := http.ListenAndServe(metricsAddress, nil); err != nil {
 				log.WithFields(log.Fields{
