mirror of
https://github.com/42wim/matterbridge.git
synced 2024-12-04 16:32:02 -08:00
83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
"strconv"
|
||
|
"sync"
|
||
|
"time"
|
||
|
|
||
|
"github.com/labstack/echo"
|
||
|
)
|
||
|
|
||
|
type (
|
||
|
Stats struct {
|
||
|
Uptime time.Time `json:"uptime"`
|
||
|
RequestCount uint64 `json:"requestCount"`
|
||
|
Statuses map[string]int `json:"statuses"`
|
||
|
mutex sync.RWMutex
|
||
|
}
|
||
|
)
|
||
|
|
||
|
func NewStats() *Stats {
|
||
|
return &Stats{
|
||
|
Uptime: time.Now(),
|
||
|
Statuses: make(map[string]int),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Process is the middleware function.
|
||
|
func (s *Stats) Process(next echo.HandlerFunc) echo.HandlerFunc {
|
||
|
return func(c echo.Context) error {
|
||
|
if err := next(c); err != nil {
|
||
|
c.Error(err)
|
||
|
}
|
||
|
s.mutex.Lock()
|
||
|
defer s.mutex.Unlock()
|
||
|
s.RequestCount++
|
||
|
status := strconv.Itoa(c.Response().Status)
|
||
|
s.Statuses[status]++
|
||
|
return nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Handle is the endpoint to get stats.
|
||
|
func (s *Stats) Handle(c echo.Context) error {
|
||
|
s.mutex.RLock()
|
||
|
defer s.mutex.RUnlock()
|
||
|
return c.JSON(http.StatusOK, s)
|
||
|
}
|
||
|
|
||
|
// ServerHeader middleware adds a `Server` header to the response.
|
||
|
func ServerHeader(next echo.HandlerFunc) echo.HandlerFunc {
|
||
|
return func(c echo.Context) error {
|
||
|
c.Response().Header().Set(echo.HeaderServer, "Echo/3.0")
|
||
|
return next(c)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
e := echo.New()
|
||
|
|
||
|
// Debug mode
|
||
|
e.Debug = true
|
||
|
|
||
|
//-------------------
|
||
|
// Custom middleware
|
||
|
//-------------------
|
||
|
// Stats
|
||
|
s := NewStats()
|
||
|
e.Use(s.Process)
|
||
|
e.GET("/stats", s.Handle) // Endpoint to get stats
|
||
|
|
||
|
// Server header
|
||
|
e.Use(ServerHeader)
|
||
|
|
||
|
// Handler
|
||
|
e.GET("/", func(c echo.Context) error {
|
||
|
return c.String(http.StatusOK, "Hello, World!")
|
||
|
})
|
||
|
|
||
|
// Start server
|
||
|
e.Logger.Fatal(e.Start(":1323"))
|
||
|
}
|