Making connection context key exported

loopholelabs · Sep 10, 2024 · 5672a73 · 5672a73
1 parent e4e8578
commit 5672a73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/templates/constants.templ b/templates/constants.templ
@@ -1,5 +1,5 @@
 {{define "constants"}}
-    const connectionContextKey int = 1000
+    const ConnectionContextKey int = 1000
 
     func SetErrorFlag(flags uint8, error bool) uint8 {
         return flags | 0x2

diff --git a/templates/server.templ b/templates/server.templ
@@ -83,7 +83,7 @@ func NewServer({{ GetServerFields .services }}, tlsConfig *tls.Config, logger ty
     {{ end -}}
 
     s.server.ConnContext = func (ctx context.Context, conn *frisbee.Async) context.Context {
-        return context.WithValue(ctx, connectionContextKey, conn)
+        return context.WithValue(ctx, ConnectionContextKey, conn)
     }
 
     return s, nil