94 lines
3.1 KiB
C
94 lines
3.1 KiB
C
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#ifndef EASY_POOL_H_
|
|
#define EASY_POOL_H_
|
|
|
|
/**
|
|
* 简单的内存池
|
|
*/
|
|
#include "easy_define.h"
|
|
#include "easy_list.h"
|
|
#include "easy_atomic.h"
|
|
|
|
EASY_CPP_START
|
|
|
|
#ifdef EASY_DEBUG_MAGIC
|
|
#define EASY_DEBUG_MAGIC_POOL 0x4c4f4f5059534145
|
|
#define EASY_DEBUG_MAGIC_MESSAGE 0x4753454d59534145
|
|
#define EASY_DEBUG_MAGIC_SESSION 0x5353455359534145
|
|
#define EASY_DEBUG_MAGIC_CONNECT 0x4e4e4f4359534145
|
|
#define EASY_DEBUG_MAGIC_REQUEST 0x5551455259534145
|
|
#endif
|
|
|
|
#define EASY_POOL_ALIGNMENT 512
|
|
#define EASY_POOL_PAGE_SIZE 4096
|
|
#define easy_pool_alloc(pool, size) easy_pool_alloc_ex(pool, size, sizeof(long))
|
|
#define easy_pool_nalloc(pool, size) easy_pool_alloc_ex(pool, size, 1)
|
|
|
|
typedef void *(*easy_pool_realloc_pt)(void *ptr, size_t size);
|
|
typedef struct easy_pool_large_t easy_pool_large_t;
|
|
typedef struct easy_pool_t easy_pool_t;
|
|
typedef void (easy_pool_cleanup_pt)(const void *data);
|
|
typedef struct easy_pool_cleanup_t easy_pool_cleanup_t;
|
|
|
|
struct easy_pool_large_t {
|
|
easy_pool_large_t *next;
|
|
uint8_t *data;
|
|
uint32_t size;
|
|
};
|
|
|
|
struct easy_pool_cleanup_t {
|
|
easy_pool_cleanup_pt *handler;
|
|
easy_pool_cleanup_t *next;
|
|
const void *data;
|
|
};
|
|
|
|
struct mod_stat_t;
|
|
struct easy_pool_t {
|
|
uint8_t *last;
|
|
uint8_t *end;
|
|
easy_pool_t *next;
|
|
uint16_t failed;
|
|
uint16_t flags;
|
|
uint32_t max;
|
|
|
|
// pool header
|
|
easy_pool_t *current;
|
|
easy_pool_large_t *large;
|
|
easy_atomic_t ref;
|
|
easy_atomic_t tlock;
|
|
easy_pool_cleanup_t *cleanup;
|
|
struct mod_stat_t* mod_stat;
|
|
#ifdef EASY_DEBUG_MAGIC
|
|
uint64_t magic;
|
|
#endif
|
|
};
|
|
|
|
extern easy_pool_realloc_pt easy_pool_realloc;
|
|
extern void *easy_pool_default_realloc (void *ptr, size_t size);
|
|
|
|
extern easy_pool_t *easy_pool_create(uint32_t size);
|
|
extern void easy_pool_clear(easy_pool_t *pool);
|
|
extern void easy_pool_destroy(easy_pool_t *pool);
|
|
extern void *easy_pool_alloc_ex(easy_pool_t *pool, uint32_t size, int align);
|
|
extern void *easy_pool_calloc(easy_pool_t *pool, uint32_t size);
|
|
extern void easy_pool_set_allocator(easy_pool_realloc_pt alloc);
|
|
extern void easy_pool_set_lock(easy_pool_t *pool);
|
|
extern easy_pool_cleanup_t *easy_pool_cleanup_new(easy_pool_t *pool, const void *data, easy_pool_cleanup_pt *handler);
|
|
extern void easy_pool_cleanup_reg(easy_pool_t *pool, easy_pool_cleanup_t *cl);
|
|
|
|
extern char *easy_pool_strdup(easy_pool_t *pool, const char *str);
|
|
|
|
EASY_CPP_END
|
|
#endif
|