mirror of
https://github.com/datarhei/core.git
synced 2025-10-05 16:07:07 +08:00
Add metrics collector for HTTP status codes
This commit is contained in:
@@ -1558,6 +1558,9 @@ func (a *api) start(ctx context.Context) error {
|
|||||||
a.sidecarserver.Handler = autocertManager.HTTPChallengeHandler(sidecarserverhandler)
|
a.sidecarserver.Handler = autocertManager.HTTPChallengeHandler(sidecarserverhandler)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
metrics.Register(monitor.NewHTTPCollector("HTTPS", mainserverhandler))
|
||||||
|
metrics.Register(monitor.NewHTTPCollector("HTTP", sidecarserverhandler))
|
||||||
|
|
||||||
wgStart.Add(1)
|
wgStart.Add(1)
|
||||||
a.wgStop.Add(1)
|
a.wgStop.Add(1)
|
||||||
|
|
||||||
@@ -1583,6 +1586,8 @@ func (a *api) start(ctx context.Context) error {
|
|||||||
|
|
||||||
sendError(err)
|
sendError(err)
|
||||||
}()
|
}()
|
||||||
|
} else {
|
||||||
|
metrics.Register(monitor.NewHTTPCollector("HTTP", mainserverhandler))
|
||||||
}
|
}
|
||||||
|
|
||||||
if a.rtmpserver != nil {
|
if a.rtmpserver != nil {
|
||||||
|
@@ -16,11 +16,13 @@ type Config struct {
|
|||||||
// Skipper defines a function to skip middleware.
|
// Skipper defines a function to skip middleware.
|
||||||
Skipper middleware.Skipper
|
Skipper middleware.Skipper
|
||||||
Logger log.Logger
|
Logger log.Logger
|
||||||
|
Status func(code int)
|
||||||
}
|
}
|
||||||
|
|
||||||
var DefaultConfig = Config{
|
var DefaultConfig = Config{
|
||||||
Skipper: middleware.DefaultSkipper,
|
Skipper: middleware.DefaultSkipper,
|
||||||
Logger: log.New("HTTP"),
|
Logger: log.New(""),
|
||||||
|
Status: nil,
|
||||||
}
|
}
|
||||||
|
|
||||||
func New() echo.MiddlewareFunc {
|
func New() echo.MiddlewareFunc {
|
||||||
@@ -76,6 +78,10 @@ func NewWithConfig(config Config) echo.MiddlewareFunc {
|
|||||||
|
|
||||||
latency := time.Since(start)
|
latency := time.Since(start)
|
||||||
|
|
||||||
|
if config.Status != nil {
|
||||||
|
config.Status(res.Status)
|
||||||
|
}
|
||||||
|
|
||||||
if raw != "" {
|
if raw != "" {
|
||||||
path = path + "?" + raw
|
path = path + "?" + raw
|
||||||
}
|
}
|
||||||
|
@@ -32,6 +32,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"github.com/datarhei/core/v16/cluster"
|
"github.com/datarhei/core/v16/cluster"
|
||||||
cfgstore "github.com/datarhei/core/v16/config/store"
|
cfgstore "github.com/datarhei/core/v16/config/store"
|
||||||
@@ -43,6 +44,7 @@ import (
|
|||||||
api "github.com/datarhei/core/v16/http/handler/api"
|
api "github.com/datarhei/core/v16/http/handler/api"
|
||||||
httplog "github.com/datarhei/core/v16/http/log"
|
httplog "github.com/datarhei/core/v16/http/log"
|
||||||
"github.com/datarhei/core/v16/http/router"
|
"github.com/datarhei/core/v16/http/router"
|
||||||
|
serverhandler "github.com/datarhei/core/v16/http/server"
|
||||||
"github.com/datarhei/core/v16/http/validator"
|
"github.com/datarhei/core/v16/http/validator"
|
||||||
"github.com/datarhei/core/v16/iam"
|
"github.com/datarhei/core/v16/iam"
|
||||||
"github.com/datarhei/core/v16/log"
|
"github.com/datarhei/core/v16/log"
|
||||||
@@ -104,10 +106,6 @@ type CorsConfig struct {
|
|||||||
Origins []string
|
Origins []string
|
||||||
}
|
}
|
||||||
|
|
||||||
type Server interface {
|
|
||||||
ServeHTTP(w http.ResponseWriter, r *http.Request)
|
|
||||||
}
|
|
||||||
|
|
||||||
type server struct {
|
type server struct {
|
||||||
logger log.Logger
|
logger log.Logger
|
||||||
|
|
||||||
@@ -154,6 +152,11 @@ type server struct {
|
|||||||
profiling bool
|
profiling bool
|
||||||
|
|
||||||
readOnly bool
|
readOnly bool
|
||||||
|
|
||||||
|
metrics struct {
|
||||||
|
lock sync.Mutex
|
||||||
|
status map[int]uint64
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type filesystem struct {
|
type filesystem struct {
|
||||||
@@ -164,7 +167,7 @@ type filesystem struct {
|
|||||||
middleware echo.MiddlewareFunc
|
middleware echo.MiddlewareFunc
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewServer(config Config) (Server, error) {
|
func NewServer(config Config) (serverhandler.Server, error) {
|
||||||
s := &server{
|
s := &server{
|
||||||
logger: config.Logger,
|
logger: config.Logger,
|
||||||
mimeTypesFile: config.MimeTypesFile,
|
mimeTypesFile: config.MimeTypesFile,
|
||||||
@@ -172,6 +175,8 @@ func NewServer(config Config) (Server, error) {
|
|||||||
readOnly: config.ReadOnly,
|
readOnly: config.ReadOnly,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
s.metrics.status = map[int]uint64{}
|
||||||
|
|
||||||
s.filesystems = map[string]*filesystem{}
|
s.filesystems = map[string]*filesystem{}
|
||||||
|
|
||||||
corsPrefixes := map[string][]string{
|
corsPrefixes := map[string][]string{
|
||||||
@@ -327,6 +332,12 @@ func NewServer(config Config) (Server, error) {
|
|||||||
|
|
||||||
s.middleware.log = mwlog.NewWithConfig(mwlog.Config{
|
s.middleware.log = mwlog.NewWithConfig(mwlog.Config{
|
||||||
Logger: s.logger,
|
Logger: s.logger,
|
||||||
|
Status: func(code int) {
|
||||||
|
s.metrics.lock.Lock()
|
||||||
|
defer s.metrics.lock.Unlock()
|
||||||
|
|
||||||
|
s.metrics.status[code]++
|
||||||
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
s.v3handler.widget = api.NewWidget(api.WidgetConfig{
|
s.v3handler.widget = api.NewWidget(api.WidgetConfig{
|
||||||
@@ -458,6 +469,19 @@ func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|||||||
s.router.ServeHTTP(w, r)
|
s.router.ServeHTTP(w, r)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *server) HTTPStatus() map[int]uint64 {
|
||||||
|
status := map[int]uint64{}
|
||||||
|
|
||||||
|
s.metrics.lock.Lock()
|
||||||
|
defer s.metrics.lock.Unlock()
|
||||||
|
|
||||||
|
for code, value := range s.metrics.status {
|
||||||
|
status[code] = value
|
||||||
|
}
|
||||||
|
|
||||||
|
return status
|
||||||
|
}
|
||||||
|
|
||||||
func (s *server) setRoutes() {
|
func (s *server) setRoutes() {
|
||||||
gzipMiddleware := mwgzip.NewWithConfig(mwgzip.Config{
|
gzipMiddleware := mwgzip.NewWithConfig(mwgzip.Config{
|
||||||
Level: mwgzip.BestSpeed,
|
Level: mwgzip.BestSpeed,
|
||||||
|
8
http/server/server.go
Normal file
8
http/server/server.go
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import "net/http"
|
||||||
|
|
||||||
|
type Server interface {
|
||||||
|
ServeHTTP(w http.ResponseWriter, r *http.Request)
|
||||||
|
HTTPStatus() map[int]uint64
|
||||||
|
}
|
49
monitor/http.go
Normal file
49
monitor/http.go
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
package monitor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/datarhei/core/v16/http/server"
|
||||||
|
"github.com/datarhei/core/v16/monitor/metric"
|
||||||
|
)
|
||||||
|
|
||||||
|
type httpCollector struct {
|
||||||
|
handler server.Server
|
||||||
|
name string
|
||||||
|
statusDescr *metric.Description
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewHTTPCollector(name string, handler server.Server) metric.Collector {
|
||||||
|
c := &httpCollector{
|
||||||
|
handler: handler,
|
||||||
|
name: name,
|
||||||
|
}
|
||||||
|
|
||||||
|
c.statusDescr = metric.NewDesc("http_status", "Total return status", []string{"name", "code"})
|
||||||
|
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *httpCollector) Prefix() string {
|
||||||
|
return "filesystem"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *httpCollector) Describe() []*metric.Description {
|
||||||
|
return []*metric.Description{
|
||||||
|
c.statusDescr,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *httpCollector) Collect() metric.Metrics {
|
||||||
|
status := c.handler.HTTPStatus()
|
||||||
|
|
||||||
|
metrics := metric.NewMetrics()
|
||||||
|
|
||||||
|
for code, count := range status {
|
||||||
|
metrics.Add(metric.NewValue(c.statusDescr, float64(count), c.name, strconv.Itoa(code)))
|
||||||
|
}
|
||||||
|
|
||||||
|
return metrics
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *httpCollector) Stop() {}
|
Reference in New Issue
Block a user