diff options
| author | Andrew Lewis <nerf@judo.za.org> | 2015-04-02 13:00:33 +0200 |
|---|---|---|
| committer | Andrew Lewis <nerf@judo.za.org> | 2015-04-02 13:00:33 +0200 |
| commit | f786fc3053c2a3bc65a8ffe9bb4859f5c904be87 (patch) | |
| tree | 8f726fc757cde9100bb3788c656a89137aaf0ce3 /src | |
| parent | 6f072228e63748c79ef8fbf74dfffd0694a07247 (diff) | |
| download | rspamd-f786fc3053c2a3bc65a8ffe9bb4859f5c904be87.tar.gz rspamd-f786fc3053c2a3bc65a8ffe9bb4859f5c904be87.zip | |
Allow getting message queue ID from Lua
Diffstat (limited to 'src')
| -rw-r--r-- | src/lua/lua_task.c | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c index 96292bee8..c4e595c8b 100644 --- a/src/lua/lua_task.c +++ b/src/lua/lua_task.c @@ -246,6 +246,11 @@ LUA_FUNCTION_DEF (task, get_raw_headers); */ LUA_FUNCTION_DEF (task, get_received_headers); /*** + * @method task:get_queue_id() + * Returns queue ID of the message being processed. + */ +LUA_FUNCTION_DEF (task, get_queue_id); +/*** * @method task:get_resolver() * Returns ready to use rspamd_resolver object suitable for making asynchronous DNS requests. * @return {rspamd_resolver} resolver object associated with the task's session @@ -464,6 +469,7 @@ static const struct luaL_reg tasklib_m[] = { LUA_INTERFACE_DEF (task, get_header_full), LUA_INTERFACE_DEF (task, get_raw_headers), LUA_INTERFACE_DEF (task, get_received_headers), + LUA_INTERFACE_DEF (task, get_queue_id), LUA_INTERFACE_DEF (task, get_resolver), LUA_INTERFACE_DEF (task, inc_dns_req), LUA_INTERFACE_DEF (task, get_recipients), @@ -1237,6 +1243,20 @@ lua_task_get_received_headers (lua_State * L) } static gint +lua_task_get_queue_id (lua_State *L) +{ + struct rspamd_task *task = lua_check_task (L, 1); + + if (task && task->queue_id != NULL && task->queue_id != "undef") { + lua_pushstring (L, task->queue_id); + return 1; + } + + lua_pushnil (L); + return 1; +} + +static gint lua_task_get_resolver (lua_State *L) { struct rspamd_task *task = lua_check_task (L, 1); |