feat: add cluster returnPeerClient() func

stream1080 · Oct 1, 2023 · 3a2751c · 3a2751c
1 parent 3d868e0
commit 3a2751c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cluster/com.go b/cluster/com.go
@@ -24,3 +24,12 @@ func (cluster *ClusterDatabases) getPeerClient(peer string) (*client.Client, err
 
 	return c, nil
 }
+
+func (cluster *ClusterDatabases) returnPeerClient(peer string, client *client.Client) error {
+	pool, ok := cluster.peerConn[peer]
+	if !ok {
+		return errors.New("conn not found")
+	}
+
+	return pool.ReturnObject(context.Background(), client)
+}