Move metrics from macaron to chi (#13601)

This commit is contained in:
Lunny Xiao 2020-11-18 04:50:06 +08:00 committed by GitHub
parent 75ebf7c5bd
commit 9ec5e6c40b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 22 additions and 22 deletions

View File

@ -6,29 +6,29 @@ package routers
import ( import (
"crypto/subtle" "crypto/subtle"
"net/http"
"code.gitea.io/gitea/modules/context"
"code.gitea.io/gitea/modules/setting" "code.gitea.io/gitea/modules/setting"
"github.com/prometheus/client_golang/prometheus/promhttp" "github.com/prometheus/client_golang/prometheus/promhttp"
) )
// Metrics validate auth token and render prometheus metrics // Metrics validate auth token and render prometheus metrics
func Metrics(ctx *context.Context) { func Metrics(resp http.ResponseWriter, req *http.Request) {
if setting.Metrics.Token == "" { if setting.Metrics.Token == "" {
promhttp.Handler().ServeHTTP(ctx.Resp, ctx.Req.Request) promhttp.Handler().ServeHTTP(resp, req)
return return
} }
header := ctx.Req.Header.Get("Authorization") header := req.Header.Get("Authorization")
if header == "" { if header == "" {
ctx.Error(401) http.Error(resp, "", 401)
return return
} }
got := []byte(header) got := []byte(header)
want := []byte("Bearer " + setting.Metrics.Token) want := []byte("Bearer " + setting.Metrics.Token)
if subtle.ConstantTimeCompare(got, want) != 1 { if subtle.ConstantTimeCompare(got, want) != 1 {
ctx.Error(401) http.Error(resp, "", 401)
return return
} }
promhttp.Handler().ServeHTTP(ctx.Resp, ctx.Req.Request) promhttp.Handler().ServeHTTP(resp, req)
} }

View File

@ -17,12 +17,15 @@ import (
"time" "time"
"code.gitea.io/gitea/modules/log" "code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/metrics"
"code.gitea.io/gitea/modules/public" "code.gitea.io/gitea/modules/public"
"code.gitea.io/gitea/modules/setting" "code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/storage" "code.gitea.io/gitea/modules/storage"
"code.gitea.io/gitea/routers"
"github.com/go-chi/chi" "github.com/go-chi/chi"
"github.com/go-chi/chi/middleware" "github.com/go-chi/chi/middleware"
"github.com/prometheus/client_golang/prometheus"
) )
type routerLoggerOptions struct { type routerLoggerOptions struct {
@ -251,6 +254,18 @@ func NormalRoutes() http.Handler {
}) })
} }
r.Get("/apple-touch-icon.png", func(w http.ResponseWriter, req *http.Request) {
http.Redirect(w, req, path.Join(setting.StaticURLPrefix, "img/apple-touch-icon.png"), 301)
})
// prometheus metrics endpoint
if setting.Metrics.Enabled {
c := metrics.NewCollector()
prometheus.MustRegister(c)
r.Get("/metrics", routers.Metrics)
}
return r return r
} }

View File

@ -6,14 +6,12 @@ package routes
import ( import (
"encoding/gob" "encoding/gob"
"path"
"code.gitea.io/gitea/models" "code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/auth" "code.gitea.io/gitea/modules/auth"
"code.gitea.io/gitea/modules/context" "code.gitea.io/gitea/modules/context"
"code.gitea.io/gitea/modules/lfs" "code.gitea.io/gitea/modules/lfs"
"code.gitea.io/gitea/modules/log" "code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/metrics"
"code.gitea.io/gitea/modules/options" "code.gitea.io/gitea/modules/options"
"code.gitea.io/gitea/modules/setting" "code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/templates" "code.gitea.io/gitea/modules/templates"
@ -43,7 +41,6 @@ import (
"gitea.com/macaron/macaron" "gitea.com/macaron/macaron"
"gitea.com/macaron/session" "gitea.com/macaron/session"
"gitea.com/macaron/toolbox" "gitea.com/macaron/toolbox"
"github.com/prometheus/client_golang/prometheus"
"github.com/tstranex/u2f" "github.com/tstranex/u2f"
) )
@ -978,23 +975,11 @@ func RegisterMacaronRoutes(m *macaron.Macaron) {
private.RegisterRoutes(m) private.RegisterRoutes(m)
}) })
m.Get("/apple-touch-icon.png", func(ctx *context.Context) {
ctx.Redirect(path.Join(setting.StaticURLPrefix, "img/apple-touch-icon.png"), 301)
})
// Progressive Web App // Progressive Web App
m.Get("/manifest.json", templates.JSONRenderer(), func(ctx *context.Context) { m.Get("/manifest.json", templates.JSONRenderer(), func(ctx *context.Context) {
ctx.HTML(200, "pwa/manifest_json") ctx.HTML(200, "pwa/manifest_json")
}) })
// prometheus metrics endpoint
if setting.Metrics.Enabled {
c := metrics.NewCollector()
prometheus.MustRegister(c)
m.Get("/metrics", routers.Metrics)
}
// Not found handler. // Not found handler.
m.NotFound(routers.NotFound) m.NotFound(routers.NotFound)
} }