Test commit

Signed-off-by: Jeny Sadadia <[email protected]>

Author: Jeny Sadadia

src/job_retry.py

@@ -80,12 +80,14 @@ def _run(self, sub_id):
                 event_data["jobfilter"] = [node["name"]]
                 event_data["platform_filter"] = [node["data"].get("platform")]
                 event_data["retry_counter"] = retry_counter + 1
+                event_data["debug"] = {"retry_by": str(node["id"])}
                 self.log.debug(f"{node['id']}:Event data retry_counter: {event_data['retry_counter']}")
                 # Added a flag `is_retry` to signal a job retry event
                 event_data["is_retry"] = True
                 event = {'data': event_data}
-                self._api_helper.api.send_event('node', event)
-                self.log.info(f"Job retry for node {node['id']} submitted")
+                self._api_helper.api.send_event('retry', event)
+                self.log.info(f"Job retry for node {node['id']} submitted. Parent node: {event_data['id']}")
+                self.log.debug(f"Event:{event}")
             else:
                 self.log.error(f"Not able to retry the job as parent kind is unknown: {node['id']}")
         return True

src/scheduler.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: LGPL-2.1-or-later
 #
-# Copyright (C) 2021, 2022, 2023 Collabora Limited
+# Copyright (C) 2021-2025 Collabora Limited
 # Author: Guillaume Tucker <[email protected]>
 # Author: Jeny Sadadia <[email protected]>
 
@@ -105,11 +105,17 @@ def _cleanup_paths(self):
         # ToDo: if stat != 0 then report error to API?
 
     def _setup(self, args):
-        return self._api.subscribe('node')
-
-    def _stop(self, sub_id):
-        if sub_id:
-            self._api_helper.unsubscribe_filters(sub_id)
+        # return self._api.subscribe('node')
+        node_sub_id = self._api.subscribe('node')
+        self.log.debug(f"Node channel sub id: {node_sub_id}")
+        retry_sub_id = self._api.subscribe('retry')
+        self.log.debug(f"Retry channel sub id: {retry_sub_id}")
+        return [node_sub_id, retry_sub_id]
+
+    def _stop(self, sub_ids):
+        for sub_id in sub_ids:
+            if sub_id:
+                self._api_helper.unsubscribe_filters(sub_id)
         self._cleanup_paths()
 
     def backup_cleanup(self):
@@ -149,6 +155,7 @@ def _run_job(self, job_config, runtime, platform, input_node, retry_counter):
             node = self._api_helper.create_job_node(job_config,
                                                     input_node,
                                                     runtime, platform, retry_counter)
+            self.log.debug(f"Job node created: {node['id']}. Parent: f{node['parent']}")
         except KeyError as e:
             self.log.error(' '.join([
                 input_node['id'],
@@ -371,7 +378,17 @@ def _verify_architecture_filter(self, job, node):
             return False
         return True
 
-    def _run(self, sub_id):
+    def _run(self, sub_ids):
+        threads = []
+        for sub_id in sub_ids:
+            thread = threading.Thread(target=self._run_scheduler, args=(sub_id,))
+            threads.append(thread)
+            thread.start()
+
+        for thread in threads:
+            thread.join()
+
+    def _run_scheduler(self, sub_id):
         self.log.info("Listening for available checkout events")
         self.log.info("Press Ctrl-C to stop.")
         subscribe_retries = 0
@@ -381,25 +398,29 @@ def _run(self, sub_id):
             event = None
             try:
                 event = self._api_helper.receive_event_data(sub_id, block=False)
+                if not event:
+                    # If we received a keep-alive event, just continue
+                    continue
             except Exception as e:
                 self.log.error(f"Error receiving event: {e}, re-subscribing in 10 seconds")
-                time.sleep(10)
-                sub_id = self._api.subscribe('node')
-                subscribe_retries += 1
-                if subscribe_retries > 3:
-                    self.log.error("Failed to re-subscribe to node events")
-                    return False
-                continue
-            if not event:
-                # If we received a keep-alive event, just continue
+                # time.sleep(10)
+                # sub_id = self._api.subscribe('node')
+                # subscribe_retries += 1
+                # if subscribe_retries > 3:
+                #     self.log.error("Failed to re-subscribe to node events")
+                #     return False
                 continue
-            subscribe_retries = 0
+            # subscribe_retries = 0
+            self.log.debug(f"Event received: {sub_id}:{event['id']}:{event.get('debug')}:{event.get('retry_counter')}")
             for job, runtime, platform, rules in self._sched.get_schedule(event):
                 input_node = self._api.node.get(event['id'])
                 jobfilter = event.get('jobfilter')
                 # Add to node data the jobfilter if it exists in event
                 if jobfilter and isinstance(jobfilter, list):
                     input_node['jobfilter'] = jobfilter
+                platform_filter = event.get('platform_filter')
+                if platform_filter and isinstance(platform_filter, list):
+                    input_node['platform_filter'] = platform_filter
                 # we cannot use rules, as we need to have info about job too
                 if job.params.get('frequency', None):
                     if not self._verify_frequency(job, input_node, platform):
@@ -448,5 +469,11 @@ def __call__(self, configs, args):
     opts = parse_opts('scheduler', globals())
     yaml_configs = opts.get_yaml_configs() or 'config'
     configs = kernelci.config.load(yaml_configs)
+    # sub_ids = sch.setup(opts)
+    # threads = []
+    # for sub_id in sub_ids:
+    #     thread = threading.Thread(target=sch._run_scheduler, args=(sub_id,))
+    #     threads.append(thread)
+    #     thread.start()
     status = opts.command(configs, opts)
     sys.exit(0 if status is True else 1)