feat(pdk): dynamic control upstream tls when kong.service.request.set…

…_scheme was called
Kong · Dec 20, 2024 · 65d9a31 · 65d9a31
1 parent 198b6ce
commit 65d9a31
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 1 deletion.
diff --git a/kong/pdk/service/request.lua b/kong/pdk/service/request.lua
@@ -6,7 +6,7 @@ local cjson = require "cjson.safe"
 local buffer = require "string.buffer"
 local checks = require "kong.pdk.private.checks"
 local phase_checker = require "kong.pdk.private.phases"
-
+local balancer = require "ngx.balancer"
 
 local ngx = ngx
 local ngx_var = ngx.var
@@ -112,6 +112,15 @@ local function new(self)
       error("invalid scheme: " .. scheme, 2)
     end
 
+    if ngx.get_phase() == "balancer" then
+      if scheme == "https" then
+        kong.service.request.enable_tls()
+      end
+      if scheme == "http" then
+        kong.service.request.disable_tls()
+      end
+    end
+
     ngx_var.upstream_scheme = scheme
   end
 
@@ -711,6 +720,14 @@ local function new(self)
 
       return disable_proxy_ssl()
     end
+  else
+    request.disable_tls = function()
+      return balancer.set_upstream_tls(false)
+    end
+
+    request.enable_tls = function()
+      return balancer.set_upstream_tls(true)
+    end
   end
 
   return request

diff --git a/t/01-pdk/06-service-request/00-phase_checks.t b/t/01-pdk/06-service-request/00-phase_checks.t
@@ -186,6 +186,30 @@ qq{
                 body_filter   = false,
                 log           = false,
                 admin_api     = "forced false",
+            },{
+                method        = "disable_tls",
+                args          = { },
+                init_worker   = "forced false",
+                certificate   = "pending",
+                rewrite       = "forced false",
+                access        = "forced false",
+                response      = "forced false",
+                header_filter = "forced false",
+                body_filter   = "forced false",
+                log           = "forced false",
+                admin_api     = "forced false",
+            },{
+                method        = "enable_tls",
+                args          = { },
+                init_worker   = "forced false",
+                certificate   = "pending",
+                rewrite       = "forced false",
+                access        = "forced false",
+                response      = "forced false",
+                header_filter = "forced false",
+                body_filter   = "forced false",
+                log           = "forced false",
+                admin_api     = "forced false",
             },
         }