lilou-1294 fix the issue that the sticky client is closed

smallnest · Apr 2, 2024 · b0e1cff · b0e1cff
1 parent aec5225
commit b0e1cff
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/client/xclient.go b/client/xclient.go
@@ -245,6 +245,10 @@ func filterByStateAndGroup(group string, servers map[string]string) {
 // selects a client from candidates base on c.selectMode
 func (c *xClient) selectClient(ctx context.Context, servicePath, serviceMethod string, args interface{}) (string, RPCClient, error) {
 	c.mu.Lock()
+	if c.option.Sticky && c.stickyRPCClient.IsClosing() || c.stickyRPCClient.IsShutdown() {
+		c.stickyRPCClient = nil
+	}
+
 	if c.option.Sticky && c.stickyRPCClient != nil {
 		c.mu.Unlock()
 		return c.stickyK, c.stickyRPCClient, nil