Skip to content

Commit

Permalink
Implement process uptime metric (#666)
Browse files Browse the repository at this point in the history
* Implement process uptime metric
* Add clarification comment
* Add another clarification comment
  • Loading branch information
sosedoff committed Mar 30, 2023
1 parent 941c0ac commit 08bbb15
Show file tree
Hide file tree
Showing 5 changed files with 35 additions and 7 deletions.
3 changes: 2 additions & 1 deletion pkg/api/routes.go
Expand Up @@ -61,6 +61,7 @@ func SetupRoutes(router *gin.Engine) {

func SetupMetrics(engine *gin.Engine) {
if command.Opts.MetricsEnabled && command.Opts.MetricsAddr == "" {
engine.GET("/metrics", gin.WrapH(metrics.Handler()))
// NOTE: We're not supporting the MetricsPath CLI option here to avoid the route conflicts.
engine.GET("/metrics", gin.WrapH(metrics.NewHandler()))
}
}
2 changes: 2 additions & 0 deletions pkg/cli/cli.go
Expand Up @@ -307,6 +307,8 @@ func Run() {
}
}

// Start a separate metrics http server. If metrics addr is not provided, we
// add the metrics endpoint in the existing application server (see api.go).
if options.MetricsEnabled && options.MetricsAddr != "" {
go startMetricsServer()
}
Expand Down
25 changes: 25 additions & 0 deletions pkg/metrics/handler.go
@@ -0,0 +1,25 @@
package metrics

import (
"net/http"
"time"

"github.com/prometheus/client_golang/prometheus/promhttp"
)

type Handler struct {
startTime time.Time
promHandler http.Handler
}

func (h Handler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
uptimeGauge.Set(time.Since(h.startTime).Seconds())
h.promHandler.ServeHTTP(rw, req)
}

func NewHandler() http.Handler {
return Handler{
startTime: time.Now(),
promHandler: promhttp.Handler(),
}
}
5 changes: 5 additions & 0 deletions pkg/metrics/metrics.go
Expand Up @@ -20,6 +20,11 @@ var (
Name: "pgweb_healthy",
Help: "Server health status",
})

uptimeGauge = promauto.NewGauge(prometheus.GaugeOpts{
Name: "pgweb_uptime",
Help: "Server application uptime in seconds",
})
)

func IncrementQueriesCount() {
Expand Down
7 changes: 1 addition & 6 deletions pkg/metrics/server.go
Expand Up @@ -3,17 +3,12 @@ package metrics
import (
"net/http"

"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/sirupsen/logrus"
)

func Handler() http.Handler {
return promhttp.Handler()
}

func StartServer(logger *logrus.Logger, path string, addr string) error {
logger.WithField("addr", addr).WithField("path", path).Info("starting prometheus metrics server")

http.Handle(path, Handler())
http.Handle(path, NewHandler())
return http.ListenAndServe(addr, nil)
}

0 comments on commit 08bbb15

Please sign in to comment.