62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package site
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
type counter struct {
|
|
hits map[string]int
|
|
mu sync.Mutex
|
|
}
|
|
|
|
type handler struct {
|
|
logger *slog.Logger
|
|
requestsCounter *counter
|
|
}
|
|
|
|
func newHandler(logger *slog.Logger) *handler {
|
|
return &handler{
|
|
logger: logger,
|
|
requestsCounter: &counter{hits: make(map[string]int)},
|
|
}
|
|
}
|
|
|
|
func (h *handler) LoggingMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
h.logger.Debug("Received request", "path", r.URL.Path)
|
|
next.ServeHTTP(w, r)
|
|
h.logger.Debug("Finished Request", "path", r.URL.Path)
|
|
})
|
|
}
|
|
|
|
func (h *handler) RequestsCounterMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
h.requestsCounter.mu.Lock()
|
|
h.requestsCounter.hits[r.URL.Path] += 1
|
|
h.requestsCounter.mu.Unlock()
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func (h *handler) Ping(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprintf(w, "Pong")
|
|
}
|
|
|
|
func (h *handler) Metrics(w http.ResponseWriter, r *http.Request) {
|
|
h.requestsCounter.mu.Lock()
|
|
defer h.requestsCounter.mu.Unlock()
|
|
for path, hits := range h.requestsCounter.hits {
|
|
fmt.Fprintf(w, "http_requests_total{handler=\"%s\"} %d\n", path, hits)
|
|
}
|
|
}
|
|
func (h *handler) Healthz(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprintln(w, "Ok")
|
|
}
|
|
|
|
func (h *handler) NotFound(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, "Page Not Found", http.StatusNotFound)
|
|
}
|