mirror of
https://github.com/Sagit-chu/flvx.git
synced 2026-09-28 07:36:38 +08:00
sagitchu
21 lines
534 B
Go
21 lines
534 B
Go
package httpserver
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"go-backend/internal/http/handler"
|
|
"go-backend/internal/http/middleware"
|
|
)
|
|
|
|
func NewRouter(h *handler.Handler, jwtSecret string) http.Handler {
|
|
mux := http.NewServeMux()
|
|
h.Register(mux)
|
|
mux.Handle("/system-info", h.WebSocketHandler())
|
|
|
|
wrapped := middleware.Recover(mux)
|
|
wrapped = middleware.JWT(middleware.AuthOptions{JWTSecret: jwtSecret, GetUserAuthState: h.GetUserAuthState})(wrapped)
|
|
wrapped = middleware.RequestLog(wrapped)
|
|
wrapped = middleware.CORS(wrapped)
|
|
return wrapped
|
|
}
|