Skip to content

Commit ca3d3c3

Browse files
Add worker dashboard (#282)
* Add worker dashboard xref #280 (comment) test-kubernetes * lint
1 parent 7becded commit ca3d3c3

File tree

2 files changed

+20
-1
lines changed

2 files changed

+20
-1
lines changed

dask-gateway-server/dask_gateway_server/backends/kubernetes/controller.py

+2
Original file line numberDiff line numberDiff line change
@@ -1048,6 +1048,8 @@ def get_worker_command(self, namespace, cluster_name, config):
10481048
str(config.worker_memory_limit),
10491049
"--scheduler-address",
10501050
f"tls://{service_name}.{namespace}:8786",
1051+
"--dashboard-address",
1052+
":8787",
10511053
]
10521054

10531055
def get_env(self, namespace, cluster_name, config):

dask-gateway/dask_gateway/dask_cli.py

+18-1
Original file line numberDiff line numberDiff line change
@@ -565,6 +565,12 @@ async def start_scheduler(
565565
worker_parser.add_argument(
566566
"--scheduler-address", default=None, help="The scheduler address"
567567
)
568+
worker_parser.add_argument(
569+
"--dashboard-address",
570+
type=str,
571+
default=":0",
572+
help="The address the dashboard should listen at. Defaults to `:0`",
573+
)
568574

569575

570576
async def start_worker(
@@ -576,6 +582,8 @@ async def start_worker(
576582
scheduler_address=None,
577583
local_directory=None,
578584
nanny=True,
585+
dashboard=True,
586+
dashboard_address=":0",
579587
):
580588
loop = IOLoop.current()
581589

@@ -592,6 +600,8 @@ async def start_worker(
592600
security=security,
593601
name=worker_name,
594602
local_directory=local_directory,
603+
dashboard=dashboard,
604+
dashboard_address=dashboard_address,
595605
)
596606

597607
if nanny:
@@ -612,6 +622,7 @@ def worker(argv=None):
612622
nthreads = args.nthreads
613623
memory_limit = args.memory_limit
614624
scheduler_address = args.scheduler_address
625+
dashboard_address = args.dashboard_address
615626

616627
gateway = make_gateway_client()
617628
security = make_security()
@@ -623,7 +634,13 @@ def worker(argv=None):
623634

624635
async def run():
625636
worker = await start_worker(
626-
gateway, security, worker_name, nthreads, memory_limit, scheduler_address
637+
gateway,
638+
security,
639+
worker_name,
640+
nthreads,
641+
memory_limit,
642+
scheduler_address,
643+
dashboard_address=dashboard_address,
627644
)
628645
await worker.finished()
629646

0 commit comments

Comments
 (0)