Init Repo

vendor/swoole/src/core/array.c

@@ -0,0 +1,149 @@
+/*
+ +----------------------------------------------------------------------+
+ | Swoole                                                               |
+ +----------------------------------------------------------------------+
+ | This source file is subject to version 2.0 of the Apache license,    |
+ | that is bundled with this package in the file LICENSE, and is        |
+ | available through the world-wide-web at the following url:           |
+ | http://www.apache.org/licenses/LICENSE-2.0.html                      |
+ | If you did not receive a copy of the Apache2.0 license and are unable|
+ | to obtain it through the world-wide-web, please send a note to       |
+ | license@swoole.com so we can mail you a copy immediately.            |
+ +----------------------------------------------------------------------+
+ | Author: Tianfeng Han  <mikan.tenny@gmail.com>                        |
+ +----------------------------------------------------------------------+
+ */
+
+#include "swoole.h"
+#include "array.h"
+
+/**
+ * Create new array
+ */
+swArray *swArray_new(int page_size, size_t item_size)
+{
+    swArray *array = sw_malloc(sizeof(swArray));
+    if (array == NULL)
+    {
+        swoole_error_log(SW_LOG_ERROR, SW_ERROR_MALLOC_FAIL, "malloc[0] failed.");
+        return NULL;
+    }
+    bzero(array, sizeof(swArray));
+
+    array->pages = sw_malloc(sizeof(void*) * SW_ARRAY_PAGE_MAX);
+    if (array->pages == NULL)
+    {
+        sw_free(array);
+        swoole_error_log(SW_LOG_ERROR, SW_ERROR_MALLOC_FAIL, "malloc[1] failed.");
+        return NULL;
+    }
+
+    array->item_size = item_size;
+    array->page_size = page_size;
+
+    swArray_extend(array);
+
+    return array;
+}
+
+/**
+ * Destory the array
+ */
+void swArray_free(swArray *array)
+{
+    int i;
+    for (i = 0; i < array->page_num; i++)
+    {
+        sw_free(array->pages[i]);
+    }
+    sw_free(array->pages);
+    sw_free(array);
+}
+
+/**
+ * Extend the memory pages of the array
+ */
+int swArray_extend(swArray *array)
+{
+    if (array->page_num == SW_ARRAY_PAGE_MAX)
+    {
+        swWarn("max page_num is %d", array->page_num);
+        return SW_ERR;
+    }
+    array->pages[array->page_num] = sw_calloc(array->page_size, array->item_size);
+    if (array->pages[array->page_num] == NULL)
+    {
+        swWarn("malloc[1] failed.");
+        return SW_ERR;
+    }
+    array->page_num++;
+    return SW_OK;
+}
+
+/**
+ * Fetch data by index of the array
+ */
+void *swArray_fetch(swArray *array, uint32_t n)
+{
+    int page = swArray_page(array, n);
+    if (page >= array->page_num)
+    {
+        return NULL;
+    }
+    return array->pages[page] + (swArray_offset(array, n) * array->item_size);
+}
+
+/**
+ * Append to the array
+ */
+int swArray_append(swArray *array, void *data)
+{
+    int n = array->offset++;
+    int page = swArray_page(array, n);
+
+    if (page >= array->page_num && swArray_extend(array) < 0)
+    {
+        return SW_ERR;
+    }
+    array->item_num++;
+    memcpy(array->pages[page] + (swArray_offset(array, n) * array->item_size), data, array->item_size);
+    return n;
+}
+
+
+int swArray_store(swArray *array, uint32_t n, void *data)
+{
+    int page = swArray_page(array, n);
+    if (page >= array->page_num)
+    {
+        swWarn("fetch index[%d] out of array", n);
+        return SW_ERR;
+    }
+    memcpy(array->pages[page] + (swArray_offset(array, n) * array->item_size), data, array->item_size);
+    return SW_OK;
+}
+
+void *swArray_alloc(swArray *array, uint32_t n)
+{
+    while (n >= array->page_num * array->page_size)
+    {
+        if (swArray_extend(array) < 0)
+        {
+            return NULL;
+        }
+    }
+
+    int page = swArray_page(array, n);
+    if (page >= array->page_num)
+    {
+        swWarn("fetch index[%d] out of array", n);
+        return NULL;
+    }
+    return array->pages[page] + (swArray_offset(array, n) * array->item_size);
+}
+
+void swArray_clear(swArray *array)
+{
+    array->offset = 0;
+    array->item_num = 0;
+}