diff --git a/brozzler/cli.py b/brozzler/cli.py index f18ad14..e4fe834 100755 --- a/brozzler/cli.py +++ b/brozzler/cli.py @@ -368,6 +368,7 @@ def brozzle_page(argv=None): metrics_port=args.metrics_port, registry_url=args.registry_url, env=args.env, + worker_id=args.worker_id, ) def on_screenshot(screenshot_jpeg): @@ -717,6 +718,7 @@ def brozzler_worker(argv=None): metrics_port=args.metrics_port, registry_url=args.registry_url, env=args.env, + worker_id=args.worker_id, ) signal.signal(signal.SIGQUIT, dump_state) diff --git a/brozzler/worker.py b/brozzler/worker.py index 807195b..9c40abc 100644 --- a/brozzler/worker.py +++ b/brozzler/worker.py @@ -82,6 +82,7 @@ class BrozzlerWorker: metrics_port=0, registry_url=None, env=None, + worker_id=None, ): self._frontier = frontier self._service_registry = service_registry @@ -97,6 +98,9 @@ class BrozzlerWorker: self._skip_visit_hashtags = skip_visit_hashtags self._skip_youtube_dl = skip_youtube_dl + if worker_id is not None: + self.logger = self.logger.bind(worker_id=worker_id) + # TODO try using importlib.util.find_spec to test for dependency # presence rather than try/except on import. # See https://docs.astral.sh/ruff/rules/unused-import/#example