-
Notifications
You must be signed in to change notification settings - Fork 4.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(APIAnalytics) ALF serializer skeleton + tests
- almost complete ALF serialization - basic data push to apianalytics - entries can be queued in the serializer, and flushed anytime - serializer can output itself in JSON, being a valid ALF object - more tests, with fixtures - introduce serializers for logging plugins Only a temporary solution before each serializer to be a module of its own (maybe?) and each logging plugin to be independent from Kong, and require serializers as dependencies. Following the discussion in #86
- Loading branch information
1 parent
f77bd4c
commit ddd82ea
Showing
18 changed files
with
413 additions
and
56 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,58 @@ | ||
| local http = require "socket.http" | ||
| local ltn12 = require "ltn12" | ||
| local BasePlugin = require "kong.plugins.base_plugin" | ||
| local ALFSerializer = require "kong.plugins.log_serializers.alf" | ||
|
|
||
| local http_client = require "kong.tools.http_client" | ||
|
|
||
| --local SERVER_URL = "http://socket.apianalytics.com/" | ||
| local SERVER_URL = "http://localhost:58000/alf_1.0.0" | ||
|
|
||
| local APIAnalyticsHandler = BasePlugin:extend() | ||
|
|
||
| function APIAnalyticsHandler:new() | ||
| APIAnalyticsHandler.super.new(self, "apianalytics") | ||
| end | ||
|
|
||
| function APIAnalyticsHandler:access(conf) | ||
| APIAnalyticsHandler.super.access(self) | ||
|
|
||
| ngx.req.read_body() | ||
| ngx.ctx.req_body = ngx.req.get_body_data() | ||
| ngx.ctx.res_body = "" | ||
| end | ||
|
|
||
| function APIAnalyticsHandler:body_filter(conf) | ||
| APIAnalyticsHandler.super.body_filter(self) | ||
|
|
||
| -- concatenate response chunks for response.content.text | ||
| local chunk = ngx.arg[1] | ||
| ngx.ctx.res_body = ngx.ctx.res_body..chunk | ||
| end | ||
|
|
||
| function APIAnalyticsHandler:log(conf) | ||
| APIAnalyticsHandler.super.log(self) | ||
|
|
||
| ALFSerializer:add_entry(ngx) | ||
|
|
||
| -- if queue is full | ||
| local message = ALFSerializer:to_json_string("54d2b98ee0d5076065fd6f93") | ||
| print("MESSAGE: "..message) | ||
|
|
||
| -- TODO: use the cosocket API | ||
| local response, status, headers = http_client.post(SERVER_URL, message, | ||
| { | ||
| ["content-length"] = string.len(message), | ||
| ["content-type"] = "application/json" | ||
| }) | ||
|
|
||
| print("STATUS: "..status) | ||
| if status ~= 200 then | ||
| ngx.log(ngx.ERR, "Could not send entry to "..SERVER_URL) | ||
| print("RESPONSE IS: "..response) | ||
| end | ||
|
|
||
| -- todo: flush | ||
| end | ||
|
|
||
| return APIAnalyticsHandler |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| return {} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,136 @@ | ||
| -- ALF serializer module. | ||
| -- ALF is the format supported by API Analytics (http://apianalytics.com) | ||
| -- | ||
| -- - ALF specifications: https://github.com/Mashape/api-log-format | ||
| -- - Nginx lua module documentation: http://wiki.nginx.org/HttpLuaModule | ||
| -- - ngx_http_core_module: http://wiki.nginx.org/HttpCoreModule#.24http_HEADER | ||
|
|
||
| local EMPTY_ARRAY_PLACEHOLDER = "__empty_array_placeholder__" | ||
|
|
||
| local alf_mt = {} | ||
| alf_mt.__index = alf_mt | ||
|
|
||
| local ALF = { | ||
| version = "1.0.0", | ||
| serviceToken = "", | ||
| har = { | ||
| log = { | ||
| version = "1.2", | ||
| creator = { | ||
| name = "kong-api-analytics-plugin", | ||
| version = "0.1" | ||
| }, | ||
| entries = {} | ||
| } | ||
| } | ||
| } | ||
|
|
||
| -- Transform a key/value lua table into an array of elements with `name`, `value` | ||
| -- Since Lua won't recognize {} as an empty array but an empty object, we need to force it | ||
| -- to be an array, hence we will do "[__empty_array_placeholder__]". | ||
| -- Then once the ALF will be stringified, we will remove the placeholder so the only left element will be "[]". | ||
| -- @param hash key/value dictionary to serialize | ||
| -- @return an array, or nil | ||
| local function dic_to_array(hash) | ||
| local arr = {} | ||
| for k, v in pairs(hash) do | ||
| table.insert(arr, { name = k, value = v }) | ||
| end | ||
|
|
||
| if #arr > 0 then | ||
| return arr | ||
| else | ||
| return {EMPTY_ARRAY_PLACEHOLDER} | ||
| end | ||
| end | ||
|
|
||
| -- Serialize into one ALF entry | ||
| -- For performance reasons, it tries to use the NGINX Lua API | ||
| -- instead of ngx_http_core_module when possible. | ||
| function alf_mt:serialize_entry(ngx) | ||
| -- Extracted data | ||
| local req_headers = ngx.req.get_headers() | ||
| local res_headers = ngx.resp.get_headers() | ||
|
|
||
| local req_body = ngx.ctx.req_body | ||
| local res_body = ngx.ctx.res_body | ||
|
|
||
| -- ALF format | ||
| local alf_req_mimeType = req_headers["Content-Type"] and req_headers["Content-Type"] or "application/octet-stream" | ||
| local alf_res_mimeType = res_headers["Content-Type"] and res_headers["Content-Type"] or "application/octet-stream" | ||
| local alf_req_bodySize = req_headers["Content-Length"] and req_headers["Content-Length"] or 0 | ||
| local alf_res_bodySize = res_headers["Content-Length"] and res_headers["Content-Length"] or 0 | ||
|
|
||
| return { | ||
| startedDateTime = os.date("!%Y-%m-%dT%TZ", ngx.req.start_time()), | ||
| clientIPAddress = ngx.var.remote_addr, | ||
| time = 3, | ||
| -- REQUEST | ||
| request = { | ||
| method = ngx.req.get_method(), | ||
| url = ngx.var.scheme.."://"..ngx.var.host..ngx.var.uri, | ||
| httpVersion = "HTTP/"..ngx.req.http_version(), | ||
| queryString = dic_to_array(ngx.req.get_uri_args()), | ||
| headers = dic_to_array(req_headers), | ||
| headersSize = 10, | ||
| cookies = {EMPTY_ARRAY_PLACEHOLDER}, | ||
| bodySize = tonumber(alf_req_bodySize), | ||
| content = { | ||
| size = tonumber(ngx.var.request_length), | ||
| mimeType = alf_req_mimeType, | ||
| text = req_body and req_body or "" | ||
| } | ||
| }, | ||
| -- RESPONSE | ||
| response = { | ||
| status = ngx.status, | ||
| statusText = "", | ||
| httpVersion = "", | ||
| headers = dic_to_array(res_headers), | ||
| headersSize = 10, | ||
| cookies = {EMPTY_ARRAY_PLACEHOLDER}, | ||
| bodySize = tonumber(alf_res_bodySize), | ||
| redirectURL = "", | ||
| content = { | ||
| size = tonumber(ngx.var.bytes_sent), | ||
| mimeType = alf_res_mimeType, | ||
| text = res_body and res_body or "" | ||
| } | ||
| }, | ||
| cache = {}, | ||
| -- TIMINGS | ||
| timings = { | ||
| send = 1, | ||
| wait = 1, | ||
| receive = 1, | ||
| blocked = 0, | ||
| connect = 0, | ||
| dns = 0, | ||
| ssl = 0 | ||
| } | ||
| } -- end of entry | ||
| end | ||
|
|
||
| function alf_mt:add_entry(ngx) | ||
| table.insert(self.har.log.entries, self:serialize_entry(ngx)) | ||
| end | ||
|
|
||
| function alf_mt:to_json_string(token) | ||
| if not token then | ||
| error("API Analytics serviceToken required", 2) | ||
| end | ||
|
|
||
| local cjson = require "cjson" | ||
|
|
||
| -- inject token | ||
| self.serviceToken = token | ||
|
|
||
| local str = cjson.encode(self) | ||
| return str:gsub("\""..EMPTY_ARRAY_PLACEHOLDER.."\"", ""):gsub("\\/", "/") | ||
| end | ||
|
|
||
| function alf_mt:flush_entries() | ||
| self.har.log.entries = {} | ||
| end | ||
|
|
||
| return setmetatable(ALF, alf_mt) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| local _M = {} | ||
|
|
||
| function _M.serialize(ngx) | ||
| return { | ||
| uri = ngx.var.request_uri, | ||
| request_uri = ngx.var.scheme.."://"..ngx.var.host..":"..ngx.var.server_port..ngx.var.request_uri, | ||
| querystring = ngx.req.get_uri_args(), -- parameters, as a table | ||
| method = ngx.req.get_method(), | ||
| headers = ngx.req.get_headers(), | ||
| size = ngx.var.request_length | ||
| }, | ||
| response = { | ||
| status = ngx.status, | ||
| headers = ngx.resp.get_headers(), | ||
| size = ngx.var.bytes_sent | ||
| }, | ||
| authenticated_entity = ngx.ctx.authenticated_entity, | ||
| api = ngx.ctx.api, | ||
| client_ip = ngx.var.remote_addr, | ||
| started_at = ngx.req.start_time() * 1000 | ||
| } | ||
| end | ||
|
|
||
| return _M |
Oops, something went wrong.