You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.0 KiB
41 lines
1.0 KiB
4 years ago
|
#ifndef _MEMORY_H_
|
||
|
|
#define _MEMORY_H_
|
||
|
|
|
||
|
|
#include <cstdlib>
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Pre-allocate a chunk of memory and allocate them in small pieces.
|
||
|
|
* Those memory are never freed after allocation. Used for persist
|
||
|
|
* data like dictionary contents that will never be destroyed unless
|
||
|
|
* the application exited.
|
||
|
|
*/
|
||
|
|
|
||
|
|
namespace rmmseg {
|
||
|
|
const size_t REALLOC_SIZE = 2048; /* 2KB */
|
||
|
|
|
||
|
|
extern size_t _pool_size;
|
||
|
|
extern char *_pool_base;
|
||
|
|
|
||
|
|
inline void *pool_alloc(size_t len) {
|
||
|
|
void *mem = _pool_base;
|
||
|
|
|
||
|
|
if (len <= _pool_size) {
|
||
|
|
_pool_size -= len;
|
||
|
|
_pool_base += len;
|
||
|
|
return mem;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* NOTE: the remaining memory is simply discard, which WILL
|
||
|
|
* cause memory leak. However, this function is not for allocating
|
||
|
|
* large object. Larger pre-alloc chunk size will also reduce the
|
||
|
|
* impact of this leak. So this is generally not a problem. */
|
||
|
|
_pool_base = static_cast<char *>(std::malloc(REALLOC_SIZE));
|
||
|
|
mem = _pool_base;
|
||
|
|
_pool_base += len;
|
||
|
|
_pool_size = REALLOC_SIZE - len;
|
||
|
|
return mem;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif /* _MEMORY_H_ */
|