Prometheus - track wallet requests

metrics/metrics.go

@@ -0,0 +1,19 @@
+package metrics
+
+import (
+	"github.com/prometheus/client_golang/prometheus"
+)
+
+var (
+	RequestsCount = prometheus.NewCounterVec(
+		prometheus.CounterOpts{
+			Name: "requests_count",
+			Help: "Total number of requests to various endpoints",
+		},
+		[]string{"method"},
+	)
+)
+
+func init() {
+	prometheus.MustRegister(RequestsCount)
+}

server/server.go

@@ -8,6 +8,8 @@ import (
 	"net/http"
 	"net/mail"
 
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+
 	"lbryio/lbry-id/auth"
 	"lbryio/lbry-id/store"
 )
@@ -17,6 +19,8 @@ import (
 const ApiVersion = "3"
 const PathPrefix = "/api/" + ApiVersion
 
+const PathPrometheus = "/metrics"
+
 const PathAuthToken = PathPrefix + "/auth/full"
 const PathRegister = PathPrefix + "/signup"
 const PathPassword = PathPrefix + "/password"
@@ -208,6 +212,8 @@ func (s *Server) Serve() {
 	http.HandleFunc(PathUnknownEndpoint, s.unknownEndpoint)
 	http.HandleFunc(PathWrongApiVersion, s.wrongApiVersion)
 
+	http.Handle(PathPrometheus, promhttp.Handler())
+
 	fmt.Println("Serving at localhost:8090")
 	http.ListenAndServe("localhost:8090", nil)
 }

server/wallet.go

@@ -5,7 +5,10 @@ import (
 	"fmt"
 	"net/http"
 
+	"github.com/prometheus/client_golang/prometheus"
+
 	"lbryio/lbry-id/auth"
+	"lbryio/lbry-id/metrics"
 	"lbryio/lbry-id/store"
 	"lbryio/lbry-id/wallet"
 )
@@ -66,6 +69,8 @@ func getWalletParams(req *http.Request) (token auth.TokenString, err error) {
 }
 
 func (s *Server) getWallet(w http.ResponseWriter, req *http.Request) {
+	metrics.RequestsCount.With(prometheus.Labels{"method": "GET wallet"}).Inc()
+
 	if !getGetData(w, req) {
 		return
 	}
@@ -118,6 +123,8 @@ func (s *Server) getWallet(w http.ResponseWriter, req *http.Request) {
 //     current wallet's sequence
 //   500: Update unsuccessful for unanticipated reasons
 func (s *Server) postWallet(w http.ResponseWriter, req *http.Request) {
+	metrics.RequestsCount.With(prometheus.Labels{"method": "POST wallet"}).Inc()
+
 	var walletRequest WalletRequest
 	if !getPostData(w, req, &walletRequest) {
 		return