support retry times in session pool

vesoft-inc · Sep 11, 2024 · f533579 · f533579
1 parent 530bb43
commit f533579
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/configs.go b/configs.go
@@ -224,6 +224,14 @@ func WithHandshakeKey(handshakeKey string) SessionPoolConfOption {
 		conf.handshakeKey = handshakeKey
 	}
 }
+func WithRetryTimes(retryTimes int) SessionPoolConfOption {
+	if retryTimes < 0 {
+		retryTimes = 0
+	}
+	return func(conf *SessionPoolConf) {
+		conf.retryGetSessionTimes = retryTimes
+	}
+}
 
 func (conf *SessionPoolConf) checkMandatoryFields() error {
 	// Check mandatory fields