feat: add API to DisableGopool (cloudwego#28)

connection_onevent.go

@@ -22,6 +22,15 @@ import (
 	"github.com/bytedance/gopkg/util/gopool"
 )
 
+var runTask = gopool.CtxGo
+
+func disableGopool() error {
+	runTask = func(ctx context.Context, f func()) {
+		go f()
+	}
+	return nil
+}
+
 // ------------------------------------ implement OnPrepare, OnRequest, CloseCallback ------------------------------------
 
 type gracefulExit interface {
@@ -116,7 +125,7 @@ func (c *connection) onRequest() (err error) {
 			goto START
 		}
 	}
-	gopool.CtxGo(c.ctx, task)
+	runTask(c.ctx, task)
 	return nil
 }
 

netpoll_options.go

@@ -35,6 +35,15 @@ func SetLoadBalance(lb LoadBalance) error {
 	return setLoadBalance(lb)
 }
 
+// DisableGopool will remove gopool(the goroutine pool used to run OnRequest),
+// which means that OnRequest will be run via `go OnRequest(...)`.
+// Usually, OnRequest will cause stack expansion, which can be solved by reusing goroutine.
+// But if you can confirm that the OnRequest will not cause stack expansion,
+// it is recommended to use DisableGopool to reduce redundancy and improve performance.
+func DisableGopool() error {
+	return disableGopool()
+}
+
 // WithOnPrepare registers the OnPrepare method to EventLoop.
 func WithOnPrepare(onPrepare OnPrepare) Option {
 	return Option{func(op *options) {