Browse Source

Merge pull request #1172 from fatalbanana/f

Clickhouse plugin
pull/1174/head
Vsevolod Stakhov 9 years ago
committed by GitHub
parent
commit
30d8bbafc5
  1. 10
      src/plugins/lua/clickhouse.lua

10
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
}
@ -177,7 +178,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",
@ -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

Loading…
Cancel
Save