450 lines
13 KiB
Bash
450 lines
13 KiB
Bash
#!/bin/bash
|
|
# Ver: 1.2 by Endial Fang (endial@126.com)
|
|
#
|
|
# 服务管理函数库
|
|
|
|
# shellcheck disable=SC1091
|
|
|
|
# 加载依赖项
|
|
source "/usr/local/lib/liblog.sh"
|
|
source "/usr/local/lib/libvalidations.sh"
|
|
|
|
# 函数列表
|
|
|
|
# 获取并返回服务 PID
|
|
# 参数:
|
|
# $1 - PID 文件
|
|
# 返回值:
|
|
# PID
|
|
get_pid_from_file() {
|
|
local pid_file="${1:?pid file is missing}"
|
|
|
|
if [[ -f "$pid_file" ]]; then
|
|
if [[ -n "$(< "$pid_file")" ]] && [[ "$(< "$pid_file")" -gt 0 ]]; then
|
|
echo "$(< "$pid_file")"
|
|
fi
|
|
fi
|
|
}
|
|
|
|
# 检测 PID 对应的服务是否在运行中
|
|
# 参数:
|
|
# $1 - PID
|
|
# 返回值:
|
|
# 0 / 1
|
|
is_service_running() {
|
|
local pid="${1:?pid is missing}"
|
|
|
|
kill -0 "$pid" 2>/dev/null
|
|
}
|
|
|
|
# 通过发送信号停止一个指定 PID 的服务
|
|
# 参数:
|
|
# $1 - PID 文件
|
|
# $2 - 信号 (可选)
|
|
stop_service_using_pid() {
|
|
local pid_file="${1:?pid file is missing}"
|
|
local signal="${2:-}"
|
|
local pid
|
|
|
|
pid="$(get_pid_from_file "$pid_file")"
|
|
[[ -z "$pid" ]] || ! is_service_running "$pid" && return
|
|
|
|
if [[ -n "$signal" ]]; then
|
|
kill "-${signal}" "$pid"
|
|
else
|
|
kill "$pid"
|
|
fi
|
|
|
|
local counter=10
|
|
while [[ "$counter" -ne 0 ]] && is_service_running "$pid"; do
|
|
sleep 1
|
|
counter=$((counter - 1))
|
|
done
|
|
}
|
|
|
|
# 启动一个 cron 守护进程
|
|
# 返回值:
|
|
# true / false
|
|
cron_start() {
|
|
if [[ -x "/usr/sbin/cron" ]]; then
|
|
/usr/sbin/cron
|
|
elif [[ -x "/usr/sbin/crond" ]]; then
|
|
/usr/sbin/crond
|
|
else
|
|
false
|
|
fi
|
|
}
|
|
|
|
# 为指定的服务生成 cron 配置文件
|
|
# 参数:
|
|
# $1 - 服务名称
|
|
# $2 - 命令
|
|
# 标志位:
|
|
# --run-as - 运行的用户. 默认值: root
|
|
# --schedule - Cron 周期配置. 默认值: * * * * *
|
|
generate_cron_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local cmd="${2:?command is missing}"
|
|
local run_as="root"
|
|
local schedule="* * * * *"
|
|
local clean="true"
|
|
|
|
# 检测标志位
|
|
shift 2
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case "$1" in
|
|
--run-as)
|
|
shift
|
|
run_as="$1"
|
|
;;
|
|
--schedule)
|
|
shift
|
|
schedule="$1"
|
|
;;
|
|
--no-clean)
|
|
clean="false"
|
|
;;
|
|
*)
|
|
echo "Invalid command line flag ${1}" >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
mkdir -p /etc/cron.d
|
|
if "$clean"; then
|
|
cat > "/etc/cron.d/${service_name}" <<EOF
|
|
${schedule} ${run_as} ${cmd}
|
|
EOF
|
|
else
|
|
echo "${schedule} ${run_as} ${cmd}" >> /etc/cron.d/"$service_name"
|
|
fi
|
|
}
|
|
|
|
# 删除指定服务的 cron 配置文件
|
|
# 参数:
|
|
# $1 - 服务名称
|
|
remove_cron_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local cron_conf_dir="/etc/monit/conf.d"
|
|
|
|
rm -f "${cron_conf_dir}/${service_name}"
|
|
}
|
|
|
|
# 为指定的服务生成 monit 配置文件
|
|
# 参数:
|
|
# $1 - 服务名
|
|
# $2 - PID 文件
|
|
# $3 - 启动命令
|
|
# $4 - 停止命令
|
|
# 标志位:
|
|
# --disabled - 是否禁用. 默认值: no
|
|
generate_monit_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local pid_file="${2:?pid file is missing}"
|
|
local start_command="${3:?start command is missing}"
|
|
local stop_command="${4:?stop command is missing}"
|
|
local monit_conf_dir="/etc/monit/conf.d"
|
|
local disabled="no"
|
|
|
|
# 检测标志位
|
|
shift 4
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case "$1" in
|
|
--disable)
|
|
disabled="yes"
|
|
;;
|
|
*)
|
|
echo "Invalid command line flag ${1}" >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
is_boolean_yes "$disabled" && conf_suffix=".disabled"
|
|
mkdir -p "$monit_conf_dir"
|
|
cat > "${monit_conf_dir}/${service_name}.conf${conf_suffix:-}" <<EOF
|
|
check process ${service_name}
|
|
with pidfile "${pid_file}"
|
|
start program = "${start_command}" with timeout 90 seconds
|
|
stop program = "${stop_command}" with timeout 90 seconds
|
|
EOF
|
|
}
|
|
|
|
# 删除指定服务的 monit 配置文件
|
|
# 参数:
|
|
# $1 - 服务名称
|
|
remove_monit_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local monit_conf_dir="/etc/monit/conf.d"
|
|
|
|
rm -f "${monit_conf_dir}/${service_name}.conf"
|
|
}
|
|
|
|
# 为指定的服务生成 Logrotate 配置文件
|
|
# 参数:
|
|
# $1 - 应用名称
|
|
# $2 - 日志路径
|
|
# 标志位:
|
|
# --period - 周期
|
|
# --rotations - Rotations 存储的数量
|
|
# --extra - 扩展参数 (可选)
|
|
generate_logrotate_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local log_path="${2:?log path is missing}"
|
|
local period="weekly"
|
|
local rotations="150"
|
|
local extra=""
|
|
local logrotate_conf_dir="/etc/logrotate.d"
|
|
local var_name
|
|
|
|
# 检测标志位
|
|
shift 2
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case "$1" in
|
|
--period|--rotations|--extra)
|
|
var_name="$(echo "$1" | sed -e "s/^--//" -e "s/-/_/g")"
|
|
shift
|
|
declare "$var_name"="${1:?"$var_name" is missing}"
|
|
;;
|
|
*)
|
|
echo "Invalid command line flag ${1}" >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
mkdir -p "$logrotate_conf_dir"
|
|
cat <<EOF | sed '/^\s*$/d' > "${logrotate_conf_dir}/${service_name}"
|
|
${log_path} {
|
|
${period}
|
|
rotate ${rotations}
|
|
dateext
|
|
compress
|
|
copytruncate
|
|
missingok
|
|
$(indent "$extra" 2)
|
|
}
|
|
EOF
|
|
}
|
|
|
|
# 删除指定服务的 Logrotate 配置文件
|
|
# 参数:
|
|
# $1 - 服务名称
|
|
remove_logrotate_conf() {
|
|
local service_name="${1:?service name is missing}"
|
|
local logrotate_conf_dir="/etc/logrotate.d"
|
|
rm -f "${logrotate_conf_dir}/${service_name}"
|
|
}
|
|
|
|
# 为指定的服务生成 Systemd 配置文件
|
|
# 参数:
|
|
# $1 - 服务名称
|
|
# 标志位:
|
|
# --custom-service-content - 自定义内容
|
|
# --environment - 环境变量（可选）
|
|
# --environment-file - 环境变量文件（可选）
|
|
# --exec-start - 启动命令（必须）
|
|
# --exec-start-pre - 启动前命令（可选）
|
|
# --exec-start-post - 启动后命令（可选）
|
|
# --exec-stop - 停止命令（可选）
|
|
# --exec-reload - 重载命令（可选）
|
|
# --group - 系统组
|
|
# --name - 服务名称（默认为 $1）
|
|
# --restart - 重启策略
|
|
# --pid-file - PID 文件
|
|
# --standard-output - 标准输出文件
|
|
# --standard-error - 标准错误文件
|
|
# --success-exit-status - 成功退出状态码
|
|
# --type - 服务类型（默认为 Fork）
|
|
# --user - 系统用户
|
|
# --working-directory - 工作目录
|
|
generate_systemd_conf() {
|
|
local -r service_name="${1:?service name is missing}"
|
|
local -r systemd_units_dir="/etc/systemd/system"
|
|
local -r service_file="${systemd_units_dir}/bitnami.${service_name}.service"
|
|
# Default values
|
|
local name="$service_name"
|
|
local type="forking"
|
|
local user=""
|
|
local group=""
|
|
local environment=""
|
|
local environment_file=""
|
|
local exec_start=""
|
|
local exec_start_pre=""
|
|
local exec_start_post=""
|
|
local exec_stop=""
|
|
local exec_reload=""
|
|
local restart="always"
|
|
local pid_file=""
|
|
local standard_output="journal"
|
|
local standard_error=""
|
|
local limits_content=""
|
|
local success_exit_status=""
|
|
local custom_service_content=""
|
|
local working_directory=""
|
|
# Parse CLI flags
|
|
shift
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case "$1" in
|
|
--name \
|
|
| --type \
|
|
| --user \
|
|
| --group \
|
|
| --exec-start \
|
|
| --exec-stop \
|
|
| --exec-reload \
|
|
| --restart \
|
|
| --pid-file \
|
|
| --standard-output \
|
|
| --standard-error \
|
|
| --success-exit-status \
|
|
| --custom-service-content \
|
|
| --working-directory \
|
|
)
|
|
var_name="$(echo "$1" | sed -e "s/^--//" -e "s/-/_/g")"
|
|
shift
|
|
declare "$var_name"="${1:?"${var_name} value is missing"}"
|
|
;;
|
|
--limit-*)
|
|
[[ -n "$limits_content" ]] && limits_content+=$'\n'
|
|
var_name="${1//--limit-}"
|
|
shift
|
|
limits_content+="Limit${var_name^^}=${1:?"--limit-${var_name} value is missing"}"
|
|
;;
|
|
--exec-start-pre)
|
|
shift
|
|
[[ -n "$exec_start_pre" ]] && exec_start_pre+=$'\n'
|
|
exec_start_pre+="ExecStartPre=${1:?"--exec-start-pre value is missing"}"
|
|
;;
|
|
--exec-start-post)
|
|
shift
|
|
[[ -n "$exec_start_post" ]] && exec_start_post+=$'\n'
|
|
exec_start_post+="ExecStartPost=${1:?"--exec-start-post value is missing"}"
|
|
;;
|
|
--environment)
|
|
shift
|
|
# It is possible to add multiple environment lines
|
|
[[ -n "$environment" ]] && environment+=$'\n'
|
|
environment+="Environment=${1:?"--environment value is missing"}"
|
|
;;
|
|
--environment-file)
|
|
shift
|
|
# It is possible to add multiple environment-file lines
|
|
[[ -n "$environment_file" ]] && environment_file+=$'\n'
|
|
environment_file+="EnvironmentFile=${1:?"--environment-file value is missing"}"
|
|
;;
|
|
*)
|
|
echo "Invalid command line flag ${1}" >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
# Validate inputs
|
|
local error="no"
|
|
if [[ -z "$exec_start" ]]; then
|
|
error "The --exec-start option is required"
|
|
error="yes"
|
|
fi
|
|
if [[ "$error" != "no" ]]; then
|
|
return 1
|
|
fi
|
|
# Generate the Systemd unit
|
|
cat > "$service_file" <<EOF
|
|
# Copyright Broadcom, Inc. All Rights Reserved.
|
|
# SPDX-License-Identifier: APACHE-2.0
|
|
|
|
[Unit]
|
|
Description=Bitnami service for ${name}
|
|
# Starting/stopping the main bitnami service should cause the same effect for this service
|
|
PartOf=bitnami.service
|
|
|
|
[Service]
|
|
Type=${type}
|
|
EOF
|
|
if [[ -n "$working_directory" ]]; then
|
|
cat >> "$service_file" <<< "WorkingDirectory=${working_directory}"
|
|
fi
|
|
if [[ -n "$exec_start_pre" ]]; then
|
|
# This variable may contain multiple ExecStartPre= directives
|
|
cat >> "$service_file" <<< "$exec_start_pre"
|
|
fi
|
|
if [[ -n "$exec_start" ]]; then
|
|
cat >> "$service_file" <<< "ExecStart=${exec_start}"
|
|
fi
|
|
if [[ -n "$exec_start_post" ]]; then
|
|
# This variable may contain multiple ExecStartPost= directives
|
|
cat >> "$service_file" <<< "$exec_start_post"
|
|
fi
|
|
# Optional stop and reload commands
|
|
if [[ -n "$exec_stop" ]]; then
|
|
cat >> "$service_file" <<< "ExecStop=${exec_stop}"
|
|
fi
|
|
if [[ -n "$exec_reload" ]]; then
|
|
cat >> "$service_file" <<< "ExecReload=${exec_reload}"
|
|
fi
|
|
# User and group
|
|
if [[ -n "$user" ]]; then
|
|
cat >> "$service_file" <<< "User=${user}"
|
|
fi
|
|
if [[ -n "$group" ]]; then
|
|
cat >> "$service_file" <<< "Group=${group}"
|
|
fi
|
|
# PID file allows to determine if the main process is running properly (for Restart=always)
|
|
if [[ -n "$pid_file" ]]; then
|
|
cat >> "$service_file" <<< "PIDFile=${pid_file}"
|
|
fi
|
|
if [[ -n "$restart" ]]; then
|
|
cat >> "$service_file" <<< "Restart=${restart}"
|
|
fi
|
|
# Environment flags
|
|
if [[ -n "$environment" ]]; then
|
|
# This variable may contain multiple Environment= directives
|
|
cat >> "$service_file" <<< "$environment"
|
|
fi
|
|
if [[ -n "$environment_file" ]]; then
|
|
# This variable may contain multiple EnvironmentFile= directives
|
|
cat >> "$service_file" <<< "$environment_file"
|
|
fi
|
|
# Logging
|
|
if [[ -n "$standard_output" ]]; then
|
|
cat >> "$service_file" <<< "StandardOutput=${standard_output}"
|
|
fi
|
|
if [[ -n "$standard_error" ]]; then
|
|
cat >> "$service_file" <<< "StandardError=${standard_error}"
|
|
fi
|
|
if [[ -n "$custom_service_content" ]]; then
|
|
# This variable may contain multiple miscellaneous directives
|
|
cat >> "$service_file" <<< "$custom_service_content"
|
|
fi
|
|
if [[ -n "$success_exit_status" ]]; then
|
|
cat >> "$service_file" <<EOF
|
|
# When the process receives a SIGTERM signal, it exits with code ${success_exit_status}
|
|
SuccessExitStatus=${success_exit_status}
|
|
EOF
|
|
fi
|
|
cat >> "$service_file" <<EOF
|
|
# Optimizations
|
|
TimeoutStartSec=2min
|
|
TimeoutStopSec=30s
|
|
IgnoreSIGPIPE=no
|
|
KillMode=mixed
|
|
EOF
|
|
if [[ -n "$limits_content" ]]; then
|
|
cat >> "$service_file" <<EOF
|
|
# Limits
|
|
${limits_content}
|
|
EOF
|
|
fi
|
|
cat >> "$service_file" <<EOF
|
|
|
|
[Install]
|
|
# Enabling/disabling the main bitnami service should cause the same effect for this service
|
|
WantedBy=bitnami.service
|
|
EOF
|
|
}
|