From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Sat, 27 May 2023 14:36:19 +0000 (+0100)
Subject: [Minor] Propagate on_load via Lua wrapper
X-Git-Tag: 3.6~95
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=a3745962855863cb180d7a5b6cf86ad46a7570b8;p=rspamd.git

[Minor] Propagate on_load via Lua wrapper
---

diff --git a/lualib/lua_maps.lua b/lualib/lua_maps.lua
index e3ce2af20..d65481177 100644
--- a/lualib/lua_maps.lua
+++ b/lualib/lua_maps.lua
@@ -231,6 +231,12 @@ local function rspamd_map_add_from_ucl(opt, mtype, description, callback)
       end
 
       return nil
+    end,
+    foreach = function(t, cb)
+      return t.__data:foreach(cb)
+    end,
+    on_load = function(t, cb)
+      t.__data:on_load(cb)
     end
   }
   local ret_mt = {
@@ -414,6 +420,20 @@ local function rspamd_map_add_from_ucl(opt, mtype, description, callback)
 
               return nil
             end
+            ret.foreach = function(_, func)
+              for k,v in pairs(ret.__data) do
+                if not func(k, v) then
+                  return false
+                end
+              end
+
+              return true
+            end
+            ret.on_load = function(_, cb)
+              rspamd_config:add_on_load(function(_, _, _)
+                cb()
+              end)
+            end
 
             maps_cache[cache_key] = ret
             maybe_register_selector()