trafficlunar/api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: websockets not working with prometheus middleware

Browse source
This commit is contained in:
trafficlunar 2025-07-20 15:30:48 +01:00
parent f4e24d9440
commit b45149d731
1 changed files with 19 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
internal/server/router.go
View file

@ -31,7 +31,6 @@ func NewRouter() {
r := chi.NewRouter()
// Middleware
r.Use(app_middleware.PrometheusMiddleware)
r.Use(middleware.RequestID)
r.Use(middleware.RealIP)
r.Use(middleware.Logger)
@ -46,6 +45,10 @@ func NewRouter() {
MaxAge: 300,
}))
// Add Prometheus middleware to all routes except WebSockets
r.Group(func(r chi.Router) {
r.Use(app_middleware.PrometheusMiddleware)
// Prometheus
r.Handle("/metrics", promhttp.Handler())
@ -62,8 +65,10 @@ func NewRouter() {
r.Get("/song", handler.HandleGetCurrentlyPlaying)
r.Get("/projects", handler.HandleGetProjects)
r.Get("/computer", handler.HandleComputerGraphData)
r.Get("/computer/ws", handler.HandleComputerWebSocket)
r.Get("/roblox/grow-a-robloxian/likes", handler.HandleGetGrowARobloxianLikesCount)
})
r.Get("/computer/ws", handler.HandleComputerWebSocket)
port := os.Getenv("PORT")
if len(port) == 0 {