From 6e5511614df73e2e5f1bf8ab0dd7232fb7bbe7ce Mon Sep 17 00:00:00 2001
From: YuQing <384681@qq.com>
Date: Thu, 20 Feb 2020 16:03:28 +0800
Subject: [PATCH] add function fc_server_check_connect_ex

---
 HISTORY              |  2 +-
 src/server_id_func.c | 56 ++++++++++++++++++++++++++++++++++++++++++++
 src/server_id_func.h | 11 +++++++++
 3 files changed, 68 insertions(+), 1 deletion(-)

diff --git a/HISTORY b/HISTORY
index 1612543..f2d0693 100644
--- a/HISTORY
+++ b/HISTORY
@@ -1,5 +1,5 @@
 
-Version 1.44  2020-02-19
+Version 1.44  2020-02-20
   * add test file src/tests/test_pthread_lock.c
   * add uniq_skiplist.[hc]
   * add function split_string_ex
diff --git a/src/server_id_func.c b/src/server_id_func.c
index 2fffca1..957ceac 100644
--- a/src/server_id_func.c
+++ b/src/server_id_func.c
@@ -1451,3 +1451,59 @@ void fc_server_to_log(FCServerConfig *ctx)
     fc_server_log_groups(ctx);
     fc_server_log_servers(ctx);
 }
+
+ConnectionInfo *fc_server_check_connect_ex(FCAddressPtrArray *addr_array,
+        const int connect_timeout, const char *bind_ipaddr,
+        const bool log_connect_error, int *err_no)
+{
+    FCAddressInfo **current;
+    FCAddressInfo **addr;
+    FCAddressInfo **end;
+
+    if (addr_array->count <= 0) {
+        *err_no = ENOENT;
+        return NULL;
+    }
+
+    current = addr_array->addrs + addr_array->index;
+    if ((*current)->conn.sock >= 0) {
+        return &(*current)->conn;
+    }
+
+    if ((*err_no= conn_pool_connect_server_ex(&(*current)->conn,
+                    connect_timeout, bind_ipaddr, log_connect_error)) == 0)
+    {
+        return &(*current)->conn;
+    }
+
+    if (addr_array->count == 1) {
+        return NULL;
+    }
+
+    end = addr_array->addrs + addr_array->count;
+    for (addr=addr_array->addrs; addr<end; addr++) {
+        if (addr == current) {
+            continue;
+        }
+        if ((*err_no= conn_pool_connect_server_ex(&(*addr)->conn,
+                        connect_timeout, bind_ipaddr,
+                        log_connect_error)) == 0)
+        {
+            addr_array->index = addr - addr_array->addrs;
+            return &(*addr)->conn;
+        }
+    }
+
+    return NULL;
+}
+
+void fc_server_disconnect(FCAddressPtrArray *addr_array)
+{
+    FCAddressInfo **current;
+
+    current = addr_array->addrs + addr_array->index;
+    if ((*current)->conn.sock >= 0) {
+        close((*current)->conn.sock);
+        (*current)->conn.sock = -1;
+    }
+}
diff --git a/src/server_id_func.h b/src/server_id_func.h
index cc540e7..3a7bc10 100644
--- a/src/server_id_func.h
+++ b/src/server_id_func.h
@@ -33,6 +33,7 @@ typedef struct {
 typedef struct {
     int alloc;
     int count;
+    int index;
     FCAddressInfo **addrs;
 } FCAddressPtrArray;
 
@@ -172,6 +173,16 @@ int fc_server_to_config_string(FCServerConfig *ctx, FastBuffer *buffer);
 
 void fc_server_to_log(FCServerConfig *ctx);
 
+ConnectionInfo *fc_server_check_connect_ex(FCAddressPtrArray *addr_array,
+        const int connect_timeout, const char *bind_ipaddr,
+        const bool log_connect_error, int *err_no);
+
+#define fc_server_check_connect(addr_array, connect_timeout, err_no) \
+    fc_server_check_connect_ex(addr_array, connect_timeout, NULL, true, err_no)
+
+
+void fc_server_disconnect(FCAddressPtrArray *addr_array);
+
 #ifdef __cplusplus
 }
 #endif