95 lines
2.6 KiB
C++
95 lines
2.6 KiB
C++
#ifndef _MSG_ALLOCATOR_H_
|
|
#define _MSG_ALLOCATOR_H_
|
|
|
|
#include "os_def.h"
|
|
#include "x_tick.h"
|
|
#include "base_allocator.hpp"
|
|
#include "x_lock.h"
|
|
|
|
class BaseAllocator;
|
|
|
|
using namespace tick64;
|
|
using namespace lock;
|
|
|
|
class BufferAllocator: public BaseAllocator
|
|
{
|
|
|
|
public:
|
|
static const unsigned int SmallBufferSize = 64; //小块内存块大小
|
|
static const unsigned int MiddleBufferSize = 256; //中等内存块大小
|
|
static const unsigned int LargeBufferSize = 1024; //大块内存块大小
|
|
|
|
#pragma pack(push, 1)
|
|
typedef struct tagBuffer
|
|
{
|
|
size_t size_; //内存数据段大小
|
|
tagBuffer* prev_buffer_;//指向上一个内存块的指针
|
|
int64_t free_tick_; //即将被销毁的时间
|
|
#ifdef _MEMORY_TRACE_
|
|
tagBuffer* next_buffer_;//指向上下个内存块的指针，用于调试
|
|
struct
|
|
{
|
|
const char* fn_; //申请内存块的代码位置，用于调试
|
|
int line_;
|
|
} al, fl;
|
|
bool used_flag_; //是否使用的标记
|
|
#endif
|
|
} BUFFER, *PBUFFER;
|
|
#pragma pack(pop)
|
|
|
|
struct BufferStatic
|
|
{
|
|
size_t alloc_size_;
|
|
size_t free_size_;
|
|
};
|
|
|
|
typedef struct tagAllocatorMemoryInfo
|
|
{
|
|
BufferStatic SmallBuffer;
|
|
BufferStatic MiddleBuffer;
|
|
BufferStatic LargeBuffer;
|
|
BufferStatic SuperBuffer;
|
|
} ALLOCATOR_MEMORY_INFO;
|
|
|
|
private:
|
|
Mutex buffer_lock_; //内存块申请锁
|
|
PBUFFER last_small_buffer_; //指向最后一个空闲的小型内存块的指针
|
|
PBUFFER last_middle_buffer_; //指向最后一个空闲的中型内存块的指针
|
|
PBUFFER last_large_buffer_; //指向最后一个空闲的大型内存块的指针
|
|
PBUFFER last_super_buffer_; //指向最后一个空闲的超级内存块的指针
|
|
#ifdef _MEMORY_TRACE_
|
|
PBUFFER last_alloced_buffer_; //指向最后一个被申请的内存块
|
|
#endif
|
|
size_t alloced_small_size_; //已被申请的小型内存块大小
|
|
size_t alloced_middle_size_; //已被申请的中型内存块大小
|
|
size_t alloced_large_size_; //已被申请的大型内存块大小
|
|
size_t alloced_super_size_; //已被申请的超级内存块大小
|
|
size_t freed_small_size_; //已被释放的小型内存块大小
|
|
size_t freed_middle_size_; //已被释放的中型内存块大小
|
|
size_t freed_large_size_; //已被释放的大型内存块大小
|
|
size_t freed_super_size_; //已被释放的超级内存块大小
|
|
|
|
bool mult_thread_;
|
|
public:
|
|
BufferAllocator(const char* namestr);
|
|
~BufferAllocator();
|
|
|
|
#ifndef _MEMORY_TRACE_
|
|
void* AllocBuffer(size_t size);
|
|
void* ReAllocBuffer(void* ptr, size_t new_size);
|
|
void FreeBuffer(void* ptr);
|
|
#else
|
|
void* _AllocBuffer(size_t size, const char* fn, int line);
|
|
void* _ReAllocBuffer(void* ptr, size_t new_size, const char* fn, int line);
|
|
void _FreeBuffer(void* ptr, const char* fn, int line);
|
|
#endif
|
|
|
|
void GetMemoryInfo(BufferAllocator::ALLOCATOR_MEMORY_INFO* pMemoryInfo);
|
|
void CheckFreeBuffers();
|
|
void SetMultiThread(bool flag);
|
|
void GC();
|
|
};
|
|
|
|
#endif
|
|
|