Add function to run shell command with timeout

common/lib/idds/common/utils.py

@@ -1101,3 +1101,44 @@ def get_process_thread_info():
     thread_id = hb_thread.ident
     thread_name = hb_thread.name
     return hostname, pid, thread_id, thread_name
+
+
+def run_command_with_timeout(command, timeout=600, stdout=sys.stdout, stderr=sys.stderr):
+    """
+    Run a command and monitor its output. Terminate if no output within timeout.
+    """
+    last_output_time = time.time()
+
+    def monitor_output(stream, output, timeout):
+        nonlocal last_output_time
+        for line in iter(stream.readline, b""):
+            output.buffer.write(line)
+            output.flush()
+            last_output_time = time.time()  # Reset timer on new output
+
+    # Start the process
+    process = subprocess.Popen(command,
+                               preexec_fn=os.setsid,    # setpgrp
+                               stdout=subprocess.PIPE,
+                               stderr=subprocess.PIPE)
+
+    # Start the monitoring thread
+    stdout_thread = threading.Thread(target=monitor_output, args=(process.stdout, stdout, timeout))
+    stderr_thread = threading.Thread(target=monitor_output, args=(process.stderr, stderr, timeout))
+    stdout_thread.start()
+    stderr_thread.start()
+
+    # monitor the output and enforce timeout
+    while process.poll() is None:
+        time_elapsed = time.time() - last_output_time
+        if time_elapsed > timeout:
+            print(f"No output for {time_elapsed} seconds. Terminating process.")
+            kill_all(process)
+            break
+        time.sleep(10)  # Check every second
+
+    # Wait for the process to complete and join the monitoring thread
+    stdout_thread.join()
+    stderr_thread.join()
+    process.wait()
+    return process

main/lib/idds/orm/base/utils.py

@@ -47,7 +47,7 @@ def build_database(echo=True, tests=False):
                         conn.execute(CreateSchema(schema))
             except Exception as e:
                 print('Cannot create schema, please validate manually if schema creation is needed, continuing:', e)
-                print(traceback.format_exc())
+                # print(traceback.format_exc())
 
     models.register_models(engine)
 

monitor/data/conf.js

@@ -1,9 +1,9 @@
 
 var appConfig = {
-    'iddsAPI_request': "https://lxplus961.cern.ch:443/idds/monitor_request/null/null",
-    'iddsAPI_transform': "https://lxplus961.cern.ch:443/idds/monitor_transform/null/null",
-    'iddsAPI_processing': "https://lxplus961.cern.ch:443/idds/monitor_processing/null/null",
-    'iddsAPI_request_detail': "https://lxplus961.cern.ch:443/idds/monitor/null/null/true/false/false",
-    'iddsAPI_transform_detail': "https://lxplus961.cern.ch:443/idds/monitor/null/null/false/true/false",
-    'iddsAPI_processing_detail': "https://lxplus961.cern.ch:443/idds/monitor/null/null/false/false/true"
+    'iddsAPI_request': "https://lxplus947.cern.ch:443/idds/monitor_request/null/null",
+    'iddsAPI_transform': "https://lxplus947.cern.ch:443/idds/monitor_transform/null/null",
+    'iddsAPI_processing': "https://lxplus947.cern.ch:443/idds/monitor_processing/null/null",
+    'iddsAPI_request_detail': "https://lxplus947.cern.ch:443/idds/monitor/null/null/true/false/false",
+    'iddsAPI_transform_detail': "https://lxplus947.cern.ch:443/idds/monitor/null/null/false/true/false",
+    'iddsAPI_processing_detail': "https://lxplus947.cern.ch:443/idds/monitor/null/null/false/false/true"
 }