add buffer count metric

api/routes.go

@@ -6,9 +6,9 @@ import (
 	"github.com/lbryio/lbrytv/app/publish"
 	"github.com/lbryio/lbrytv/app/users"
 	"github.com/lbryio/lbrytv/internal/status"
-	"github.com/prometheus/client_golang/prometheus/promhttp"
 
 	"github.com/gorilla/mux"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
 // InstallRoutes sets up global API handlers
@@ -31,6 +31,7 @@ func InstallRoutes(proxyService *proxy.ProxyService, r *mux.Router) {
 	internalRouter.Handle("/metrics", promhttp.Handler())
 	internalRouter.HandleFunc("/status", status.GetStatus)
 	internalRouter.HandleFunc("/whoami", status.WhoAMI)
+	internalRouter.HandleFunc("/ui_metric", status.TrackUIMetric)
 
 	player.InstallRoutes(r)
 }

go.mod

@@ -25,6 +25,7 @@ require (
 	github.com/rubenv/sql-migrate v0.0.0-20190618074426-f4d34eae5a5c
 	github.com/sirupsen/logrus v1.4.2
 	github.com/spf13/afero v1.2.2 // indirect
+	github.com/spf13/cast v1.3.0
 	github.com/spf13/cobra v0.0.5
 	github.com/spf13/jwalterweatherman v1.1.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect

internal/metrics/metrics.go

@@ -10,6 +10,7 @@ const (
 	nsIAPI    = "iapi"
 	nsProxy   = "proxy"
 	nsLbrynet = "lbrynet"
+	nsUI      = "ui"
 
 	LabelSource   = "source"
 	LabelInstance = "instance"
@@ -120,4 +121,11 @@ var (
 		Name:      "count",
 		Help:      "Number of wallets currently loaded",
 	}, []string{LabelSource})
+
+	UIBufferCount = promauto.NewCounter(prometheus.CounterOpts{
+		Namespace: nsUI,
+		Subsystem: "buffer",
+		Name:      "count",
+		Help:      "Video buffer events",
+	})
 )

internal/status/status.go

@@ -7,6 +7,7 @@ import (
 	"time"
 
 	"github.com/lbryio/lbrytv/app/router"
+	"github.com/lbryio/lbrytv/internal/metrics"
 	"github.com/lbryio/lbrytv/internal/monitor"
 
 	ljsonrpc "github.com/lbryio/lbry.go/v2/extras/jsonrpc"
@@ -63,7 +64,7 @@ func GetStatus(w http.ResponseWriter, req *http.Request) {
 		sdks := router.GetSDKServerList()
 		for _, s := range sdks {
 			c := ljsonrpc.NewClient(s.Address)
-			c.SetRPCTimeout(5*time.Second)
+			c.SetRPCTimeout(5 * time.Second)
 			status, err := c.Status()
 			srv := ServerItem{Address: s.Address, Status: StatusOK}
 			if err != nil {
@@ -118,3 +119,23 @@ func WhoAMI(w http.ResponseWriter, req *http.Request) {
 	respByte, _ := json.MarshalIndent(&details, "", "  ")
 	w.Write(respByte)
 }
+
+func TrackUIMetric(w http.ResponseWriter, req *http.Request) {
+	w.Header().Add("content-type", "application/json; charset=utf-8")
+	resp := make(map[string]string)
+
+	metricName := req.FormValue("name")
+	resp["name"] = metricName
+
+	switch metricName {
+	case "buffer":
+		metrics.UIBufferCount.Inc()
+		resp["status"] = "ok"
+	default:
+		w.WriteHeader(http.StatusBadRequest)
+		resp["error"] = "Unknown metric"
+	}
+
+	respByte, _ := json.MarshalIndent(&resp, "", "  ")
+	w.Write(respByte)
+}