From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Mon, 30 Apr 2018 16:50:32 +0000 (+0100)
Subject: [Feature] Implement stop symbols for Clickhouse collection
X-Git-Tag: 1.7.4~8
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=2cf3ff82308ecdc16a70571bf97aa947262be602;p=rspamd.git

[Feature] Implement stop symbols for Clickhouse collection
---

diff --git a/src/plugins/lua/clickhouse.lua b/src/plugins/lua/clickhouse.lua
index edfd3bc0d..82762edfe 100644
--- a/src/plugins/lua/clickhouse.lua
+++ b/src/plugins/lua/clickhouse.lua
@@ -48,6 +48,7 @@ local settings = {
   dkim_reject_symbols = {'R_DKIM_REJECT'},
   dmarc_allow_symbols = {'DMARC_POLICY_ALLOW'},
   dmarc_reject_symbols = {'DMARC_POLICY_REJECT', 'DMARC_POLICY_QUARANTINE'},
+  stop_symbols = {},
   table = 'rspamd',
   attachments_table = 'rspamd_attachments',
   urls_table = 'rspamd_urls',
@@ -408,6 +409,14 @@ end
 
 local function clickhouse_collect(task)
   if not settings.allow_local and rspamd_lua_utils.is_rspamc_or_controller(task) then return end
+
+  for _,sym in ipairs(settings.stop_symbols) do
+    if task:has_symbol(sym) then
+      rspamd_logger.debugm(N, task, 'skip collection as symbol %s has fired', sym)
+      return
+    end
+  end
+
   local from_domain = ''
   local from_user = ''
   if task:has_from('smtp') then