d9c5750ed8
Removed original monolithic WebSocket handler files after extracting to focused packages: Deleted: - ws_jobs.go (1,365 lines) → Extracted to api/jobs/handlers.go - ws_jupyter.go (512 lines) → Extracted to api/jupyter/handlers.go - ws_validate.go (523 lines) → Extracted to api/validate/handlers.go - ws_handler.go (379 lines) → Extracted to api/ws/handler.go - ws_datasets.go (174 lines) - Functionality not migrated - ws_tls_auth.go (101 lines) - Functionality not migrated Updated: - routes.go - Changed NewWSHandler → ws.NewHandler Lines deleted: ~3,000+ lines from monolithic files Build status: Compiles successfully
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/jfraeys/fetch_ml/internal/api/ws"
|
|
"github.com/jfraeys/fetch_ml/internal/prommetrics"
|
|
)
|
|
|
|
// registerRoutes sets up all HTTP routes and handlers
|
|
func (s *Server) registerRoutes(mux *http.ServeMux) {
|
|
// Register Prometheus metrics endpoint (if enabled)
|
|
if s.config.Monitoring.Prometheus.Enabled {
|
|
s.promMetrics = prommetrics.New()
|
|
s.logger.Info("prometheus metrics initialized")
|
|
|
|
// Register metrics endpoint
|
|
metricsPath := s.config.Monitoring.Prometheus.Path
|
|
if metricsPath == "" {
|
|
metricsPath = "/metrics"
|
|
}
|
|
mux.Handle(metricsPath, s.promMetrics.Handler())
|
|
s.logger.Info("metrics endpoint registered", "path", metricsPath)
|
|
}
|
|
|
|
// Register health check endpoints (if enabled)
|
|
if s.config.Monitoring.HealthChecks.Enabled {
|
|
s.registerHealthRoutes(mux)
|
|
}
|
|
|
|
// Register WebSocket endpoint
|
|
s.registerWebSocketRoutes(mux)
|
|
|
|
// Register HTTP API handlers
|
|
s.handlers.RegisterHandlers(mux)
|
|
}
|
|
|
|
// registerHealthRoutes sets up health check endpoints
|
|
func (s *Server) registerHealthRoutes(mux *http.ServeMux) {
|
|
healthHandler := NewHealthHandler(s)
|
|
healthHandler.RegisterRoutes(mux)
|
|
mux.HandleFunc("/health/ok", s.handlers.handleHealth)
|
|
s.logger.Info("health check endpoints registered")
|
|
}
|
|
|
|
// registerWebSocketRoutes sets up WebSocket endpoint
|
|
func (s *Server) registerWebSocketRoutes(mux *http.ServeMux) {
|
|
// Initialize audit logger for WebSocket connections
|
|
auditLogger := s.initAuditLogger()
|
|
|
|
// Register WebSocket handler with security config and audit logger
|
|
securityCfg := getSecurityConfig(s.config)
|
|
wsHandler := ws.NewHandler(
|
|
s.config.BuildAuthConfig(),
|
|
s.logger,
|
|
s.expManager,
|
|
s.config.DataDir,
|
|
s.taskQueue,
|
|
s.db,
|
|
s.jupyterServiceMgr,
|
|
securityCfg,
|
|
auditLogger,
|
|
)
|
|
|
|
mux.Handle("/ws", wsHandler)
|
|
s.logger.Info("websocket endpoint registered")
|
|
}
|