From eb712f0db93448222475c91a7379fe762b031ba8 Mon Sep 17 00:00:00 2001
From: wei liu <wei.liu@zilliz.com>
Date: Tue, 5 Nov 2024 10:46:32 +0800
Subject: [PATCH] fix: dead lock if query node crash during shard client init
 (#37354)

issue: #37115

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
---
 internal/proxy/shard_client.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/internal/proxy/shard_client.go b/internal/proxy/shard_client.go
index dd0b36e494062..8475494e6659c 100644
--- a/internal/proxy/shard_client.go
+++ b/internal/proxy/shard_client.go
@@ -47,6 +47,7 @@ func (n *shardClient) getClient(ctx context.Context) (types.QueryNodeClient, err
 		n.Lock()
 		if !n.initialized.Load() {
 			if err := n.initClients(); err != nil {
+				n.Unlock()
 				return nil, err
 			}
 			n.initialized.Store(true)