INTERNAL: Refactor the structure of additional item in mblck

naver · Apr 29, 2024 · 779a775 · 779a775
1 parent c1539bd
commit 779a775
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 8 deletions.
diff --git a/mc_util.c b/mc_util.c
@@ -31,7 +31,7 @@ static void do_mblck_pool_init(mblck_pool_t *pool, uint32_t blck_len)
     pool->tail = NULL;
     pool->head = NULL;
     pool->blck_len = blck_len;
-    pool->body_len = blck_len - sizeof(void *);
+    pool->body_len = blck_len - (sizeof(void *) + sizeof(uint32_t));
     pool->used_cnt = 0;
     pool->free_cnt = 0;
 }
@@ -107,16 +107,29 @@ int mblck_list_alloc(mblck_pool_t *pool, uint32_t item_len, uint32_t item_cnt,
     }
     assert(pool->free_cnt >= blck_cnt);
 
+    if (blck_cnt > 1) {
+        list->rvec = (value_item **)malloc(sizeof(value_item *) * (blck_cnt - 1));
+        if (list->rvec == NULL) {
+            return -1;
+        }
+    } else {
+        list->rvec = NULL;
+    }
+
+    uint32_t cnt = 0;
     list->pool = (void*)pool;
     list->head = pool->head;
     list->tail = list->head;
     list->blck_cnt = 1;
-    while (list->blck_cnt < blck_cnt) {
+    while (list->blck_cnt + cnt < blck_cnt) {
         list->tail = list->tail->next;
-        list->blck_cnt += 1;
+        list->rvec[cnt] = &list->tail->data;
+        list->rvec[cnt]->len = nitems_per_blck * item_len;
+        cnt += 1;
     }
     pool->head = list->tail->next;
     list->tail->next = NULL;
+    list->blck_cnt += cnt;
 
     if (pool->head == NULL) {
         pool->tail = NULL;
@@ -141,7 +154,11 @@ void mblck_list_merge(mblck_list_t *pri_list, mblck_list_t *add_list)
     add_list->head = NULL;
     add_list->tail = NULL;
     add_list->blck_cnt = 0;
-    /* FIXME: item_cnt and item_len: how to merge them ? */
+    if (add_list->rvec != NULL) {
+        free(add_list->rvec);
+        add_list->rvec = NULL;
+    }
+    /* FIXME: item_cnt, item_len and rvec: how to merge them ? */
 }
 
 void mblck_list_free(mblck_pool_t *pool, mblck_list_t *list)
@@ -160,6 +177,10 @@ void mblck_list_free(mblck_pool_t *pool, mblck_list_t *list)
         list->head = NULL;
         list->tail = NULL;
         list->blck_cnt = 0;
+        if (list->rvec != NULL) {
+            free(list->rvec);
+            list->rvec = NULL;
+        }
     }
     list->pool = NULL;
 }

diff --git a/mc_util.h b/mc_util.h
@@ -37,13 +37,14 @@ typedef struct _token_buff {
  */
 typedef struct _mblck_node {
     struct _mblck_node *next;
-    char data[1];
+    value_item data;
 } mblck_node_t;
 
 typedef struct _mblck_list {
     void         *pool;
     mblck_node_t *head;
     mblck_node_t *tail;
+    value_item  **rvec;
     uint32_t  blck_cnt;
     uint32_t  body_len;
     uint32_t  item_cnt;
@@ -69,7 +70,7 @@ typedef struct _mblck_pool {
 #define MBLCK_GET_ITEMCNT(l) ((l)->item_cnt)
 #define MBLCK_GET_ITEMLEN(l) ((l)->item_len)
 #define MBLCK_GET_NEXTBLK(b) ((b)->next)
-#define MBLCK_GET_BODYPTR(b) ((b)->data)
+#define MBLCK_GET_BODYPTR(b) ((b)->data.ptr)
 
 /* memory block functions */
 int  mblck_pool_create(mblck_pool_t *pool, uint32_t blck_len, uint32_t blck_cnt);

diff --git a/memcached.c b/memcached.c
@@ -1022,6 +1022,7 @@ static void ritem_set_first(conn *c, int rtype, int vleng)
         c->rlbytes = vleng < MBLCK_GET_BODYLEN(&c->memblist)
                    ? vleng : MBLCK_GET_BODYLEN(&c->memblist);
         c->rltotal = vleng;
+        c->rindex = 0;
     }
     else if (c->rtype == CONN_RTYPE_HINFO) {
         if (c->hinfo.naddnl == 0) {
@@ -1070,10 +1071,10 @@ static void ritem_set_next(conn *c)
     assert(c->rltotal > 0);
 
     if (c->rtype == CONN_RTYPE_MBLCK) {
-        c->membk = MBLCK_GET_NEXTBLK(c->membk);
-        c->ritem = MBLCK_GET_BODYPTR(c->membk);
+        c->ritem = c->memblist.rvec[c->rindex]->ptr;
         c->rlbytes = c->rltotal < MBLCK_GET_BODYLEN(&c->memblist)
                    ? c->rltotal : MBLCK_GET_BODYLEN(&c->memblist);
+        c->rindex += 1;
     }
     else if (c->rtype == CONN_RTYPE_HINFO) {
         c->ritem = c->hinfo.addnl[c->rindex]->ptr;