From 25dacd361f84cc5e41241f9fe3ab662d5931014e Mon Sep 17 00:00:00 2001
From: YuQing <384681@qq.com>
Date: Wed, 17 Feb 2021 11:18:56 +0800
Subject: [PATCH] impl connection_manager init and add

---
 src/sf_connection_manager.c | 111 +++++++++++++++++++++++++++++++++++-
 src/sf_connection_manager.h |   9 ++-
 2 files changed, 116 insertions(+), 4 deletions(-)

diff --git a/src/sf_connection_manager.c b/src/sf_connection_manager.c
index 7fbeafd..4bbfaaa 100644
--- a/src/sf_connection_manager.c
+++ b/src/sf_connection_manager.c
@@ -23,12 +23,119 @@
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <errno.h>
+#include "sf/sf_global.h"
 #include "fastcommon/shared_func.h"
 #include "fastcommon/logger.h"
 #include "sf_connection_manager.h"
 
-int sf_connection_manager_init(SFConnectionManager *cm, const int group_count,
-        const int server_group_index)
+static int init_group_array(SFCMConnGroupArray *garray, const int group_count,
+        const int min_group_id)
 {
+    int result;
+    int bytes;
+    SFCMConnGroupEntry *group;
+    SFCMConnGroupEntry *end;
+
+    bytes = sizeof(SFCMConnGroupEntry) * group_count;
+    garray->entries = (SFCMConnGroupEntry *)fc_malloc(bytes);
+    if (garray->entries == NULL) {
+        return ENOMEM;
+    }
+    memset(garray->entries, 0, bytes);
+
+    end = garray->entries + group_count;
+    for (group=garray->entries; group<end; group++) {
+        if ((result=init_pthread_lock(&group->lock)) != 0) {
+            return result;
+        }
+    }
+
+    garray->count = group_count;
+    garray->min_group_id = min_group_id;
+    garray->max_group_id = min_group_id + group_count - 1;
     return 0;
 }
+
+int sf_connection_manager_init(SFConnectionManager *cm, const int group_count,
+        const int min_group_id, const int server_group_index,
+        const SFDataReadRule read_rule)
+{
+    int result;
+
+    if ((result=init_group_array(&cm->groups, group_count,
+                    min_group_id)) != 0)
+    {
+        return result;
+    }
+
+    cm->server_group_index = server_group_index;
+    cm->read_rule = read_rule;
+    return 0;
+}
+
+int sf_connection_manager_add(SFConnectionManager *cm, const int group_id,
+        FCServerInfo **servers, const int count)
+{
+    SFCMConnGroupEntry *group;
+    FCServerInfo **server;
+    FCServerInfo **end;
+    SFCMServerEntry *entry;
+
+    if (group_id < cm->groups.min_group_id) {
+        logError("file: "__FILE__", line: %d, "
+                "invalid group id: %d which < min group id: %d",
+                __LINE__, group_id, cm->groups.min_group_id);
+        return EINVAL;
+    }
+    if (group_id > cm->groups.max_group_id) {
+        logError("file: "__FILE__", line: %d, "
+                "invalid group id: %d which > max group id: %d",
+                __LINE__, group_id, cm->groups.max_group_id);
+        return EINVAL;
+    }
+
+    group = cm->groups.entries + (group_id - cm->groups.min_group_id);
+    group->all.servers = (SFCMServerEntry *)fc_malloc(
+            sizeof(SFCMServerEntry) * count);
+    if (group->all.servers == NULL) {
+        return ENOMEM;
+    }
+    group->all.count = count;
+
+    end = servers + count;
+    for (entry=group->all.servers, server=servers;
+            server<end; entry++, server++)
+    {
+        entry->server_id = (*server)->id;
+        entry->addr_array = &(*server)->group_addrs[
+            cm->server_group_index].address_array;
+        entry->conn = NULL;
+    }
+
+    group->alives.servers = (SFCMServerEntry **)fc_malloc(
+            sizeof(SFCMServerEntry *) * count);
+    if (group->alives.servers == NULL) {
+        return ENOMEM;
+    }
+
+    return 0;
+}
+
+static void *connection_manager_thread_func(void *arg)
+{
+    SFConnectionManager *cm;
+
+    cm = (SFConnectionManager *)arg;
+    while (1) {
+        //TODO
+    }
+
+    return NULL;
+}
+
+int sf_connection_manager_start(SFConnectionManager *cm)
+{
+    pthread_t tid;
+    return fc_create_thread(&tid, connection_manager_thread_func,
+            cm, SF_G_THREAD_STACK_SIZE);
+}
diff --git a/src/sf_connection_manager.h b/src/sf_connection_manager.h
index de831c9..eb5d8b6 100644
--- a/src/sf_connection_manager.h
+++ b/src/sf_connection_manager.h
@@ -40,6 +40,7 @@ typedef struct sf_cm_server_ptr_array {
 
 typedef struct sf_cm_conn_group_entry {
     SFCMServerEntry *master;
+    SFCMServerArray all;
     SFCMServerPtrArray alives;
     pthread_mutex_t lock;
 } SFCMConnGroupEntry;
@@ -47,7 +48,8 @@ typedef struct sf_cm_conn_group_entry {
 typedef struct sf_cm_conn_group_array {
     SFCMConnGroupEntry *entries;
     int count;
-    int base_id;
+    int min_group_id;
+    int max_group_id;
 } SFCMConnGroupArray;
 
 typedef struct sf_connection_manager {
@@ -57,11 +59,14 @@ typedef struct sf_connection_manager {
 } SFConnectionManager;
 
 int sf_connection_manager_init(SFConnectionManager *cm, const int group_count,
-        const int server_group_index);
+        const int min_group_id, const int server_group_index,
+        const SFDataReadRule read_rule);
 
 int sf_connection_manager_add(SFConnectionManager *cm, const int group_id,
         FCServerInfo **servers, const int count);
 
+int sf_connection_manager_start(SFConnectionManager *cm);
+
 ConnectionInfo *sf_connection_manager_get_master(SFConnectionManager *cm,
         const int group_index, int *err_no);