cpp/src/common/allocator/alloc_base.h

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * License); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#ifndef COMMON_ALLOCATOR_ALLOC_BASE_H
#define COMMON_ALLOCATOR_ALLOC_BASE_H

#include <stddef.h>

#include <cstring>

#include "utils/util_define.h"

namespace common {

enum AllocModID {
    __FIRST_MOD_ID = 0,
    // if you are sure you will not consume too much memory, you can use
    // MOD_DEFAULT.
    MOD_DEFAULT = 0,
    MOD_MEMTABLE = 1,
    MOD_SCHEMA = 2,
    MOD_SQL = 3,
    MOD_NET = 4,
    MOD_NET_DATA = 5,
    MOD_TVLIST_DATA = 6,
    MOD_TVLIST_OBJ = 7,
    MOD_TSBLOCK = 8,
    MOD_CONTAINER = 9,
    MOD_TSSTORE_OBJ = 10,
    MOD_FLUSH_TASK_OBJ = 11,
    MOD_PAGE_WRITER_OUTPUT_STREAM = 12,
    MOD_CW_PAGES_DATA = 13,
    MOD_CHUNK_WRITER_OBJ = 14,
    MOD_STATISTIC_OBJ = 15,
    MOD_ENCODER_OBJ = 16,
    MOD_DECODER_OBJ = 17,
    MOD_TSFILE_WRITER_META = 18,
    MOD_TSFILE_WRITE_STREAM = 19,
    MOD_TIMESERIES_INDEX_OBJ = 20,
    MOD_BLOOM_FILTER = 21,
    MOD_OPEN_FILE_OBJ = 22,
    MOD_TSFILE_READER = 23,
    MOD_CHUNK_READER = 24,
    MOD_COMPRESSOR_OBJ = 25,
    MOD_ARRAY = 26,
    MOD_HASH_TABLE = 27,
    MOD_WRITER_INDEX_NODE = 28,
    MOD_TS2DIFF_OBJ = 29,
    MOD_BITENCODE_OBJ = 30,
    MOD_DICENCODE_OBJ = 31,
    MOD_ZIGZAG_OBJ = 32,
    MOD_DEVICE_META_ITER = 33,
    MOD_DEVICE_TASK_ITER = 34,
    MOD_DEVICE_ORDER_TSBLOCK_READER = 35,
    __LAST_MOD_ID = 36,  // prev + 1,
    __MAX_MOD_ID = 127,  // leave 1 bit to detect header size
};

extern const char *g_mod_names[__LAST_MOD_ID];

/* very basic alloc/free interface in C style */
void *mem_alloc(size_t size, AllocModID mid);
void mem_free(void *ptr);
void *mem_realloc(void *ptr, size_t size);

/* base allocator */
class BaseAllocator {
   public:
    static void *alloc(const size_t size, const AllocModID mid) { return mem_alloc(size, mid); }
    static void free(void *ptr) { mem_free(ptr); }
};

extern BaseAllocator g_base_allocator;

}  // end namespace common

#endif  // COMMON_ALLOCATOR_ALLOC_BASE_H