mirror of
https://github.com/Sagit-chu/flvx.git
synced 2026-09-28 07:36:38 +08:00
20 lines
454 B
Go
20 lines
454 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})(wrapped)
|
|
wrapped = middleware.CORS(wrapped)
|
|
return wrapped
|
|
}
|