Changeset 416

Timestamp:

Nov 24, 2019, 5:40:06 PM (4 years ago)

Author:

g7moreau

Message:

• Add support for parallel and xargs

File:

• trunk/tssh/tssh (modified) (4 diffs)

trunk/tssh/tssh

@@ -21 +21 @@
 
 OPTIONS
-   -w             windows to open (integer, default 16)
+   -w windows     windows to open (integer, default 16)
    -o ssh_option  option to pass to ssh
    -f             fast, no nmap scan to eliminate sleeping computer
    -v             verbose
    -c remote_cmd  launch the remote command on hosts and exit
+   -p             launch in parallel (only with -c) with command parallel or xargs
    -h             help
 
@@ -120 +121 @@
 # get options
 if [ $# -eq 0 ]; then usage; exit 1; fi 
-while getopts "w:o:c:fvh" options
+while getopts "w:o:c:fvph" options
 do
    case ${options} in
@@ -139 +140 @@
             usage
             exit 2
-         fi
+         fi         
+         ;;
+      p)
+         export parallel='yes'
          ;;
       o)
@@ -246 +250 @@
 done
 
+if [ "${parallel}" == 'yes' -a -n "${remote_command}" ]
+then
+   if which parallel > /dev/null
+   then
+      sort -u "/tmp/${base_path}/master++" | parallel -j ${split_number} "ssh ${ssh_option} {} ${remote_command}"
+   else
+      sort -u "/tmp/${base_path}/master++" | xargs -n 1 -P ${split_number} -I{} ssh ${ssh_option} {} "${remote_command}"
+   fi
+
+   # Clean temporary folder
+   [ -d "/tmp/${base_path}" ] && rm -rf "/tmp/${base_path}"
+   
+   exit 0
+fi
+
+
 # split master list in paquet of split_number computer
 sort -u "/tmp/${base_path}/master++" | split -l ${split_number} - /tmp/${base_path}/__splitted_