Cleanups
This commit is contained in:
parent
a7a1d60ecb
commit
d41b55b25b
|
@ -3,35 +3,35 @@
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
int libretrodb_lua_to_rmsgpack_value(
|
int libretrodb_lua_to_rmsgpack_value(lua_State * L, int index, struct rmsgpack_dom_value * out)
|
||||||
lua_State * L,
|
{
|
||||||
int index,
|
lua_Number tmp_num;
|
||||||
struct rmsgpack_dom_value * out
|
size_t tmp_len;
|
||||||
) {
|
int i, rv = -1;
|
||||||
|
|
||||||
int rv = -1;
|
|
||||||
int i;
|
|
||||||
const char * tmp_string = NULL;
|
const char * tmp_string = NULL;
|
||||||
char * tmp_buff = NULL;
|
char * tmp_buff = NULL;
|
||||||
struct rmsgpack_dom_value * tmp_value;
|
struct rmsgpack_dom_value * tmp_value;
|
||||||
const int key_idx = -2;
|
const int key_idx = -2;
|
||||||
const int value_idx = -1;
|
const int value_idx = -1;
|
||||||
const int MAX_FIELDS = 100;
|
const int MAX_FIELDS = 100;
|
||||||
size_t tmp_len;
|
|
||||||
lua_Number tmp_num;
|
|
||||||
|
|
||||||
out->type = RDT_MAP;
|
out->type = RDT_MAP;
|
||||||
out->map.len = 0;
|
out->map.len = 0;
|
||||||
out->map.items = calloc(MAX_FIELDS, sizeof(struct rmsgpack_dom_pair));
|
out->map.items = calloc(MAX_FIELDS, sizeof(struct rmsgpack_dom_pair));
|
||||||
lua_pushnil(L);
|
lua_pushnil(L);
|
||||||
while (lua_next(L, index - 1) != 0) {
|
|
||||||
if (out->map.len > MAX_FIELDS) {
|
while (lua_next(L, index - 1) != 0)
|
||||||
|
{
|
||||||
|
if (out->map.len > MAX_FIELDS)
|
||||||
printf("skipping due to too many keys\n");
|
printf("skipping due to too many keys\n");
|
||||||
} else if (!lua_isstring(L, key_idx)) {
|
else if (!lua_isstring(L, key_idx))
|
||||||
printf("skipping non string key\n");
|
printf("skipping non string key\n");
|
||||||
} else if (lua_isnil(L, value_idx)) {
|
else if (lua_isnil(L, value_idx))
|
||||||
|
{
|
||||||
// Skipping nil value fields to save disk space
|
// Skipping nil value fields to save disk space
|
||||||
} else {
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
i = out->map.len;
|
i = out->map.len;
|
||||||
tmp_buff = strdup(lua_tostring(L, key_idx));
|
tmp_buff = strdup(lua_tostring(L, key_idx));
|
||||||
out->map.items[i].key.type = RDT_STRING;
|
out->map.items[i].key.type = RDT_STRING;
|
||||||
|
@ -39,7 +39,8 @@ int libretrodb_lua_to_rmsgpack_value(
|
||||||
out->map.items[i].key.string.buff = tmp_buff;
|
out->map.items[i].key.string.buff = tmp_buff;
|
||||||
|
|
||||||
tmp_value = &out->map.items[i].value;
|
tmp_value = &out->map.items[i].value;
|
||||||
switch (lua_type(L, value_idx)) {
|
switch (lua_type(L, value_idx))
|
||||||
|
{
|
||||||
case LUA_TNUMBER:
|
case LUA_TNUMBER:
|
||||||
tmp_num = lua_tonumber(L, value_idx);
|
tmp_num = lua_tonumber(L, value_idx);
|
||||||
tmp_value->type = RDT_INT;
|
tmp_value->type = RDT_INT;
|
||||||
|
@ -57,19 +58,25 @@ int libretrodb_lua_to_rmsgpack_value(
|
||||||
break;
|
break;
|
||||||
case LUA_TTABLE:
|
case LUA_TTABLE:
|
||||||
lua_getfield(L, value_idx, "binary");
|
lua_getfield(L, value_idx, "binary");
|
||||||
if (!lua_isstring(L, -1)) {
|
if (!lua_isstring(L, -1))
|
||||||
|
{
|
||||||
lua_pop(L, 1);
|
lua_pop(L, 1);
|
||||||
lua_getfield(L, value_idx, "uint");
|
lua_getfield(L, value_idx, "uint");
|
||||||
if (!lua_isnumber(L, -1)) {
|
if (!lua_isnumber(L, -1))
|
||||||
|
{
|
||||||
lua_pop(L, 1);
|
lua_pop(L, 1);
|
||||||
goto set_nil;
|
goto set_nil;
|
||||||
} else {
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
tmp_num = lua_tonumber(L, -1);
|
tmp_num = lua_tonumber(L, -1);
|
||||||
tmp_value->type = RDT_UINT;
|
tmp_value->type = RDT_UINT;
|
||||||
tmp_value->uint_ = tmp_num;
|
tmp_value->uint_ = tmp_num;
|
||||||
lua_pop(L, 1);
|
lua_pop(L, 1);
|
||||||
}
|
}
|
||||||
} else {
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
tmp_string = lua_tolstring(L, -1, &tmp_len);
|
tmp_string = lua_tolstring(L, -1, &tmp_len);
|
||||||
tmp_buff = malloc(tmp_len);
|
tmp_buff = malloc(tmp_len);
|
||||||
memcpy(tmp_buff, tmp_string, tmp_len);
|
memcpy(tmp_buff, tmp_string, tmp_len);
|
||||||
|
@ -90,4 +97,3 @@ set_nil:
|
||||||
rv = 0;
|
rv = 0;
|
||||||
return rv;
|
return rv;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -345,11 +345,11 @@ static struct rmsgpack_dom_value q_glob(struct rmsgpack_dom_value input,
|
||||||
static struct rmsgpack_dom_value all_map(struct rmsgpack_dom_value input,
|
static struct rmsgpack_dom_value all_map(struct rmsgpack_dom_value input,
|
||||||
unsigned argc, const struct argument *argv)
|
unsigned argc, const struct argument *argv)
|
||||||
{
|
{
|
||||||
|
unsigned i;
|
||||||
struct rmsgpack_dom_value res;
|
struct rmsgpack_dom_value res;
|
||||||
struct rmsgpack_dom_value *value = NULL;
|
|
||||||
struct argument arg;
|
struct argument arg;
|
||||||
struct rmsgpack_dom_value nil_value;
|
struct rmsgpack_dom_value nil_value;
|
||||||
unsigned i;
|
struct rmsgpack_dom_value *value = NULL;
|
||||||
|
|
||||||
nil_value.type = RDT_NULL;
|
nil_value.type = RDT_NULL;
|
||||||
res.type = RDT_BOOL;
|
res.type = RDT_BOOL;
|
||||||
|
|
Loading…
Reference in New Issue