gen: add connection to the stream context

Signed-off-by: Luiz Aoqui <luizaoqui@loopholelabs.io>
loopholelabs · Nov 6, 2024 · d85630b · d85630b
1 parent 9934b70
commit d85630b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/templates/server.templ b/templates/server.templ
@@ -82,6 +82,10 @@ func NewServer({{ GetServerFields .services }}, tlsConfig *tls.Config, logger ty
         }
     })
 
+    s.server.StreamContext = func(ctx context.Context, stream *frisbee.Stream) context.Context {
+      return context.WithValue(ctx, ConnectionContextKey, stream.Conn())
+    }
+
     {{ end -}}
 
     s.server.ConnContext = func (ctx context.Context, conn *frisbee.Async) context.Context {