Fix/concurrent write conn (#61)

* fix: barry quick commit, 2025-02-11 10:28:02

* fix: barry quick fix, 2025-03-18 19:20:07

* feat(grpcs): add metadata handlers for gRPC gateway

* feat(grpcs): add context middleware to http server

Author: kooksee

lava/middleware.go

@@ -1,6 +1,13 @@
 package lava
 
-import "context"
+import (
+	"context"
+	"net/http"
+
+	"google.golang.org/grpc/metadata"
+)
+
+type GrpcGatewayMetadata func(ctx context.Context, req *http.Request, rpcPath string, httpPattern string) metadata.MD
 
 type HandlerFunc func(ctx context.Context, req Request) (Response, error)
 

servers/grpcs/server.go

@@ -92,6 +92,7 @@ func (s *serviceImpl) DixInject(
 	log log.Logger,
 	conf *Config,
 	gw []*gateway.Mux,
+	metadataHandlers []lava.GrpcGatewayMetadata,
 ) {
 	s.conf = conf
 	if conf.HttpPort == nil {
@@ -165,6 +166,11 @@ func (s *serviceImpl) DixInject(
 		}))
 	}
 
+	httpServer.Use(func(ctx *fiber.Ctx) error {
+		ctx.SetUserContext(ctx.Context())
+		return ctx.Next()
+	})
+
 	app := fiber.New()
 	app.Group("/debug", httputil.StripPrefix(filepath.Join(conf.BaseUrl, "/debug"), debug.Handler))
 
@@ -239,11 +245,17 @@ func (s *serviceImpl) DixInject(
 			return strings.ToUpper(s), true
 		}),
 		runtime.WithMetadata(func(ctx context.Context, request *http.Request) metadata.MD {
-			path, ok := runtime.HTTPPathPattern(ctx)
-			if !ok {
-				return nil
+			rpcPath, _ := runtime.RPCMethod(ctx)
+			path, _ := runtime.HTTPPathPattern(ctx)
+
+			md := metadata.Pairs("http_path", path, "http_method", request.Method, "http_url", request.URL.Path)
+			for _, h := range metadataHandlers {
+				for k, v := range h(ctx, request, rpcPath, path) {
+					md.Append(k, v...)
+				}
 			}
-			return metadata.Pairs("http_path", path, "http_method", request.Method, "http_url", request.URL.Path)
+
+			return md
 		}),
 		runtime.WithErrorHandler(func(ctx context.Context, mux *runtime.ServeMux, marshal runtime.Marshaler, w http.ResponseWriter, request *http.Request, err error) {
 			md, ok := runtime.ServerMetadataFromContext(ctx)

servers/https/server.go

@@ -19,7 +19,6 @@ import (
 	"github.com/pubgo/funk/stack"
 	"github.com/pubgo/funk/version"
 	"github.com/pubgo/opendoc/opendoc"
-	"github.com/valyala/fasthttp"
 	"google.golang.org/grpc/codes"
 
 	"github.com/pubgo/lava/core/debug"
@@ -51,8 +50,6 @@ func (s *serviceImpl) Run() {
 	defer s.stop()
 	s.start()
 	signal.Wait()
-
-	fasthttp.AcquireArgs()
 }
 
 func (s *serviceImpl) Start() { s.start() }