minor changes to validateWRP func

xmidt-org · Oct 11, 2023 · ecb3d78 · ecb3d78
1 parent 603e230
commit ecb3d78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/primaryHandler.go b/primaryHandler.go
@@ -567,8 +567,7 @@ func ValidateWRP() func(http.Handler) http.Handler {
 	return func(delegate http.Handler) http.Handler {
 		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 
-			ctx := r.Context()
-			if msg, ok := wrpcontext.Get[*wrp.Message](ctx); ok {
+			if msg, ok := wrpcontext.Get[*wrp.Message](r.Context()); ok {
 				validators := wrp.SpecValidators()
 				var err error
 				for _, v := range validators {
@@ -586,6 +585,7 @@ func ValidateWRP() func(http.Handler) http.Handler {
 					return
 				}
 			}
+
 			delegate.ServeHTTP(w, r)
 		})
 	}