stream_status to use the same terms with http_requests_total

Kong · May 30, 2022 · b619f75 · b619f75
1 parent 4d0b15c
commit b619f75
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/kong/plugins/prometheus/exporter.lua b/kong/plugins/prometheus/exporter.lua
@@ -287,7 +287,12 @@ else
     labels_table[1] = service_name
     labels_table[2] = route_name
     labels_table[3] = message.session.status
-    labels_table[4] = kong.response.get_source()
+
+    if kong.response.get_source() == "service" then
+      labels_table[4] = "service"
+    else
+      labels_table[4] = "kong"
+    end
 
     latency_labels_table[1] = service_name
     latency_labels_table[2] = route_name

diff --git a/spec/03-plugins/26-prometheus/02-access_spec.lua b/spec/03-plugins/26-prometheus/02-access_spec.lua
@@ -484,7 +484,7 @@ describe("Plugin: prometheus (access) per-consumer metrics", function()
       return body:find('http_requests_total{service="mock-service",route="http-route",code="200",source="service",consumer="alice"} 1', nil, true)
     end)
 
-    assert.matches('http_requests_total{service="mock-service",route="http-route",code="401",source="exit",consumer=""} 1', body, nil, true)
+    assert.matches('http_requests_total{service="mock-service",route="http-route",code="401",source="kong",consumer=""} 1', body, nil, true)
 
     assert.matches('kong_nginx_metric_errors_total 0', body, nil, true)
   end)