cfg.c

/*
 * Copyright 2010-2016, Tarantool AUTHORS, please see AUTHORS file.
 *
 * Redistribution and use in source and binary forms, with or
 * without modification, are permitted provided that the following
 * conditions are met:
 *
 * 1. Redistributions of source code must retain the above
 *    copyright notice, this list of conditions and the
 *    following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above
 *    copyright notice, this list of conditions and the following
 *    disclaimer in the documentation and/or other materials
 *    provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY <COPYRIGHT HOLDER> ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
 * <COPYRIGHT HOLDER> OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
 * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */
#include "say.h"
#include "cfg.h"
#include "lua/utils.h"
#include "tt_static.h"

enum { MAX_OPT_NAME_LEN = 256, MAX_OPT_VAL_LEN = 256 };

static void
cfg_get(const char *param)
{
	const char *buf =
		tt_snprintf(MAX_OPT_NAME_LEN, "return box.cfg.%s", param);
	if (luaL_dostring(tarantool_L, buf) != 0)
		panic("cfg_get('%s')", param);
}

int
cfg_geti(const char *param)
{
	cfg_get(param);
	int val;
	if (lua_isboolean(tarantool_L, -1))
		val = lua_toboolean(tarantool_L, -1);
	else
		val = lua_tointeger(tarantool_L, -1);
	lua_pop(tarantool_L, 1);
	return val;
}

int
cfg_getb(const char *param)
{
	cfg_get(param);
	int val;
	if (lua_isnil(tarantool_L, -1))
		return -1;
	val = lua_toboolean(tarantool_L, -1);
	lua_pop(tarantool_L, 1);
	return val;
}

int
cfg_geti_default(const char *param, int default_val)
{
	cfg_get(param);
	int ok;
	int val = lua_tointegerx(tarantool_L, -1, &ok);
	return ok ? val : default_val;
}

int64_t
cfg_geti64(const char *param)
{
	cfg_get(param);
	int64_t val = luaL_toint64(tarantool_L, -1);
	lua_pop(tarantool_L, 1);
	return val;
}

/* Support simultaneous cfg_gets("str1") and cfg_gets("str2") */
static const char *
cfg_tostring(struct lua_State *L)
{
	if (lua_isnil(L, -1))
		return NULL;
	return tt_snprintf(MAX_OPT_VAL_LEN, "%s", lua_tostring(L, -1));
}

const char *
cfg_gets(const char *param)
{
	cfg_get(param);
	const char *val = cfg_tostring(tarantool_L);
	lua_pop(tarantool_L, 1);
	return val;
}

double
cfg_getd(const char *param)
{
	cfg_get(param);
	double val = lua_tonumber(tarantool_L, -1);
	lua_pop(tarantool_L, 1);
	return val;
}

double
cfg_getd_default(const char *param, double default_val)
{
	cfg_get(param);
	int ok;
	double val = lua_tonumberx(tarantool_L, -1, &ok);
	return ok ? val : default_val;
}


int
cfg_getarr_size(const char *name)
{
	cfg_get(name);
	if (lua_isnil(tarantool_L, -1)) {
		/* missing value is equal to empty array */
		lua_pop(tarantool_L, 1);
		return 0;
	} else if (!lua_istable(tarantool_L, -1)) {
		/* scalars are handled like an array with one element */
		lua_pop(tarantool_L, 1);
		return 1;
	}

	int result = lua_objlen(tarantool_L, -1);
	lua_pop(tarantool_L, 1);
	return result;
}

const char *
cfg_getarr_elem(const char *name, int i)
{
	cfg_get(name);
	if (!lua_istable(tarantool_L, -1)) {
		/* scalars are handled like an array with one element */
		assert(i == 0 && !lua_isnil(tarantool_L, -1));
		const char *val = cfg_tostring(tarantool_L);
		lua_pop(tarantool_L, 1);
		return val;
	}

	lua_rawgeti(tarantool_L, -1, i + 1);
	const char *val = cfg_tostring(tarantool_L);
	lua_pop(tarantool_L, 2);
	return val;
}