always have pid in command

Author: Avi-Robusta

src/java_toolkit/main.py

@@ -9,15 +9,11 @@
 app = typer.Typer()
 
 
-def run_jdk_cmd(pid: int, cmd_missing_jdk_path:str, add_local_pid: bool, verbose: bool):
+def run_jdk_cmd_on_pid(pid: int, cmd_missing_jdk_path_and_pid: str, verbose: bool):
     with TmpRemotePodMounter(pid, JDK_PATH, LOCAL_MOUNT_PATH, verbose) as jdk_mounter:
-        cmd = cmd_missing_jdk_path
-        if add_local_pid:
-            local_pid = get_nspid(pid, verbose)
-            cmd = cmd.format(jdk_path=jdk_mounter.get_mounted_jdk_dir(),pid=local_pid)
-        else:
-            cmd = cmd.format(jdk_path=jdk_mounter.get_mounted_jdk_dir())
-        output =  run_cmd_in_proc_namespace(pid, cmd, verbose)
+        local_pid = get_nspid(pid, verbose)
+        cmd = cmd_missing_jdk_path_and_pid.format(jdk_path=jdk_mounter.get_mounted_jdk_dir(), pid=local_pid)
+        output = run_cmd_in_proc_namespace(pid, cmd, verbose)
         typer.echo(output)
 
 
@@ -28,12 +24,12 @@ def pod_ps(pod_uid: str):
 
 @app.command()
 def jmap(pid: int, verbose: bool = False):
-    run_jdk_cmd(pid, JMAP_CMD, add_local_pid=True, verbose=verbose)
+    run_jdk_cmd_on_pid(pid, JMAP_CMD, verbose=verbose)
 
 
 @app.command()
 def jstack(pid: int, verbose: bool = False):
-    run_jdk_cmd(pid, JSTACK_CMD, add_local_pid=True, verbose=verbose)
+    run_jdk_cmd_on_pid(pid, JSTACK_CMD, verbose=verbose)
 
 
 if __name__ == "__main__":