kyle
2607d56e4c
Go service with /healthz, SSRF guard (parse + resolve-all + Control), exhaustive tests, NixOS module, and Makefile as source of truth for build flags. nix flake check runs make test lint.
42 lines
976 B
Go
42 lines
976 B
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
// Version is set from main via New.
|
|
type Server struct {
|
|
Version string
|
|
Mux *http.ServeMux
|
|
}
|
|
|
|
func New(version string) *Server {
|
|
s := &Server{Version: version, Mux: http.NewServeMux()}
|
|
s.Mux.HandleFunc("GET /healthz", s.healthz)
|
|
s.Mux.HandleFunc("GET /healthz/", s.healthz)
|
|
return s
|
|
}
|
|
|
|
func (s *Server) healthz(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(http.StatusOK)
|
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
|
"status": "ok",
|
|
"version": s.Version,
|
|
})
|
|
}
|
|
|
|
// HTTPServer wraps the mux with timeouts. Listen address is the caller's.
|
|
func (s *Server) HTTPServer(addr string) *http.Server {
|
|
return &http.Server{
|
|
Addr: addr,
|
|
Handler: s.Mux,
|
|
ReadHeaderTimeout: 5 * time.Second,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
IdleTimeout: 30 * time.Second,
|
|
}
|
|
}
|