github
/
uuWAF
mirror of https://github.com/Safe3/uuWAF
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update kafka-logger.lua

pull/8/head
有安科技 2022-09-21 22:35:20 +08:00 committed by GitHub
parent 918432ed2a
commit b92e7d053a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
plugins/kafka-logger.lua
View File

@ -3,7 +3,6 @@
--- Created by Safe3. --- Created by Safe3.
--- DateTime: 2022/9/21 20:37 --- DateTime: 2022/9/21 20:37
--- ---
local cjson = require("cjson.safe")
local producer = require("resty.kafka.producer") local producer = require("resty.kafka.producer")
local log = require("waf.log") local log = require("waf.log")
@ -26,20 +25,18 @@ function _M.resp_body_filter(waf)
end end
--]] --]]
local function kafka_log(_, broker_list, info) local function kafkaLog(_, brokerList, info)
local kp = producer:new(broker_list, { producer_type = "async" }) local kp = producer:new(brokerList, { producer_type = "async" })
local json = cjson.new()
local key = "key" local key = "key"
local message = json.encode(info) local message = log.encodeJson(info)
local ok, err = kp:send("waf-log", key, message) local ok, err = kp:send("waf-log", key, message)
if not ok then if not ok then
log.errorlog(_M.name, " send err: ", err) log.errLog(_M.name, " send err: ", err)
end end
end end
function _M.log(waf) function _M.log(waf)
local broker_list = { local brokerList = {
{ {
host = "127.0.0.1", host = "127.0.0.1",
port = 9092, port = 9092,
@ -49,12 +46,12 @@ function _M.log(waf)
user = "USERNAME", user = "USERNAME",
password = "PASSWORD", password = "PASSWORD",
}, },
}, }
} }
if waf.msg then if waf.msg then
local country, province, city = log.ip2loc(waf.ip) local country, province, city = log.ip2loc(waf.ip)
local info = { rule_id = waf.rule_id, ip = waf.ip, host = waf.host, url = waf.reqUri, data = waf.msg, country = country, province = province, city = city } local info = { rule_id = waf.rule_id, ip = waf.ip, host = waf.host, url = waf.reqUri, data = waf.msg, country = country, province = province, city = city }
log.broker(kafka_log, broker_list, info) log.broker(kafkaLog, brokerList, info)
end end
end end