chore(fix-snapshot-importer-ssh-timeout): Add TCP keep alive and serv…

…er alive interval flags to snapshot_tool SSH command
input-output-hk · Nov 17, 2023 · 6bdff1b · 6bdff1b
1 parent 1a15c2a
commit 6bdff1b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/utilities/ideascale-importer/ideascale_importer/cli/snapshot.py b/utilities/ideascale-importer/ideascale_importer/cli/snapshot.py
@@ -86,7 +86,12 @@ async def inner():
             and ssh_snapshot_tool_path is not None
             and ssh_snapshot_tool_output_dir is not None
         ):
-            ssh_config = SSHConfig(ssh_keyfile, ssh_destination, ssh_snapshot_tool_path, ssh_snapshot_tool_output_dir)
+            ssh_config = SSHConfig(
+                keyfile_path=ssh_keyfile,
+                destination=ssh_destination,
+                snapshot_tool_path=ssh_snapshot_tool_path,
+                snapshot_tool_output_dir=ssh_snapshot_tool_output_dir,
+            )
         else:
             if snapshot_tool_ssh:
                 logger.error(

diff --git a/utilities/ideascale-importer/ideascale_importer/snapshot_importer.py b/utilities/ideascale-importer/ideascale_importer/snapshot_importer.py
@@ -436,6 +436,7 @@ async def _run_snapshot_tool(self):
                     snapshot_tool_cmd = (
                         "ssh"
                         f" -i {self.ssh_config.keyfile_path}"
+                        " -oTCPKeepAlive=no -oServerAliveInterval=20"
                         f" {self.ssh_config.destination}"
                         f" {self.ssh_config.snapshot_tool_path}"
                         f" --db-user {db_user}"