我最近正在尝试跟踪我们项目中的 lua 内存使用情况,并且我遇到了使用 lua_Alloc 自定义分配器来完成此任务的想法。好吧,分配器代码看起来很简单,而且看起来工作正常。
但是很快,这个小函数就遇到了两个挑战:
1.它输出的结果与collectgarbage('count')给出的值有很大的不同;
2.假设当前内存使用量是M字节,那么我们输入一些nil引用并调用gc,内存使用量会大于M字节。例如:A 返回,B 返回,C 返回,...,collectgarbage()
好吧,我听说如果你正确使用 lua,就不会有内存泄漏,所以我认为我在计算内存使用量时做错了。请帮我弄清楚。提前致谢。
下面附上可编译的代码:
extern "C"
{
#include "lua.h"
#include <lauxlib.h>
#include <lualib.h>
};
#include <cstdio>
#include <string>
#include <cstdlib>
using namespace std;
struct Tracker
{
size_t m_usage;
}g_tracker;
void*
Allocator(void* ud, void* ptr, size_t osize, size_t nsize)
{
Tracker* pTracker = (Tracker*)ud;
void* pRet = NULL;
if( nsize == 0 )
{
pTracker->m_usage -= osize;
//printf("Free %d bytes; ", osize);
free(ptr);
}
else
{
pTracker->m_usage -= osize;
//printf("first Free %d bytes; ", osize);
pTracker->m_usage += nsize;
//printf("then alloc %d bytes; ", nsize);
pRet = realloc(ptr, nsize);
}
//printf("current usage: %d bytes\n", pTracker->m_usage);
return pRet;
}
int main()
{
lua_State* L = lua_newstate(Allocator, &g_tracker );
luaL_openlibs(L);
char buf[4096];
while(true)
{
fgets(buf, 4096, stdin);
if( strlen(buf) > 1 && 0 != luaL_dostring(L, buf) )
{
const char* errmsg = lua_tostring(L, -1);
printf("%s\n", errmsg);
}
printf("current usage: %d bytes \n", g_tracker.m_usage);
}
}
关于#2的输入序列:
press enter
current usage: 18867 bytes
a=nil; b=nil; c=nil;
current usage: 19311 bytes
collectgarbage()
current usage: 18900 bytes
d=nil; e=nil; f=nil;
current usage: 19345 bytes
collectgarbage()
current usage: 18900 bytes
collectgarbage()
current usage: 18900 bytes
for a = 1, 1000, 1 do b = nil end; collectgarbage()
current usage: 19206 bytes
collectgarbage()
current usage: 18900 bytes
for i = 1, 1000, 1 do X = {}; for j = 1, 1000, 1 do X[j] = j end ; X = nil; collectgarbage() ; end
current usage: 19391 bytes
collectgarbage()
current usage: 18900 bytes