Clean up temp file after the script exits.

Author: matthewdale

etc/profile-test.sh

@@ -2,24 +2,26 @@
 # Convenience wrapper for profiling and tracing Go tests.
 set -eu
 
-# Set the prompt for the select menu.
-PS3="Choose a profile type: "
+# Create a temporary file to store the profile or trace data. Trap the exit
+# signals to clean it up after the script exits (typically via Ctrl-C).
+tmpfile=$(mktemp)
+trap 'rm -f "$tmpfile"' EXIT INT TERM
 
+# Set the prompt and options for the select menu.
+PS3="Choose a profile type: "
 options=("cpu" "mem" "block" "mutex" "trace")
 
 select choice in "${options[@]}"
 do
     case $choice in
         "cpu" | "mem" | "block" | "mutex")
-            tmpfile=$(mktemp)
             echo "Writing $choice profile to $tmpfile"
 
             go test -${choice}profile ${tmpfile} "$@"
             go tool pprof -http=: ${tmpfile}
             break
             ;;
         "trace")
-            tmpfile=$(mktemp)
             echo "Writing trace to $tmpfile"
 
             go test -trace ${tmpfile} "$@"