lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 20:23:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

set SERVICE_PATH

Browse Source
This commit is contained in:
云风
2012-09-15 11:24:37 +08:00
parent 02a6a6aac0
commit b511c9269b
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
service-src/service_lua.c
View File

@@ -5,6 +5,7 @@
#include <lauxlib.h> #include <lauxlib.h>
#include "luacompat52.h" #include "luacompat52.h"
#include <assert.h>
#include <string.h> #include <string.h>
#include <stdlib.h> #include <stdlib.h>
#include <stdio.h> #include <stdio.h>
@@ -39,12 +40,23 @@ _try_load(lua_State *L, const char * path, int pathlen, const char * name) {
fclose(f); fclose(f);
int r = luaL_loadfile(L,tmp); int r = luaL_loadfile(L,tmp);
if (r == LUA_OK) { if (r == LUA_OK) {
int i;
for (i=namelen+pathlen-2;i>=0;i--) {
if (tmp[i] == '/') {
lua_pushlstring(L,tmp,i+1);
lua_setglobal(L,"SERVICE_PATH");
break;
}
}
if (i<0) {
return 0;
}
lua_getglobal(L,"package"); lua_getglobal(L,"package");
lua_getfield(L,-1,"path"); lua_getfield(L,-1,"path");
luaL_Buffer b; luaL_Buffer b;
luaL_buffinit(L, &b); luaL_buffinit(L, &b);
luaL_addlstring(&b, path, pathlen); luaL_addlstring(&b, tmp, i+1);
luaL_addchar(&b,';'); luaL_addstring(&b, "?.lua;");
luaL_addvalue(&b); luaL_addvalue(&b);
luaL_pushresult(&b); luaL_pushresult(&b);
lua_setfield(L,-2,"path"); lua_setfield(L,-2,"path");
@@ -104,6 +116,7 @@ snlua_init(lua_State *L, struct skynet_context *ctx, const char * args) {
lua_pushcfunction(L, traceback); lua_pushcfunction(L, traceback);
int traceback_index = lua_gettop(L); int traceback_index = lua_gettop(L);
assert(traceback_index == 1);
const char * filename = parm; const char * filename = parm;
int r = _load(L, &parm); int r = _load(L, &parm);