task subgraphs selector

Author: JinZhou5042

app.py

@@ -1,5 +1,4 @@
 from routes.runtime_template import runtime_template_bp
-from routes.subgraphs import subgraphs_bp
 from routes.worker_storage_consumption import worker_storage_consumption_bp
 from routes.file_sizes import file_sizes_bp
 from routes.file_concurrent_replicas import file_concurrent_replicas_bp
@@ -81,7 +80,6 @@ def log_response_info(response):
 app.register_blueprint(file_retention_time_bp)
 
 # subgraphs
-app.register_blueprint(subgraphs_bp)
 app.register_blueprint(task_subgraphs_bp)
 
 # runtime template

routes/subgraphs.py

@@ -113,6 +113,22 @@ def plot_file2task_edge(dot, file, task):
         data['num_task_tries'] = len(task_tries)
         data['subgraph_svg_content'] = open(svg_file_path, 'r').read()
 
+        # legend: list of {'id': str(idx), 'label': ..., 'color': ..., 'checked': bool}, sorted by idx
+        data['legend'] = [
+            {
+                'id': str(idx),
+                'label': f"Subgraph {idx} ({count} task{'s' if count != 1 else ''})",
+                'color': '',
+                'checked': False
+            }
+            for idx, count in sorted(
+                [(k, len(v)) for k, v in runtime_state.subgraphs.items()],
+                key=lambda x: x[0]
+            )
+        ]
+        # check the current subgraph (subgraph_id is 1-based)
+        data['legend'][subgraph_id - 1]['checked'] = True
+
         return jsonify(data)
     except Exception as e:
         runtime_state.logger.error(f'Error in get_task_subgraphs: {e}')

routes/task_subgraphs.py

@@ -394,6 +394,8 @@ table.dataTable thead th {
     cursor: pointer;
     user-select: none;
     transition: opacity 0.2s;
+    text-overflow: ellipsis;
+    white-space: nowrap;
     color: #333;
 }
 

static/css/report.css

@@ -9,6 +9,7 @@ async function fetchAllModulesData(folder) {
         const tasks = moduleConfigs.map(({ id }) => {
             const module = moduleObjects[id];
             return (async () => {
+                module.reset();
                 await module.fetchData(folder);
                 module.initLegend();
                 module.initResetButton();