more

include/util/amalloc.h → include/util/mallocator.h

@@ -13,8 +13,8 @@
  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  */
 
-#ifndef _TD_AMALLOC_H_
-#define _TD_AMALLOC_H_
+#ifndef _TD_MALLOCATOR_H_
+#define _TD_MALLOCATOR_H_
 
 #include "os.h"
 
@@ -22,31 +22,36 @@
 extern "C" {
 #endif
 
-#define AMALLOC_APIS                                  \
-  void *(*malloc)(void *, size_t size);               \
-  void *(*calloc)(void *, size_t nmemb, size_t size); \
-  void *(*realloc)(void *, size_t size);              \
-  void (*free)(void *ptr);
+typedef struct SMemAllocator SMemAllocator;
+
+#define MALLOCATOR_APIS                                        \
+  void *(*malloc)(SMemAllocator *, size_t size);               \
+  void *(*calloc)(SMemAllocator *, size_t nmemb, size_t size); \
+  void *(*realloc)(SMemAllocator *, size_t size);              \
+  void (*free)(SMemAllocator *, void *ptr);                    \
+  size_t (*usage)(SMemAllocator *);
 
 // Interfaces to implement
 typedef struct {
-  AMALLOC_APIS
+  MALLOCATOR_APIS
 } SMemAllocatorIf;
 
-typedef struct {
-  void *impl;
-  AMALLOC_APIS
-} SMemAllocator;
+struct SMemAllocator {
+  void * impl;
+  size_t usize;
+  MALLOCATOR_APIS
+};
+
+// heap allocator
+SMemAllocator *tdCreateHeapAllocator();
+void           tdDestroyHeapAllocator(SMemAllocator *pMemAllocator);
 
-#define amalloc(allocator, size) ((allocator) ? (*((allocator)->malloc))((allocator)->impl, (size)) : malloc(size))
-#define acalloc(allocator, nmemb, size) \
-  ((allocator) ? (*((allocator)->calloc))((allocator)->impl, (nmemb), (size)) : calloc((nmemb), (size)))
-#define arealloc(allocator, ptr, size) \
-  ((allocator) ? (*((allocator)->realloc))((allocator)->impl, (ptr), (size)) : realloc((ptr), (size)))
-#define afree(allocator, ptr, size) ((allocator) ? (*((allocator)->free))((allocator)->impl, (ptr), (size)) : free(ptr))
+// arena allocator
+SMemAllocator *tdCreateArenaAllocator(size_t size);
+void           tdDestroyArenaAllocator(SMemAllocator *);
 
 #ifdef __cplusplus
 }
 #endif
 
-#endif /*_TD_AMALLOC_H_*/
\ No newline at end of file
+#endif /*_TD_MALLOCATOR_H_*/
\ No newline at end of file

source/dnode/vnode/impl/inc/vnodeInt.h

@@ -18,7 +18,6 @@
 
 #include "vnode.h"
 
-#include "amalloc.h"
 #include "meta.h"
 #include "sync.h"
 #include "tlog.h"

source/util/src/arenaAllocator.c

+/*
+ * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
+ *
+ * This program is free software: you can use, redistribute, and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3
+ * or later ("AGPL"), as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
\ No newline at end of file

source/util/src/heapAllocator.c

+/*
+ * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
+ *
+ * This program is free software: you can use, redistribute, and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3
+ * or later ("AGPL"), as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "mallocator.h"
+
+typedef struct {
+  char name[64];
+} SHeapAllocator;
+
+static SHeapAllocator *haNew();
+static void            haDestroy(SHeapAllocator *pha);
+static void *          haMalloc(SMemAllocator *pMemAllocator, size_t size);
+void *                 haCalloc(SMemAllocator *pMemAllocator, size_t nmemb, size_t size);
+static void            haFree(SMemAllocator *pMemAllocator, void *ptr);
+
+SMemAllocator *tdCreateHeapAllocator() {
+  SMemAllocator *pMemAllocator = NULL;
+
+  pMemAllocator = (SMemAllocator *)calloc(1, sizeof(*pMemAllocator));
+  if (pMemAllocator == NULL) {
+    // TODO: handle error
+    return NULL;
+  }
+
+  pMemAllocator->impl = haNew();
+  if (pMemAllocator->impl == NULL) {
+    tdDestroyHeapAllocator(pMemAllocator);
+    return NULL;
+  }
+
+  pMemAllocator->usage = 0;
+  pMemAllocator->malloc = haMalloc;
+  pMemAllocator->calloc = haCalloc;
+  pMemAllocator->realloc = NULL;
+  pMemAllocator->free = haFree;
+  pMemAllocator->usage = NULL;
+
+  return pMemAllocator;
+}
+
+void tdDestroyHeapAllocator(SMemAllocator *pMemAllocator) {
+  if (pMemAllocator) {
+    // TODO
+  }
+}
+
+/* ------------------------ STATIC METHODS ------------------------ */
+static SHeapAllocator *haNew() {
+  SHeapAllocator *pha = NULL;
+  /* TODO */
+  return pha;
+}
+
+static void haDestroy(SHeapAllocator *pha) {
+  // TODO
+}
+
+static void *haMalloc(SMemAllocator *pMemAllocator, size_t size) {
+  void *ptr = NULL;
+
+  ptr = malloc(size);
+  if (ptr) {
+  }
+
+  return ptr;
+}
+
+void *haCalloc(SMemAllocator *pMemAllocator, size_t nmemb, size_t size) {
+  /* TODO */
+  return NULL;
+}
+
+static void haFree(SMemAllocator *pMemAllocator, void *ptr) { /* TODO */
+}
\ No newline at end of file