mirror of
https://github.com/dokku/dokku.git
synced 2026-02-24 12:12:08 +01:00
51 lines
1.7 KiB
Bash
Executable File
51 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
|
|
source "$(dirname $0)/../common/functions"
|
|
|
|
generate_scale_file() {
|
|
local APP="$1"; local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
|
|
copy_from_image "$APP" "/app/DOKKU_SCALE" "$DOKKU_ROOT/$APP" 2>/dev/null || dokku_log_info1_quiet "DOKKU_SCALE not found in app image"
|
|
if [[ ! -f $DOKKU_SCALE_FILE ]]; then
|
|
local TMP_WORK_DIR=$(mktemp -d DOKKU_SCALE.XXXXX)
|
|
trap 'rm -rf $TMP_WORK_DIR' EXIT
|
|
|
|
copy_from_image "$APP" /app/Procfile $TMP_WORK_DIR 2>/dev/null || true
|
|
copy_from_image "$APP" /app/.release $TMP_WORK_DIR 2>/dev/null || true
|
|
|
|
if [[ -f $TMP_WORK_DIR/Procfile ]];then
|
|
local PROCFILE="$TMP_WORK_DIR/Procfile"
|
|
elif [[ -f $TMP_WORK_DIR/.release ]];then
|
|
local PROCFILE="$TMP_WORK_DIR/.release"
|
|
else
|
|
echo "default=1" >> $DOKKU_SCALE_FILE
|
|
fi
|
|
|
|
while read line || [ -n "$line" ]
|
|
do
|
|
local PROC_NAME=${line%%:*}
|
|
local PROC_CMD=${line#*:}
|
|
[[ "$PROC_NAME" =~ -.* ]] && continue
|
|
[[ -n "$PROC_CMD" ]] && echo "$PROC_NAME=1" >> $DOKKU_SCALE_FILE
|
|
done < "$PROCFILE"
|
|
|
|
dokku_log_info1_quiet "New DOKKU_SCALE file generated"
|
|
dokku_log_info1_quiet "$(< $DOKKU_SCALE_FILE)"
|
|
fi
|
|
}
|
|
|
|
set_scale() {
|
|
local APP="$1"; local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
|
|
shift 1
|
|
local SCALE_SETTINGS=("$@")
|
|
for procscale in "${SCALE_SETTINGS[@]}"; do
|
|
PROC_NAME=${procscale%%=*}
|
|
PROC_COUNT=${procscale#*=}
|
|
dokku_log_info1_quiet "Scaling $APP:$PROC_NAME to $PROC_COUNT"
|
|
if (egrep -q ^${PROC_NAME}= $DOKKU_SCALE_FILE > /dev/null 2>&1);then
|
|
sed --in-place "s:^${PROC_NAME}=.*:$PROC_NAME=$PROC_COUNT:g" $DOKKU_SCALE_FILE
|
|
else
|
|
echo "$PROC_NAME=$PROC_COUNT" >> $DOKKU_SCALE_FILE
|
|
fi
|
|
done
|
|
}
|