feat(exporter) export per-consumer status

README.md

@@ -112,6 +112,9 @@ kong_bandwidth{type="ingress",service="google"} 254
 # HELP kong_datastore_reachable Datastore reachable from Kong, 0 is unreachable
 # TYPE kong_datastore_reachable gauge
 kong_datastore_reachable 1
+# HELP kong_http_consumer_status HTTP status codes for customer per service/route in Kong
+# TYPE kong_http_consumer_status counter
+kong_http_consumer_status{service="upstream",route="default",code="200",consumer="consumer1"} 5185
 # HELP kong_http_status HTTP status codes per service in Kong
 # TYPE kong_http_status counter
 kong_http_status{code="301",service="google"} 2

kong/plugins/prometheus/exporter.lua

@@ -79,6 +79,10 @@ local function init()
                                          "Total bandwidth in bytes " ..
                                          "consumed per service/route in Kong",
                                          {"service", "route", "type"})
+  metrics.consumer_status = prometheus:counter("http_consumer_status",
+                                          "HTTP status codes for customer per service/route in Kong",
+                                          {"service", "route", "code", "consumer"})
+
   if enterprise then
     enterprise.init(prometheus)
   end
@@ -92,6 +96,7 @@ end
 -- Since in the prometheus library we create a new table for each diverged label
 -- so putting the "more dynamic" label at the end will save us some memory
 local labels_table = {0, 0, 0}
+local labels_table4 = {0, 0, 0, 0}
 local upstream_target_addr_health_table = {
   { value = 0, labels = { 0, 0, 0, "healthchecks_off" } },
   { value = 0, labels = { 0, 0, 0, "healthy" } },
@@ -113,7 +118,7 @@ end
 local log
 
 if ngx.config.subsystem == "http" then
-  function log(message)
+  function log(message, serialized)
     if not metrics then
       kong.log.err("prometheus: can not log metrics because of an initialization "
               .. "error, please make sure that you've declared "
@@ -168,6 +173,14 @@ if ngx.config.subsystem == "http" then
       labels_table[3] = "kong"
       metrics.latency:observe(kong_proxy_latency, labels_table)
     end
+
+    if serialized.consumer ~= nil then
+      labels_table4[1] = labels_table[1]
+      labels_table4[2] = labels_table[2]
+      labels_table4[3] = message.response.status
+      labels_table4[4] = serialized.consumer
+      metrics.consumer_status:inc(1, labels_table4)
+    end
   end
 
 else

kong/plugins/prometheus/handler.lua

@@ -15,9 +15,15 @@ function PrometheusHandler.init_worker()
 end
 
 
-function PrometheusHandler.log()
+function PrometheusHandler.log(self, conf)
   local message = kong.log.serialize()
-  prometheus.log(message)
+
+  local serialized = {}
+  if conf.per_consumer and message.consumer ~= nil then
+    serialized.consumer = message.consumer.username
+  end
+
+  prometheus.log(message, serialized)
 end
 
 

kong/plugins/prometheus/schema.lua

@@ -12,7 +12,9 @@ return {
   fields = {
     { config = {
         type = "record",
-        fields = {},
+        fields = {
+          { per_consumer = { type = "boolean", default = false }, },
+        },
         custom_validator = validate_shared_dict,
     }, },
   },