feat: add Initialize to init pollers actively

cloudwego · Feb 20, 2024 · dba7064 · dba7064
1 parent 06e5be6
commit dba7064
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/netpoll_options.go b/netpoll_options.go
@@ -45,6 +45,12 @@ func SetLoadBalance(lb LoadBalance) error {
 	return setLoadBalance(lb)
 }
 
+// Initialize the pollers actively. By default, it's lazy initialized.
+// It's safe to call it multi times.
+func Initialize() {
+	initialize()
+}
+
 func SetLoggerOutput(w io.Writer) {
 	setLoggerOutput(w)
 }

diff --git a/poll_manager.go b/poll_manager.go
@@ -34,6 +34,11 @@ func setLoadBalance(lb LoadBalance) error {
 	return pollmanager.SetLoadBalance(lb)
 }
 
+func initialize() {
+	// The first call of Pick() will init pollers
+	_ = pollmanager.Pick()
+}
+
 func setLoggerOutput(w io.Writer) {
 	logger = log.New(w, "", log.LstdFlags)
 }