mirror of
https://github.com/dokku/dokku.git
synced 2026-02-24 04:00:36 +01:00
6a01abb905
* Document new Procfile feature * Add and update tests * Fix a typo in generate_scale_file (ref #1923)
82 lines
2.7 KiB
Bash
Executable File
82 lines
2.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
|
|
source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
|
|
|
|
print_dokku_scale_file() {
|
|
local APP="$1"; local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
|
|
while read -r line || [[ -n "$line" ]]; do
|
|
dokku_log_info2_quiet "$line"
|
|
done < "$DOKKU_SCALE_FILE"
|
|
}
|
|
|
|
extract_procfile() {
|
|
local APP="$1"
|
|
local IMAGE_TAG="$(get_running_image_tag $APP)"
|
|
local IMAGE="$(get_app_image_name $APP $IMAGE_TAG)"
|
|
local DOKKU_PROCFILE="$DOKKU_ROOT/$APP/DOKKU_PROCFILE"
|
|
verify_app_name "$APP"
|
|
|
|
copy_from_image "$IMAGE" "Procfile" "$DOKKU_PROCFILE" 2>/dev/null || true
|
|
if [[ -f "$DOKKU_PROCFILE" ]]; then
|
|
dokku_log_info1_quiet "App Procfile file found ($DOKKU_PROCFILE)"
|
|
else
|
|
dokku_log_info1_quiet "No Procfile found in app image"
|
|
fi
|
|
}
|
|
|
|
remove_procfile() {
|
|
local APP="$1"; local DOKKU_PROCFILE="$DOKKU_ROOT/$APP/DOKKU_PROCFILE"
|
|
if [[ -f "$DOKKU_PROCFILE" ]]; then
|
|
rm -f "$DOKKU_PROCFILE"
|
|
fi
|
|
}
|
|
|
|
generate_scale_file() {
|
|
local APP="$1"; local IMAGE_TAG="$2"; local IMAGE=$(get_app_image_name $APP $IMAGE_TAG); local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
|
|
local DOKKU_PROCFILE="$DOKKU_ROOT/$APP/DOKKU_PROCFILE"
|
|
verify_app_name "$APP"
|
|
|
|
copy_from_image "$IMAGE" "/app/DOKKU_SCALE" "$DOKKU_ROOT/$APP" 2>/dev/null || true
|
|
if [[ ! -f $DOKKU_SCALE_FILE ]]; then
|
|
dokku_log_info1_quiet "DOKKU_SCALE file not found in app image. Generating one based on Procfile..."
|
|
|
|
TMP_WORK_DIR=$(mktemp -d)
|
|
trap 'rm -rf "$TMP_WORK_DIR" > /dev/null' RETURN
|
|
|
|
if [[ -f $DOKKU_PROCFILE ]]; then
|
|
while read -r line || [[ -n "$line" ]]; do
|
|
if [[ -z "$line" ]] || [[ "$line" == "#"* ]]; then
|
|
continue
|
|
fi
|
|
NAME=${line%%:*}
|
|
NUM_PROCS=0
|
|
[[ "$NAME" == "web" ]] && NUM_PROCS=1
|
|
[[ -n "$NAME" ]] && echo "$NAME=$NUM_PROCS" >> $DOKKU_SCALE_FILE
|
|
done < $DOKKU_PROCFILE
|
|
else
|
|
echo "web=1" >> $DOKKU_SCALE_FILE
|
|
fi
|
|
dokku_log_info1_quiet "New DOKKU_SCALE file generated"
|
|
else
|
|
dokku_log_info1_quiet "DOKKU_SCALE file found ($DOKKU_SCALE_FILE)"
|
|
fi
|
|
print_dokku_scale_file $APP
|
|
}
|
|
|
|
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#*=}
|
|
is_number $PROC_COUNT || dokku_log_fail "ps:scale $PROC_COUNT is not a number"
|
|
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
|
|
}
|