diff options
author | Vsevolod Stakhov <vsevolod@rspamd.com> | 2024-01-17 14:48:24 +0000 |
---|---|---|
committer | Vsevolod Stakhov <vsevolod@rspamd.com> | 2024-01-17 14:48:24 +0000 |
commit | c2fd943dc8454742cd8be85c05e20adaf4719a2d (patch) | |
tree | fb16d097374ac9e748b8f64049946cd798074d88 /lualib/redis_scripts/bayes_cache_check.lua | |
parent | 903b60db99a502377bcb4e7f035efcdb801f8941 (diff) | |
download | rspamd-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.lua | 20 |
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 |