From 70773eef7084d51619c0ccc497f701ba4d8f0665 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Sun, 15 Dec 2024 09:29:07 +0000 Subject: [CritFix] Allow to reupload scripts when Redis is restarted Issue: #5256 --- lualib/lua_redis.lua | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lualib/lua_redis.lua b/lualib/lua_redis.lua index 7b85f44c4..003fee957 100644 --- a/lualib/lua_redis.lua +++ b/lualib/lua_redis.lua @@ -1140,6 +1140,8 @@ local function script_set_loaded(script) script.loaded = true end + script.servers_ready = nil -- Allow further reload + local wait_table = {} for _, s in ipairs(script.waitq) do table.insert(wait_table, s) -- cgit v1.2.3 From 16957bfccc4ce6b131bff2cb73ce433611233f9c Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Sun, 15 Dec 2024 09:43:04 +0000 Subject: [Minor] Slightly improve logging --- lualib/lua_redis.lua | 45 +++++++++++++++++++++++++-------------------- 1 file changed, 25 insertions(+), 20 deletions(-) diff --git a/lualib/lua_redis.lua b/lualib/lua_redis.lua index 003fee957..2a917b56c 100644 --- a/lualib/lua_redis.lua +++ b/lualib/lua_redis.lua @@ -1463,20 +1463,25 @@ local function exec_redis_script(id, params, callback, keys, args) if not err then callback(err, data) elseif string.match(err, 'NOSCRIPT') then - -- Schedule restart - logger.infox(params.task or rspamd_config, - 'redis script %s is not loaded (NOSCRIPT returned), scheduling reload', - script_description(script)) - script.sha = nil + -- Schedule restart if possible if can_reload then table.insert(script.waitq, do_call) if not script.servers_ready then + logger.infox(params.task or rspamd_config, + 'redis script %s is not loaded (NOSCRIPT returned), scheduling reload', + script_description(script)) + script.sha = nil + script.loaded = nil -- Reload scripts if this has not been initiated yet if params.task then load_script_task(script, params.task) else load_script_taskless(script, rspamd_config, params.ev_base) end + else + logger.infox(params.task or rspamd_config, + 'redis script %s is not ready (NOSCRIPT returned), waiting it to be loaded', + script_description(script)) end else callback(err, data) @@ -1514,24 +1519,24 @@ local function exec_redis_script(id, params, callback, keys, args) end end - if script.loaded then + if script.loaded and script.sha then do_call(true) else - -- Delayed until scripts are loaded - logger.infox(params.task or rspamd_config, 'redis script %s is not loaded, trying to load', - script_description(script)) - if not params.task then - table.insert(script.waitq, do_call) + table.insert(script.waitq, do_call) + if not script.servers_ready then + -- Delayed until scripts are loaded + logger.infox(params.task or rspamd_config, 'redis script %s is not loaded, trying to load', + script_description(script)) + -- Reload scripts if this has not been initiated yet + if params.task then + load_script_task(script, params.task) + else + load_script_taskless(script, rspamd_config, params.ev_base) + end else - -- TODO: fix taskfull requests - table.insert(script.waitq, function() - if script.loaded then - do_call(false) - else - callback('NOSCRIPT', nil) - end - end) - load_script_task(script, params.task, params.is_write) + -- Already loaded, but not yet ready + logger.infox(params.task or rspamd_config, 'redis script %s is not ready, waiting it to be loaded', + script_description(script)) end end -- cgit v1.2.3