skynet/lualib-src/lua-ssm.c

#define LUA_LIB

#include <lua.h>
#include <lauxlib.h>
#include <string.h>

#include "lstring.h"

static int
linfo(lua_State *L) {
	struct ssm_info info;
	memset(&info, 0, sizeof(info));
	luaS_infossm(&info);
	lua_createtable(L, 0, 5);
	lua_pushinteger(L, info.total);
	lua_setfield(L, -2, "n");
	lua_pushinteger(L, info.longest);
	lua_setfield(L, -2, "longest");
	lua_pushinteger(L, info.slots);
	lua_setfield(L, -2, "slots");
	lua_pushinteger(L, info.size);
	lua_setfield(L, -2, "size");
	lua_pushnumber(L, info.variance);
	lua_setfield(L, -2, "variance");

	return 1;
}

static int
lcollect(lua_State *L) {
	int loop = lua_toboolean(L, 1);
	if (loop) {
		int n = 0;
		struct ssm_collect info;
		while (luaS_collectssm(&info)) {
			n+=info.n;
		}
		lua_pushinteger(L, n);
		return 1;
	} else {
		struct ssm_collect info;
		int again = luaS_collectssm(&info);
		if (again && lua_istable(L, 2)) {
			lua_pushinteger(L, info.n);
			lua_setfield(L, 2, "n");
			lua_pushlightuserdata(L, info.key);
			lua_setfield(L, 2, "key");
		}
		lua_pushboolean(L, again);
		return 1;
	}
}

LUAMOD_API int
luaopen_skynet_ssm(lua_State *L) {
	luaL_checkversion(L);

	luaL_Reg l[] = {
		{ "info", linfo },
		{ "collect", lcollect },
		{ NULL, NULL },
	};

	luaL_newlib(L,l);

#ifndef ENABLE_SHORT_STRING_TABLE
	lua_pushboolean(L, 1);
	lua_setfield(L, -2, "disable");
#endif
	return 1;
}