pass request context to OnRequest

webrpc · Oct 1, 2024 · 512dc61 · 512dc61
1 parent be32331
commit 512dc61
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 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/server.go.tmpl b/server.go.tmpl
@@ -20,7 +20,7 @@ type WebRPCServer interface {
 type {{$serviceName}} struct {
 	{{$typePrefix}}{{$service.Name}}
 	OnError func(r *http.Request, rpcErr *WebRPCError)
-	OnRequest func(w http.ResponseWriter, r *http.Request) error
+	OnRequest func(ctx context.Context, w http.ResponseWriter, r *http.Request) error
 }
 
 func New{{firstLetterToUpper $service.Name}}Server(svc {{$typePrefix}}{{.Name}}) *{{$serviceName}} {
@@ -71,7 +71,7 @@ func (s *{{$serviceName}}) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	switch contentType {
 	case "application/json":
 		if s.OnRequest != nil {
-			if err := s.OnRequest(w, r); err != nil {
+			if err := s.OnRequest(ctx, w, r); err != nil {
 				rpcErr, ok := err.(WebRPCError)
 				if !ok {
 					rpcErr = ErrWebrpcEndpoint.WithCause(err)