apps/monibuca
1
0
Fork 0
mirror of https://github.com/langhuihui/monibuca.git synced 2025-10-06 00:37:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add cors to grpc gateway

Browse Source
This commit is contained in:
langhuihui
2024-10-09 17:20:17 +08:00
parent cb7b013897
commit b39cfebca7
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server.go
View File

@@ -14,6 +14,7 @@ import (
"time" "time"
"github.com/shirou/gopsutil/v3/cpu" "github.com/shirou/gopsutil/v3/cpu"
"google.golang.org/protobuf/proto"
"m7s.live/m7s/v5/pkg/task" "m7s.live/m7s/v5/pkg/task"
@@ -179,7 +180,15 @@ func (s *Server) Start() (err error) {
s.LogHandler.SetLevel(slog.LevelDebug) s.LogHandler.SetLevel(slog.LevelDebug)
s.LogHandler.Add(defaultLogHandler) s.LogHandler.Add(defaultLogHandler)
s.Logger = slog.New(&s.LogHandler).With("server", s.ID) s.Logger = slog.New(&s.LogHandler).With("server", s.ID)
mux := runtime.NewServeMux(runtime.WithMarshalerOption("text/plain", &pb.TextPlain{}), runtime.WithRoutingErrorHandler(func(_ context.Context, _ *runtime.ServeMux, _ runtime.Marshaler, w http.ResponseWriter, r *http.Request, _ int) { mux := runtime.NewServeMux(runtime.WithMarshalerOption("text/plain", &pb.TextPlain{}), runtime.WithForwardResponseOption(func(ctx context.Context, w http.ResponseWriter, m proto.Message) error {
header := w.Header()
header.Set("Access-Control-Allow-Credentials", "true")
header.Set("Cross-Origin-Resource-Policy", "cross-origin")
header.Set("Access-Control-Allow-Headers", "Content-Type,Access-Token")
header.Set("Access-Control-Allow-Private-Network", "true")
header.Set("Access-Control-Allow-Origin", "*")
return nil
}), runtime.WithRoutingErrorHandler(func(_ context.Context, _ *runtime.ServeMux, _ runtime.Marshaler, w http.ResponseWriter, r *http.Request, _ int) {
httpConf.GetHttpMux().ServeHTTP(w, r) httpConf.GetHttpMux().ServeHTTP(w, r)
})) }))
httpConf.SetMux(mux) httpConf.SetMux(mux)