Fix dashboard forwarding (#525)

jacobtomlinson · web-flow · commit ba2f3aa6390f · 2022-06-07T17:20:29.000+01:00
diff --git a/dask_kubernetes/classic/kubecluster.py b/dask_kubernetes/classic/kubecluster.py
@@ -31,7 +31,7 @@
 )
 from ..common.networking import (
     get_external_address_for_scheduler_service,
-    port_forward_dashboard,
+    get_scheduler_address,
 )
 
 logger = logging.getLogger(__name__)
@@ -644,9 +644,12 @@ async def _start(self):
 
         await super()._start()
 
-        self.forwarded_dashboard_port = await port_forward_dashboard(
-            self.name, self.namespace
+        dashboard_address = await get_scheduler_address(
+            self.scheduler.service.metadata.name,
+            self.namespace,
+            port_name="http-dashboard",
         )
+        self.forwarded_dashboard_port = dashboard_address.split(":")[-1]
 
     @classmethod
     def from_dict(cls, pod_spec, **kwargs):

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@`
`31`	`31`	`)`
`32`	`32`	`from ..common.networking import (`
`33`	`33`	`get_external_address_for_scheduler_service,`
`34`		`- port_forward_dashboard,`
	`34`	`+ get_scheduler_address,`
`35`	`35`	`)`
`36`	`36`
`37`	`37`	`logger = logging.getLogger(__name__)`
`@@ -644,9 +644,12 @@ async def _start(self):`
`644`	`644`
`645`	`645`	`await super()._start()`
`646`	`646`
`647`		`- self.forwarded_dashboard_port = await port_forward_dashboard(`
`648`		`- self.name, self.namespace`
	`647`	`+ dashboard_address = await get_scheduler_address(`
	`648`	`+ self.scheduler.service.metadata.name,`
	`649`	`+ self.namespace,`
	`650`	`+ port_name="http-dashboard",`
`649`	`651`	`)`
	`652`	`+ self.forwarded_dashboard_port = dashboard_address.split(":")[-1]`
`650`	`653`
`651`	`654`	`@classmethod`
`652`	`655`	`def from_dict(cls, pod_spec, **kwargs):`