feat: add method to exile requestContext (#1101)

cloudwego · May 9, 2024 · 2d13882 · 2d13882
1 parent b5b3f9d
commit 2d13882
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/pkg/app/context.go b/pkg/app/context.go
@@ -236,6 +236,17 @@ type RequestContext struct {
 
 	binder    binding.Binder
 	validator binding.StructValidator
+	exiled    bool
+}
+
+// Exile marks this RequestContext as not to be recycled.
+// Experimental features: Use with caution, it may have a slight impact on performance.
+func (ctx *RequestContext) Exile() {
+	ctx.exiled = true
+}
+
+func (ctx *RequestContext) IsExiled() bool {
+	return ctx.exiled
 }
 
 // Flush is the shortcut for ctx.Response.GetHijackWriter().Flush().

diff --git a/pkg/protocol/http1/server.go b/pkg/protocol/http1/server.go
@@ -133,6 +133,11 @@ func (s Server) Serve(c context.Context, conn network.Conn) (err error) {
 			zr.Release() //nolint:errcheck
 			zr = nil
 		}
+
+		if ctx.IsExiled() {
+			return
+		}
+
 		ctx.Reset()
 		s.Core.GetCtxPool().Put(ctx)
 	}()