Add error handler to async step

This commit is contained in:
ShadowNinja 2013-11-30 12:24:54 -05:00
parent 4594ba6522
commit 4696c59a5f
1 changed files with 11 additions and 41 deletions

View File

@ -29,36 +29,7 @@ int luaopen_marshal(lua_State *L);
#include "log.h" #include "log.h"
#include "filesys.h" #include "filesys.h"
#include "porting.h" #include "porting.h"
#include "common/c_internal.h"
//TODO replace by ShadowNinja version not yet merged to master
static int script_error_handler(lua_State *L) {
lua_getfield(L, LUA_GLOBALSINDEX, "debug");
if (!lua_istable(L, -1)) {
lua_pop(L, 1);
return 1;
}
lua_getfield(L, -1, "traceback");
if (!lua_isfunction(L, -1)) {
lua_pop(L, 2);
return 1;
}
lua_pushvalue(L, 1);
lua_pushinteger(L, 2);
lua_call(L, 2, 1);
return 1;
}
/******************************************************************************/
static void scriptError(const char *fmt, ...)
{
va_list argp;
va_start(argp, fmt);
char buf[10000];
vsnprintf(buf, 10000, fmt, argp);
va_end(argp);
errorstream<<"ERROR: "<<buf;
fprintf(stderr,"ERROR: %s\n",buf);
}
/******************************************************************************/ /******************************************************************************/
AsyncEngine::AsyncEngine() : AsyncEngine::AsyncEngine() :
@ -170,12 +141,13 @@ void AsyncEngine::putJobResult(LuaJobInfo result) {
/******************************************************************************/ /******************************************************************************/
void AsyncEngine::Step(lua_State *L) { void AsyncEngine::Step(lua_State *L) {
lua_pushcfunction(L, script_error_handler);
int errorhandler = lua_gettop(L);
lua_getglobal(L, "engine");
m_ResultQueueMutex.Lock(); m_ResultQueueMutex.Lock();
while(!m_ResultQueue.empty()) { while(!m_ResultQueue.empty()) {
LuaJobInfo jobdone = m_ResultQueue.front(); LuaJobInfo jobdone = m_ResultQueue.front();
m_ResultQueue.erase(m_ResultQueue.begin()); m_ResultQueue.erase(m_ResultQueue.begin());
lua_getglobal(L, "engine");
lua_getfield(L, -1, "async_event_handler"); lua_getfield(L, -1, "async_event_handler");
@ -186,15 +158,14 @@ void AsyncEngine::Step(lua_State *L) {
lua_pushinteger(L, jobdone.JobId); lua_pushinteger(L, jobdone.JobId);
lua_pushlstring(L, jobdone.serializedResult.c_str(), lua_pushlstring(L, jobdone.serializedResult.c_str(),
jobdone.serializedResult.length()); jobdone.serializedResult.length());
if(lua_pcall(L, 2, 0, 0)) { if(lua_pcall(L, 2, 0, errorhandler)) {
scriptError("Async ENGINE step: %s", lua_tostring(L, -1)); script_error(L);
} }
lua_pop(L,1);
} }
m_ResultQueueMutex.Unlock(); m_ResultQueueMutex.Unlock();
lua_pop(L, 2); // Pop engine and error handler
} }
/******************************************************************************/ /******************************************************************************/
@ -355,10 +326,9 @@ void* AsyncWorkerThread::worker_thread_main() {
if (StopRequested()) { continue; } if (StopRequested()) { continue; }
if(lua_pcall(m_LuaStack, 2, 2, errorhandler)) { if(lua_pcall(m_LuaStack, 2, 2, errorhandler)) {
scriptError("Async WORKER thread: %s\n", lua_tostring(m_LuaStack, -1)); script_error(m_LuaStack);
toprocess.serializedResult="ERROR"; toprocess.serializedResult = "ERROR";
} } else {
else {
//fetch result //fetch result
const char *retval = lua_tostring(m_LuaStack, -2); const char *retval = lua_tostring(m_LuaStack, -2);
unsigned int lenght = lua_tointeger(m_LuaStack,-1); unsigned int lenght = lua_tointeger(m_LuaStack,-1);