From e7fe6003a2cfb05cdb9eb3b89237595184461271 Mon Sep 17 00:00:00 2001 From: Andrew Lewis Date: Thu, 24 Nov 2016 11:27:45 +0200 Subject: [Minor] Fix logline in clickhouse plugin --- src/plugins/lua/clickhouse.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src') diff --git a/src/plugins/lua/clickhouse.lua b/src/plugins/lua/clickhouse.lua index 0d6420195..aa39b47da 100644 --- a/src/plugins/lua/clickhouse.lua +++ b/src/plugins/lua/clickhouse.lua @@ -177,7 +177,7 @@ end local function clickhouse_send_data(task) local function http_cb(err_message, code, _, _) if code ~= 200 or err_message then - rspamd_logger.errx(task, "cannot send data to clickhouse server %s: %d:%s", + rspamd_logger.errx(task, "cannot send data to clickhouse server %s: %s:%s", settings['server'], code, err_message) else rspamd_logger.infox(task, "sent %s rows to clickhouse server %s", -- cgit v1.2.3 From aed5dec62bbc44e173cd2de62297a44dad8fb126 Mon Sep 17 00:00:00 2001 From: Andrew Lewis Date: Thu, 24 Nov 2016 11:58:36 +0200 Subject: [Feature] Clickhouse: support different masks for IPv4/IPv6 --- src/plugins/lua/clickhouse.lua | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src') diff --git a/src/plugins/lua/clickhouse.lua b/src/plugins/lua/clickhouse.lua index aa39b47da..045e4a175 100644 --- a/src/plugins/lua/clickhouse.lua +++ b/src/plugins/lua/clickhouse.lua @@ -40,6 +40,7 @@ local settings = { attachments_table = 'rspamd_attachments', urls_table = 'rspamd_urls', ipmask = 19, + ipmask6 = 48, full_urls = false, from_tables = nil } @@ -288,7 +289,12 @@ local function clickhouse_collect(task) local ip_str = 'undefined' local ip = task:get_from_ip() if ip and ip:is_valid() then - local ipnet = ip:apply_mask(settings['ipmask']) + local ipnet + if ip:get_version() == 4 then + ipnet = ip:apply_mask(settings['ipmask']) + else + ipnet = ip:apply_mask(settings['ipmask6']) + end ip_str = ipnet:to_string() end -- cgit v1.2.3