add webrpc ctx containing multiple ctx values

webrpc · Sep 27, 2024 · 20a35c5 · 20a35c5
1 parent 1af2de2
commit 20a35c5
Show file tree

Hide file tree

Showing 4 changed files with 66 additions and 3 deletions.
diff --git a/_examples/golang-basics/example.gen.go b/_examples/golang-basics/example.gen.go
diff --git a/_examples/golang-imports/api.gen.go b/_examples/golang-imports/api.gen.go
diff --git a/helpers.go.tmpl b/helpers.go.tmpl
@@ -13,6 +13,13 @@ func (k *contextKey) String() string {
 	return "webrpc context value " + k.name
 }
 
+type WebRrpCtx struct {
+    ServiceName string
+    Request *http.Request
+    Annotations map[string]string
+    Writer http.ResponseWriter
+}
+
 var (
 {{- if $opts.client}}
 	HTTPClientRequestHeadersCtxKey = &contextKey{"HTTPClientRequestHeaders"}
@@ -28,6 +35,8 @@ var (
 	MethodNameCtxKey = &contextKey{"MethodName"}
 
 	MethodAnnotationsCtxKey = &contextKey{"MethodAnnotations"}
+
+	WebRpcCtxKey = &contextKey{"WebRPC"}
 )
 
 func ServiceNameFromContext(ctx context.Context) string {
@@ -50,7 +59,12 @@ func MethodAnnotationsFromContext(ctx context.Context) map[string]string {
 	return annotations
 }
 
-{{- if $opts.server}}
+func WebRPCContext(ctx context.Context) map[string]string {
+	annotations, _ := ctx.Value(MethodAnnotationsCtxKey).(map[string]string)
+	return annotations
+}
+
+{{if $opts.server}}
 func ResponseWriterFromContext(ctx context.Context) http.ResponseWriter {
 	w, _ := ctx.Value(HTTPResponseWriterCtxKey).(http.ResponseWriter)
 	return w

diff --git a/server.go.tmpl b/server.go.tmpl
@@ -43,6 +43,12 @@ func (s *{{$serviceName}}) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	ctx = context.WithValue(ctx, HTTPRequestCtxKey, r)
 	ctx = context.WithValue(ctx, ServiceNameCtxKey, "{{.Name}}")
 	ctx = context.WithValue(ctx, MethodAnnotationsCtxKey, methodAnnotations[r.URL.Path])
+    ctx = context.WithValue(ctx, WebRpcCtxKey, &WebRrpCtx{
+        Annotations:  methodAnnotations[r.URL.Path],
+        Request: r,
+        ServiceName: "{{.Name}}",
+        Writer: w,
+    })
 
     if s.OnRequest != nil {
         s.OnRequest(w, r)