run-daisy-nfsd.sh

#!/bin/bash

set -eu

#
# Usage:  ./run-daisy-nfsd.sh  go run ./cmd/fs-smallfile
#

DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" >/dev/null 2>&1 && pwd)"
# root of repo
cd "$DIR"/..

disk_file=/dev/shm/nfs.img
size_mb=400
cpu_list=""
nfs_mount_path="/mnt/nfs"
# go-journal patch
patch_file=""
extra_args=()
while [[ $# -gt 0 ]]; do
    case "$1" in
    -disk)
        shift
        disk_file="$1"
        shift
        ;;
    -size)
        shift
        size_mb="$1"
        shift
        ;;
    -mount-path)
        shift
        nfs_mount_path="$1"
        extra_args+=("-mount-path" "$nfs_mount_path")
        shift
        ;;
    -patch)
        shift
        patch_file="$1"
        shift
        ;;
    --cpu-list)
        shift
        cpu_list="$1"
        shift
        ;;
    # some argument in -foo=value syntax
    -*=*)
        extra_args+=("$1")
        shift
        ;;
    -*)
        extra_args+=("$1" "$2")
        shift
        shift
        ;;
    *)
        break
        ;;
    esac
done

if [[ -n "$patch_file" && ! -f "$patch_file" ]]; then
    echo "Invalid patch file $patch_file" 1>&2
    exit 1
fi

if [ -n "$patch_file" ]; then
    ./bench/set-gojournal-patch.sh "$patch_file"
fi

if [ -z "$cpu_list" ]; then
    ./bench/start-daisy-nfsd.sh -disk "$disk_file" -size "$size_mb" "${extra_args[@]}" || exit 1
else
    taskset --cpu-list "$cpu_list" ./bench/start-daisy-nfsd.sh -size "$size_mb" -disk "$disk_file" "${extra_args[@]}" || exit 1
fi

function cleanup {
    ./bench/stop-daisy-nfsd.sh "$nfs_mount_path"
    if [ -n "$patch_file" ]; then
        ./bench/set-gojournal-patch.sh --undo
    fi

    # only delete regular files
    if [ -f "$disk_file" ]; then
        rm -f "$disk_file"
    fi
}
trap cleanup EXIT

echo "# daisy-nfsd -disk $disk_file ${extra_args[*]}" 1>&2
echo "run $*" 1>&2
"$@"