package health
import (
"context"
"encoding/json"
"fmt"
"net/http"
"time"
"github.com/azoic/wormhole-server/pkg/metrics"
"github.com/sirupsen/logrus"
)
type HealthCheckServer struct {
server *http.Server
logger *logrus.Logger
metrics *metrics.Metrics
}
type HealthResponse struct {
Status string `json:"status"`
Timestamp string `json:"timestamp"`
Uptime string `json:"uptime"`
Metrics map[string]interface{} `json:"metrics,omitempty"`
}
func NewHealthCheckServer(addr, port string, logger *logrus.Logger, metrics *metrics.Metrics) *HealthCheckServer {
mux := http.NewServeMux()
hcs := &HealthCheckServer{
logger: logger,
metrics: metrics,
}
mux.HandleFunc("/health", hcs.healthHandler)
mux.HandleFunc("/metrics", hcs.metricsHandler)
mux.HandleFunc("/", hcs.rootHandler)
hcs.server = &http.Server{
Addr: fmt.Sprintf("%s:%s", addr, port),
Handler: mux,
}
return hcs
}
func (hcs *HealthCheckServer) Start() error {
hcs.logger.WithField("address", hcs.server.Addr).Info("Health check server starting")
return hcs.server.ListenAndServe()
}
func (hcs *HealthCheckServer) Stop(ctx context.Context) error {
hcs.logger.Info("Stopping health check server...")
return hcs.server.Shutdown(ctx)
}
func (hcs *HealthCheckServer) healthHandler(w http.ResponseWriter, r *http.Request) {
response := HealthResponse{
Status: "healthy",
Timestamp: time.Now().Format(time.RFC3339),
Uptime: time.Since(hcs.metrics.StartTime).String(),
}
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(response)
}
func (hcs *HealthCheckServer) metricsHandler(w http.ResponseWriter, r *http.Request) {
response := HealthResponse{
Status: "healthy",
Timestamp: time.Now().Format(time.RFC3339),
Uptime: time.Since(hcs.metrics.StartTime).String(),
Metrics: hcs.metrics.GetStats(),
}
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(response)
}
func (hcs *HealthCheckServer) rootHandler(w http.ResponseWriter, r *http.Request) {
html := `
Wormhole SOCKS5 Proxy
Wormhole SOCKS5 Proxy
Status: Running
Endpoints:
`
w.Header().Set("Content-Type", "text/html")
w.WriteHeader(http.StatusOK)
w.Write([]byte(html))
}