aboutsummaryrefslogtreecommitdiffstats
path: root/lualib/redis_scripts/bayes_cache_check.lua
diff options
context:
space:
mode:
authorVsevolod Stakhov <vsevolod@rspamd.com>2024-01-17 14:48:24 +0000
committerVsevolod Stakhov <vsevolod@rspamd.com>2024-01-17 14:48:24 +0000
commitc2fd943dc8454742cd8be85c05e20adaf4719a2d (patch)
treefb16d097374ac9e748b8f64049946cd798074d88 /lualib/redis_scripts/bayes_cache_check.lua
parent903b60db99a502377bcb4e7f035efcdb801f8941 (diff)
downloadrspamd-c2fd943dc8454742cd8be85c05e20adaf4719a2d.tar.gz
rspamd-c2fd943dc8454742cd8be85c05e20adaf4719a2d.zip
[Project] Final things on redis cache rework
Diffstat (limited to 'lualib/redis_scripts/bayes_cache_check.lua')
-rw-r--r--lualib/redis_scripts/bayes_cache_check.lua20
1 files changed, 20 insertions, 0 deletions
diff --git a/lualib/redis_scripts/bayes_cache_check.lua b/lualib/redis_scripts/bayes_cache_check.lua
index e69de29bb..3bea57d5f 100644
--- a/lualib/redis_scripts/bayes_cache_check.lua
+++ b/lualib/redis_scripts/bayes_cache_check.lua
@@ -0,0 +1,20 @@
+-- Lua script to perform cache checking for bayes classification
+-- This script accepts the following parameters:
+-- key1 - cache id
+-- key2 - configuration table in message pack
+
+local cache_id = KEYS[1]
+local conf = cmsgpack.unpack(KEYS[2])
+cache_id = string.sub(cache_id, 1, conf.cache_prefix_len)
+
+-- Try each prefix that is in Redis
+for i = 0, conf.cache_max_keys do
+ local prefix = conf.cache_prefix .. string.rep("X", i)
+ local have = redis.call('HGET', prefix, cache_id)
+
+ if have then
+ return tonumber(have)
+ end
+end
+
+return nil