42 lines
1.2 KiB
Bash
42 lines
1.2 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
set -e
|
||
|
|
set -o pipefail
|
||
|
|
|
||
|
|
echo $@
|
||
|
|
|
||
|
|
if [ -z "$2" ]; then
|
||
|
|
echo "USAGE: upload.sh SIZE NUMBER [TEMP DIR]"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
size="$1"
|
||
|
|
number="$2"
|
||
|
|
directory="${3:-/tmp}"
|
||
|
|
file="${directory}/${size}.bin"
|
||
|
|
|
||
|
|
openssl rand -out "$file" "$size"
|
||
|
|
|
||
|
|
# Get upload size in bytes
|
||
|
|
upload_size=$(stat -c "%s" "$file")
|
||
|
|
echo "Generated file with size: ${upload_size} bytes."
|
||
|
|
|
||
|
|
# Create uploads
|
||
|
|
for i in $(seq 1 $number); do
|
||
|
|
# Note: I wanted to use the new feature for extracting header values
|
||
|
|
# (https://daniel.haxx.se/blog/2022/03/24/easier-header-picking-with-curl/)
|
||
|
|
# but this is not yet available on the current curl version in Alpine Linux.
|
||
|
|
upload_urls[${i}]="$(curl -X POST -H 'Tus-Resumable: 1.0.0' -H "Upload-Length: ${upload_size}" --fail --silent -i http://tusd:1080/files/ | grep -i ^Location: | cut -d: -f2- | sed 's/^ *\(.*\).*/\1/' | tr -d '\r')"
|
||
|
|
done
|
||
|
|
|
||
|
|
# Perform the uploads in parallel
|
||
|
|
for i in $(seq 1 $number); do
|
||
|
|
curl -X PATCH -H 'Tus-Resumable: 1.0.0' -H 'Upload-Offset: 0' -H 'Content-Type: application/offset+octet-stream' --data-binary "@${file}" "${upload_urls[${i}]}" &
|
||
|
|
pids[${i}]=$!
|
||
|
|
done
|
||
|
|
|
||
|
|
# Wait for all uploads to complete
|
||
|
|
for pid in ${pids[*]}; do
|
||
|
|
wait $pid
|
||
|
|
done
|