Skip to content

Commit 2df0875

Browse files
authored
Data collection group registration hooks (#483)
Add ability to inject extra functionality after registering data collection groups via entry points
1 parent 577044f commit 2df0875

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/murfey/server/__init__.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2557,6 +2557,16 @@ def feedback_callback(header: dict, message: dict) -> None:
25572557
message.get("tag"): dcgid
25582558
}
25592559
_transport_object.transport.ack(header)
2560+
if dcg_hooks := entry_points().select(
2561+
group="murfey.hooks", name="data_collection_group"
2562+
):
2563+
try:
2564+
for hook in dcg_hooks:
2565+
hook(dcgid, session_id=message["session_id"])
2566+
except Exception:
2567+
logger.error(
2568+
"Call to data collection group hook failed", exc_info=True
2569+
)
25602570
return None
25612571
elif message["register"] == "atlas_update":
25622572
if _transport_object:

0 commit comments

Comments
 (0)