software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Added --cleanup option

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-03-25 15:24:59 -07:00
parent 1755dfd114
commit b8930b05a1
1 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
launch-cluster.sh
View File

@@ -43,6 +43,7 @@ MEM_LIMIT_GB="110"
MEM_SWAP_LIMIT_GB="" MEM_SWAP_LIMIT_GB=""
PIDS_LIMIT="4096" PIDS_LIMIT="4096"
SHM_SIZE_GB="64" SHM_SIZE_GB="64"
CLEANUP_MODE="false"
# Function to print usage # Function to print usage
usage() { usage() {
@@ -69,6 +70,7 @@ usage() {
echo " --pids-limit Process limit (default: 4096, only with --non-privileged)" echo " --pids-limit Process limit (default: 4096, only with --non-privileged)"
echo " --shm-size-gb Shared memory size in GB (default: 64, only with --non-privileged)" echo " --shm-size-gb Shared memory size in GB (default: 64, only with --non-privileged)"
echo " --config Path to .env configuration file (default: .env in script directory)" echo " --config Path to .env configuration file (default: .env in script directory)"
echo " --cleanup Remove all *.whl and *.-commit files in wheels directory"
echo " action start | stop | status | exec (Default: start). Not compatible with --launch-script." echo " action start | stop | status | exec (Default: start). Not compatible with --launch-script."
echo " command Command to run (only for 'exec' action). Not compatible with --launch-script." echo " command Command to run (only for 'exec' action). Not compatible with --launch-script."
echo "" echo ""
@@ -129,6 +131,7 @@ while [[ "$#" -gt 0 ]]; do
--pids-limit) PIDS_LIMIT="$2"; shift ;; --pids-limit) PIDS_LIMIT="$2"; shift ;;
--shm-size-gb) SHM_SIZE_GB="$2"; shift ;; --shm-size-gb) SHM_SIZE_GB="$2"; shift ;;
-d) DAEMON_MODE="true" ;; -d) DAEMON_MODE="true" ;;
--cleanup) CLEANUP_MODE="true" ;;
-h|--help) usage ;; -h|--help) usage ;;
--config) CONFIG_FILE="$2"; shift ;; --config) CONFIG_FILE="$2"; shift ;;
start|stop|status) start|stop|status)
@@ -535,6 +538,31 @@ if [[ "$ACTION" == "stop" ]]; then
exit 0 exit 0
fi fi
# Handle 'cleanup' action
if [[ "$CLEANUP_MODE" == "true" ]]; then
WHEELS_DIR="$SCRIPT_DIR/wheels"
echo "Cleaning up wheels directory..."
# Remove all .whl files
if compgen -G "$WHEELS_DIR/*.whl" > /dev/null 2>&1; then
rm -f "$WHEELS_DIR"/*.whl
echo "Removed *.whl files from $WHEELS_DIR"
else
echo "No *.whl files found in $WHEELS_DIR"
fi
# Remove all .-commit files
if compgen -G "$WHEELS_DIR/*.-commit" > /dev/null 2>&1; then
rm -f "$WHEELS_DIR"/*.-commit
echo "Removed *.-commit files from $WHEELS_DIR"
else
echo "No *.-commit files found in $WHEELS_DIR"
fi
echo "Cleanup complete."
exit 0
fi
# Handle 'status' action # Handle 'status' action
if [[ "$ACTION" == "status" ]]; then if [[ "$ACTION" == "status" ]]; then
echo "Checking status..." echo "Checking status..."