nginx_lua_request_monitor/reqstat.lua at master · wangyapu/nginx_lua_request_monitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
--
-- yapu.wang@dianping.com
-- Created by wangyapu on 2016/07/05 15:29.
--

-- 共享内存statistics_dict记录所有监控数据
statistics_dict = ngx.shared.statistics_dict
local stat_util = require("stat_util")

-- query conut
function query_count(identifier)
    local query_count_key = identifier .. "__query_count"
    local new_value, err = statistics_dict:incr(query_count_key, 1)
    if not new_value and err == "not found" then
        statistics_dict:add(query_count_key, 0, 86400)
        stat_util.incr(statistics_dict, query_count_key, 1)
    end
end

-- request time count
function request_time_count(identifier)
    local request_time_count_key = identifier .. "__request_time_count"
    local current_request_time = tonumber(ngx.var.request_time) or 0
    local pre_sum = statistics_dict:get(request_time_count_key) or 0
    local sum = pre_sum + current_request_time
    statistics_dict:set(request_time_count_key, sum, 86400)
end

-- flow count
function flow_count(identifier)
    local flow_sum_key = identifier .. "__bytes_sent_count"
    local current_bytes_sent = tonumber(ngx.var.bytes_sent) or 0
    local pre_sum = statistics_dict:get(flow_sum_key) or 0
    local sum = pre_sum + current_bytes_sent
    statistics_dict:set(flow_sum_key, sum, 86400)
end

-- request 4xx code count
function request_4xx_code_count(identifier)
    local status_code = tonumber(ngx.var.status)
    local request_4xx_count_key = identifier .. "__4xx_code_count"

    if status_code >= 400 and status_code < 500 then
        local new_value, err = statistics_dict:incr(request_4xx_count_key, 1)
        if not new_value and err == "not found" then
            statistics_dict:add(request_4xx_count_key, 0, 86400)
            stat_util.incr(statistics_dict, request_4xx_count_key, 1)
        end
    end
end

-- request 5xx code count
function request_5xx_code_count(identifier)
    local status_code = tonumber(ngx.var.status)
    local request_5xx_count_key = identifier .. "__5xx_code_count"

    if status_code >= 500 then
        local new_value, err = statistics_dict:incr(request_5xx_count_key, 1)
        if not new_value and err == "not found" then
            statistics_dict:add(request_5xx_count_key, 0, 86400)
            stat_util.incr(statistics_dict, request_5xx_count_key, 1)
        end
    end
end

if ngx.var.domain_monitor and string.len(ngx.var.domain_monitor) > 0 then
    local var_prefix = ngx.var.domain_monitor
    query_count(var_prefix)
    request_time_count(var_prefix)
    flow_count(var_prefix)
    request_4xx_code_count(var_prefix)
    request_5xx_code_count(var_prefix)
end