From 39e6b7901bcc3ecf575ce0400448075b6584af33 Mon Sep 17 00:00:00 2001 From: Xiaochen Wang Date: Wed, 7 Aug 2024 16:15:09 +0800 Subject: [PATCH] fix(dns): added /status/dns API to status_listen port --- kong-3.8.0-0.rockspec | 1 + kong/api/routes/dns.lua | 23 ++++++ kong/status/init.lua | 1 + .../05-dns_client_spec.lua} | 75 +++++++++++++++++-- 4 files changed, 93 insertions(+), 7 deletions(-) create mode 100644 kong/api/routes/dns.lua rename spec/02-integration/{04-admin_api/26-dns_client_spec.lua => 08-status_api/05-dns_client_spec.lua} (53%) diff --git a/kong-3.8.0-0.rockspec b/kong-3.8.0-0.rockspec index 3dc070e8684e..eb138b14094a 100644 --- a/kong-3.8.0-0.rockspec +++ b/kong-3.8.0-0.rockspec @@ -169,6 +169,7 @@ build = { ["kong.api.routes.tags"] = "kong/api/routes/tags.lua", ["kong.api.routes.targets"] = "kong/api/routes/targets.lua", ["kong.api.routes.upstreams"] = "kong/api/routes/upstreams.lua", + ["kong.api.routes.dns"] = "kong/api/routes/dns.lua", ["kong.admin_gui"] = "kong/admin_gui/init.lua", ["kong.admin_gui.utils"] = "kong/admin_gui/utils.lua", diff --git a/kong/api/routes/dns.lua b/kong/api/routes/dns.lua new file mode 100644 index 000000000000..37892a74b21a --- /dev/null +++ b/kong/api/routes/dns.lua @@ -0,0 +1,23 @@ +local kong = kong + + +return { + ["/status/dns"] = { + GET = function (self, db, helpers) + + if kong.configuration.legacy_dns_client then + return kong.response.exit(501, { + message = "not implemented with the legacy DNS client" + }) + end + + return kong.response.exit(200, { + worker = { + id = ngx.worker.id() or -1, + count = ngx.worker.count(), + }, + stats = kong.dns.stats(), + }) + end + }, +} diff --git a/kong/status/init.lua b/kong/status/init.lua index ffe7ca2e54cf..f6c7c41e3cd5 100644 --- a/kong/status/init.lua +++ b/kong/status/init.lua @@ -27,6 +27,7 @@ ngx.log(ngx.DEBUG, "Loading Status API endpoints") -- Load core health route api_helpers.attach_routes(app, require "kong.api.routes.health") api_helpers.attach_routes(app, require "kong.status.ready") +api_helpers.attach_routes(app, require "kong.api.routes.dns") if kong.configuration.database == "off" then diff --git a/spec/02-integration/04-admin_api/26-dns_client_spec.lua b/spec/02-integration/08-status_api/05-dns_client_spec.lua similarity index 53% rename from spec/02-integration/04-admin_api/26-dns_client_spec.lua rename to spec/02-integration/08-status_api/05-dns_client_spec.lua index 036671732a8a..8ebeb757e645 100644 --- a/spec/02-integration/04-admin_api/26-dns_client_spec.lua +++ b/spec/02-integration/08-status_api/05-dns_client_spec.lua @@ -1,9 +1,10 @@ local helpers = require "spec.helpers" local cjson = require "cjson" +local tcp_status_port = helpers.get_available_port() for _, strategy in helpers.each_strategy() do - describe("Admin API - DNS client route with [#" .. strategy .. "]" , function() + describe("[#traditional] Status API - DNS client route with [#" .. strategy .. "]" , function() local client lazy_setup(function() @@ -20,11 +21,11 @@ for _, strategy in helpers.each_strategy() do assert(helpers.start_kong({ database = strategy, - nginx_conf = "spec/fixtures/custom_nginx.template", + status_listen = "127.0.0.1:" .. tcp_status_port, legacy_dns_client = "off", })) - client = helpers.admin_client() + client = helpers.http_client("127.0.0.1", tcp_status_port, 20000) end) teardown(function() @@ -65,7 +66,7 @@ for _, strategy in helpers.each_strategy() do end) end) - describe("Admin API - DNS client route with [#" .. strategy .. "]" , function() + describe("[#traditional] Status API - DNS client route with [#" .. strategy .. "]" , function() local client lazy_setup(function() @@ -73,11 +74,11 @@ for _, strategy in helpers.each_strategy() do assert(helpers.start_kong({ database = strategy, - nginx_conf = "spec/fixtures/custom_nginx.template", - legacy_dns_client = true, + status_listen = "127.0.0.1:" .. tcp_status_port, + legacy_dns_client = "on", })) - client = helpers.admin_client() + client = helpers.http_client("127.0.0.1", tcp_status_port, 20000) end) teardown(function() @@ -100,3 +101,63 @@ for _, strategy in helpers.each_strategy() do end) end) end + + +-- hybrid mode + +for _, strategy in helpers.each_strategy() do + + describe("[#hybrid] Status API - DNS client route with [#" .. strategy .. "]" , function() + local client + + lazy_setup(function() + helpers.get_db_utils(strategy) -- runs migrations + + assert(helpers.start_kong({ + role = "control_plane", + cluster_cert = "spec/fixtures/kong_clustering.crt", + cluster_cert_key = "spec/fixtures/kong_clustering.key", + database = strategy, + cluster_listen = "127.0.0.1:9005", + nginx_conf = "spec/fixtures/custom_nginx.template", + legacy_dns_client = "off", + })) + + assert(helpers.start_kong({ + role = "data_plane", + database = "off", + prefix = "servroot2", + cluster_cert = "spec/fixtures/kong_clustering.crt", + cluster_cert_key = "spec/fixtures/kong_clustering.key", + cluster_control_plane = "127.0.0.1:9005", + proxy_listen = "0.0.0.0:9002", + nginx_conf = "spec/fixtures/custom_nginx.template", + status_listen = "127.0.0.1:" .. tcp_status_port, + legacy_dns_client = "off", + })) + + client = helpers.http_client("127.0.0.1", tcp_status_port, 20000) + end) + + teardown(function() + if client then + client:close() + end + helpers.stop_kong("servroot2") + helpers.stop_kong() + end) + + it("/status/dns - status code 200", function () + local res = assert(client:send { + method = "GET", + path = "/status/dns", + headers = { ["Content-Type"] = "application/json" } + }) + + local body = assert.res_status(200 , res) + local json = assert(cjson.decode(body)) + assert(type(json.stats) == "table") + end) + + end) +end