Skip to content

Commit c970571

Browse files
committed
Add request_id match logic for radas message receiver
1 parent ae5ec19 commit c970571

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

charon/pkgs/radas_signature_handler.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,10 @@ class RadasReceiver(MessagingHandler):
3737
this value transfers from the cmd flag, should register UmbListener when the client starts
3838
"""
3939

40-
def __init__(self, sign_result_loc: str) -> None:
40+
def __init__(self, sign_result_loc: str, request_id: str) -> None:
4141
super().__init__()
4242
self.sign_result_loc = sign_result_loc
43+
self.request_id = request_id
4344

4445
def on_start(self, event: Event) -> None:
4546
"""
@@ -82,8 +83,19 @@ def _process_message(self, msg: Any) -> None:
8283
msg: The message body received
8384
"""
8485
msg_dict = json.loads(msg)
85-
result_reference_url = msg_dict.get("result_reference")
86+
msg_request_id = msg_dict.get("request_id")
87+
if msg_request_id != self.request_id:
88+
logger.info(
89+
"Message request_id %s does not match the request_id %s from sender, ignoring",
90+
msg_request_id,
91+
self.request_id,
92+
)
93+
return
8694

95+
logger.info(
96+
"Start to process the sign event message, request_id %s is matched", msg_request_id
97+
)
98+
result_reference_url = msg_dict.get("result_reference")
8799
if not result_reference_url:
88100
logger.warning("Not found result_reference in message,ignore.")
89101
return
@@ -98,7 +110,6 @@ def _process_message(self, msg: Any) -> None:
98110
)
99111
logger.info("Number of files pulled: %d, path: %s", len(files), files[0])
100112

101-
102113
def generate_radas_sign(top_level: str, sign_result_loc: str) -> Tuple[List[str], List[str]]:
103114
"""
104115
Generate .asc files based on RADAS sign result json file

0 commit comments

Comments
 (0)