使用自定义Lua分配器来计算内存使用量,但其结果与collectgarbage不同(' count')

时间:2012-04-15 11:37:18

标签: memory garbage-collection lua allocator

我最近试图跟踪项目中的lua内存使用情况,并且我想到了使用lua_Alloc自定义分配器来完成此任务。好吧,分配器代码看起来很简单,看起来效果很好。

但很快,这个小功能遇到了两个挑战:
1.它输出的结果与collectgarbage给出的值完全不同(' count');
2.假设当前内存使用量为M字节,然后我们输入一些nil引用并调用gc,内存使用量将大于M字节。例如:返回,B返回,C返回,...,collectgarbage()

好吧,我听说如果你正确地使用lua,就没有mem-leak,所以我认为我在计算内存使用量方面做错了。请帮我搞清楚。提前谢谢。

可编辑的代码如下:

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

1 个答案:

答案 0 :(得分:5)

如Lua文档collectgarbage('collect')中所述,返回千字节中的值。 1KiB == 1024字节。因此,当collectgarbage表示您使用的是19.4775390625KiB时,转换为19945字节。这完全等于分配器获得的许多值之一。

你的记忆检查功能只会与Lua在一瞬间得到的东西相匹配:collectgarbage返回的那一刻。在它之前或之后发生的任何Lua分配/解除分配(例如为了显示值而调用print所需的那些)将会使你的计数失效。

简而言之,当您无法获得准确性时,您会期望字节精度。至少,不是你在那里的代码。

一般来说,你不应该担心它。