Skip to content

Commit 036bd02

Browse files
Allow configuring the event emit level (#946)
1 parent 75fe3b9 commit 036bd02

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

dask_kubernetes/kubernetes.yaml

+4
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,10 @@ kubernetes:
3737
worker-allocation:
3838
batch-size: null
3939
delay: null
40+
# Configure whether logs should be emitted as Kubernetes events and minimum level to emit
41+
events:
42+
enabled: true
43+
level: INFO
4044

4145
# Timeout to wait for the scheduler service to be up (in seconds)
4246
# Set it to 0 to wait indefinitely (not recommended)

dask_kubernetes/operator/controller/controller.py

+10
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import asyncio
44
import copy
5+
import logging
56
import time
67
from collections import defaultdict
78
from contextlib import suppress
@@ -295,6 +296,15 @@ async def startup(settings: kopf.OperatorSettings, **__: Any) -> None:
295296
# https://kopf.readthedocs.io/en/latest/configuration/#networking-timeouts
296297
settings.networking.request_timeout = 10
297298

299+
# Configure events
300+
event_enabled = dask.config.get("kubernetes.controller.events.enabled", True)
301+
settings.posting.enabled = event_enabled
302+
event_level = dask.config.get("kubernetes.controller.events.level", "INFO")
303+
if isinstance(event_level, str) and hasattr(logging, event_level):
304+
event_level = getattr(logging, event_level)
305+
if isinstance(event_level, int):
306+
settings.posting.level = event_level
307+
298308

299309
# There may be useful things for us to expose via the liveness probe
300310
# https://kopf.readthedocs.io/en/stable/probing/#probe-handlers

0 commit comments

Comments
 (0)