From 94fa4f5e721a9f789d5f8983c9879a70d4a0b399 Mon Sep 17 00:00:00 2001
From: Michael Krayer <michael.krayer@partner.kit.edu>
Date: Tue, 2 Apr 2024 15:32:06 +0200
Subject: [PATCH] Added flag for fail and removed debug directory

---
 prsync | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/prsync b/prsync
index 9a604d0..3ee4105 100755
--- a/prsync
+++ b/prsync
@@ -110,8 +110,7 @@ src=($(echo "$tmp" | awk '{print $2}'))
 totalsize=$(IFS=+; echo "$((${filesize[*]}))")
 
 # Create file lists
-# dir_temp=$(mktemp -dt rsync.XXXXX)
-dir_temp="/scratch/rsync.MCbI5"
+dir_temp=$(mktemp -dt rsync.XXXXX)
 for ((istream=0;istream<num_streams;istream++)); do
     file_temp="${dir_temp}/rsync-stream-${istream}.files"
     echo -n > $file_temp
@@ -131,6 +130,7 @@ function rsync_sigint {
 }
 rsync_pids=()
 timer=$(date +%s%N)
+flag_fail=0
 trap rsync_sigint SIGINT
 for ((istream=0;istream<num_streams;istream++)); do
     file_temp="${dir_temp}/rsync-stream-${istream}.files"
@@ -141,8 +141,14 @@ for ((istream=0;istream<num_streams;istream++)); do
 done
 for pid in ${rsync_pids[@]}; do
     wait $pid 
-    echo "Process ${pid} finished with exit code $?"
+    ec_=$?
+    echo "Process ${pid} finished with exit code $ec_"
+    [[ $ec_ -ne 0 ]] && flag_fail=1
 done
+if [ $flag_fail -ne 0 ]; then
+    echo "Transfer failed."
+    exit 1
+fi
 
 # Evaluate time and print transfer info
 dt_transfer=$(($(date +%s%N) - timer))