mirror of
https://github.com/cloudwu/skynet.git
synced 2026-07-22 02:53:09 +00:00
118 lines
2.6 KiB
C
118 lines
2.6 KiB
C
#include "skynet_imp.h"
|
|
#include "skynet_env.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <lua.h>
|
|
#include <lualib.h>
|
|
#include <lauxlib.h>
|
|
|
|
static int
|
|
optint(const char *key, int opt) {
|
|
const char * str = skynet_getenv(key);
|
|
if (str == NULL) {
|
|
char tmp[20];
|
|
sprintf(tmp,"%d",opt);
|
|
skynet_setenv(key, tmp);
|
|
return opt;
|
|
}
|
|
return strtol(str, NULL, 10);
|
|
}
|
|
|
|
static int
|
|
optboolean(const char *key, int opt) {
|
|
const char * str = skynet_getenv(key);
|
|
if (str == NULL) {
|
|
skynet_setenv(key, opt ? "true" : "false");
|
|
return opt;
|
|
}
|
|
return strcmp(str,"true")==0;
|
|
}
|
|
|
|
static const char *
|
|
optstring(const char *key,const char * opt) {
|
|
const char * str = skynet_getenv(key);
|
|
if (str == NULL) {
|
|
if (opt) {
|
|
skynet_setenv(key, opt);
|
|
}
|
|
return opt;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
static void
|
|
_init_env(lua_State *L) {
|
|
lua_rawgeti(L, LUA_REGISTRYINDEX, LUA_RIDX_GLOBALS);
|
|
lua_pushnil(L); /* first key */
|
|
while (lua_next(L, -2) != 0) {
|
|
int keyt = lua_type(L, -2);
|
|
if (keyt != LUA_TSTRING) {
|
|
fprintf(stderr, "Invalid config table\n");
|
|
exit(1);
|
|
}
|
|
const char * key = lua_tostring(L,-2);
|
|
if (lua_type(L,-1) == LUA_TBOOLEAN) {
|
|
int b = lua_toboolean(L,-1);
|
|
skynet_setenv(key,b ? "true" : "false" );
|
|
} else {
|
|
const char * value = lua_tostring(L,-1);
|
|
if (value == NULL) {
|
|
fprintf(stderr, "Invalid config table key = %s\n", key);
|
|
exit(1);
|
|
}
|
|
skynet_setenv(key,value);
|
|
}
|
|
lua_pop(L,1);
|
|
}
|
|
lua_pop(L,1);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[]) {
|
|
const char * config_file = "config";
|
|
if (argc > 1) {
|
|
config_file = argv[1];
|
|
}
|
|
skynet_env_init();
|
|
|
|
struct skynet_config config;
|
|
|
|
struct lua_State *L = luaL_newstate();
|
|
luaL_openlibs(L); // link lua lib
|
|
lua_close(L);
|
|
|
|
L = luaL_newstate();
|
|
|
|
int err = luaL_dofile(L, config_file);
|
|
if (err) {
|
|
fprintf(stderr,"%s\n",lua_tostring(L,-1));
|
|
lua_close(L);
|
|
return 1;
|
|
}
|
|
_init_env(L);
|
|
|
|
const char *path = optstring("lua_path","./lualib/?.lua;./lualib/?/init.lua");
|
|
setenv("LUA_PATH",path,1);
|
|
const char *cpath = optstring("lua_cpath","./lualib/?.so");
|
|
setenv("LUA_CPATH",cpath,1);
|
|
optstring("luaservice","./service/?.lua");
|
|
|
|
config.thread = optint("thread",8);
|
|
config.mqueue_size = optint("mqueue",256);
|
|
config.module_path = optstring("cpath","./service/?.so");
|
|
config.logger = optstring("logger",NULL);
|
|
config.harbor = optint("harbor", 1);
|
|
config.master = optstring("master","tcp://127.0.0.1:2012");
|
|
config.start = optstring("start","main.lua");
|
|
config.local = optstring("address","tcp://127.0.0.1:2525");
|
|
config.standalone = optboolean("standalone",0);
|
|
|
|
lua_close(L);
|
|
|
|
skynet_start(&config);
|
|
|
|
return 0;
|
|
}
|