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

cache lua code load

Browse Source
This commit is contained in:
云风
2013-12-17 18:26:08 +08:00
parent 2bf65084cd
commit 8e031c1b21
5 changed files with 340 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
service-src/service_lua.c
View File

@@ -6,6 +6,7 @@
#include <assert.h>
#include "luacompat52.h"
#include "service_lua.h"
#include "luacode_cache.h"
#include <assert.h>
#include <string.h>
@@ -66,38 +67,35 @@ _try_load(lua_State *L, const char * path, int pathlen, const char * name) {
exit(1);
}
tmp[namelen+pathlen-1] = '\0';
FILE *f = fopen(tmp,"rb");
if (f == NULL) {
return -1;
} else {
fclose(f);
int r = luaL_loadfile(L,tmp);
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;
}
// luacode_loadfile is the same with luaL_loadfile except cache.
int r = luacode_loadfile(L,tmp);
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_getfield(L,-1,"path");
luaL_Buffer b;
luaL_buffinit(L, &b);
luaL_addlstring(&b, tmp, i+1);
luaL_addstring(&b, "?.lua;");
luaL_addvalue(&b);
luaL_pushresult(&b);
lua_setfield(L,-2,"path");
lua_pop(L,1);
}
if (i<0) {
return 0;
}
return 1;
lua_getglobal(L,"package");
lua_getfield(L,-1,"path");
luaL_Buffer b;
luaL_buffinit(L, &b);
luaL_addlstring(&b, tmp, i+1);
luaL_addstring(&b, "?.lua;");
luaL_addvalue(&b);
luaL_pushresult(&b);
lua_setfield(L,-2,"path");
lua_pop(L,1);
return 0;
} else if (r == LUA_ERRFILE) {
return -1;
}
return 1;
}
static int
@@ -168,6 +166,8 @@ _init(struct snlua *l, struct skynet_context *ctx, const char * args) {
luaL_openlibs(L);
lua_pushlightuserdata(L, l);
lua_setfield(L, LUA_REGISTRYINDEX, "skynet_lua");
luaL_requiref(L, "skynet.codecache", luacode_lib , 0);
lua_pop(L,1);
lua_gc(L, LUA_GCRESTART, 0);
char tmp[strlen(args)+1];