ClickHouse/debian/clickhouse-server.init

221 lines
4.9 KiB
Plaintext
Raw Normal View History

#!/bin/sh
### BEGIN INIT INFO
# Provides: clickhouse-server
# Default-Start: 2 3 4 5
# Default-Stop: 0 1 6
# Required-Start: $network
# Required-Stop: $network
# Short-Description: Yandex clickhouse-server daemon
### END INIT INFO
CLICKHOUSE_USER=clickhouse
CLICKHOUSE_GROUP=${CLICKHOUSE_USER}
SHELL=/bin/bash
PROGRAM=clickhouse-server
2019-01-31 13:03:17 +00:00
CLICKHOUSE_GENERIC_PROGRAM=clickhouse
2018-11-19 22:07:34 +00:00
CLICKHOUSE_PROGRAM_ENV=""
2019-01-31 13:03:17 +00:00
EXTRACT_FROM_CONFIG=${CLICKHOUSE_GENERIC_PROGRAM}-extract-from-config
CLICKHOUSE_CONFDIR=/etc/$PROGRAM
2017-02-03 13:52:59 +00:00
CLICKHOUSE_LOGDIR=/var/log/clickhouse-server
CLICKHOUSE_LOGDIR_USER=root
2019-01-31 13:03:17 +00:00
CLICKHOUSE_DATADIR=/var/lib/clickhouse
2020-01-09 12:59:45 +00:00
if [ -d "/var/lock" ]; then
LOCALSTATEDIR=/var/lock
else
LOCALSTATEDIR=/run/lock
2020-01-09 12:59:45 +00:00
fi
if [ ! -d "$LOCALSTATEDIR" ]; then
mkdir -p "$LOCALSTATEDIR"
fi
2019-01-31 13:03:17 +00:00
CLICKHOUSE_BINDIR=/usr/bin
CLICKHOUSE_CRONFILE=/etc/cron.d/clickhouse-server
2019-01-31 13:03:17 +00:00
CLICKHOUSE_CONFIG=$CLICKHOUSE_CONFDIR/config.xml
LOCKFILE=$LOCALSTATEDIR/$PROGRAM
CLICKHOUSE_PIDDIR=/var/run/$PROGRAM
CLICKHOUSE_PIDFILE="$CLICKHOUSE_PIDDIR/$PROGRAM.pid"
# CLICKHOUSE_STOP_TIMEOUT=60 # Disabled by default. Place to /etc/default/clickhouse if you need.
2018-05-16 00:34:56 +00:00
# Some systems lack "flock"
command -v flock >/dev/null && FLOCK=flock
# Override defaults from optional config file
test -f /etc/default/clickhouse && . /etc/default/clickhouse
die()
{
2017-07-08 16:09:32 +00:00
echo $1 >&2
exit 1
}
# Check that configuration file is Ok.
check_config()
{
2019-01-31 13:03:17 +00:00
if [ -x "$CLICKHOUSE_BINDIR/$EXTRACT_FROM_CONFIG" ]; then
su -s $SHELL ${CLICKHOUSE_USER} -c "$CLICKHOUSE_BINDIR/$EXTRACT_FROM_CONFIG --config-file=\"$CLICKHOUSE_CONFIG\" --key=path" >/dev/null || die "Configuration file ${CLICKHOUSE_CONFIG} doesn't parse successfully. Won't restart server. You may use forcerestart if you are sure.";
2017-07-08 16:09:32 +00:00
fi
}
initdb()
{
2020-11-26 05:13:45 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} install --user "${CLICKHOUSE_USER}" --pid-path "${CLICKHOUSE_PIDDIR}" --config-path "${CLICKHOUSE_CONFDIR}" --binary-path "${CLICKHOUSE_BINDIR}"
}
start()
{
2020-08-27 18:50:13 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} start --user "${CLICKHOUSE_USER}" --pid-path "${CLICKHOUSE_PIDDIR}" --config-path "${CLICKHOUSE_CONFDIR}" --binary-path "${CLICKHOUSE_BINDIR}"
}
stop()
{
2020-08-26 19:53:49 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} stop --pid-path "${CLICKHOUSE_PIDDIR}"
}
restart()
{
2020-08-27 18:50:13 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} restart --user "${CLICKHOUSE_USER}" --pid-path "${CLICKHOUSE_PIDDIR}" --config-path "${CLICKHOUSE_CONFDIR}" --binary-path "${CLICKHOUSE_BINDIR}"
}
2010-03-09 19:01:07 +00:00
forcestop()
{
2020-11-26 04:54:18 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} stop --force --pid-path "${CLICKHOUSE_PIDDIR}"
2010-03-09 19:01:07 +00:00
}
2019-05-06 13:24:22 +00:00
service_or_func()
{
if [ -x "/bin/systemctl" ] && [ -f /etc/systemd/system/clickhouse-server.service ] && [ -d /run/systemd/system ]; then
2021-07-02 20:02:27 +00:00
systemctl $1 $PROGRAM
2019-05-06 13:24:22 +00:00
else
$1
fi
}
2010-03-09 19:01:07 +00:00
forcerestart()
{
2017-07-08 16:09:32 +00:00
forcestop
# Should not use 'start' function if systemd active
2019-05-06 13:24:22 +00:00
service_or_func start
2010-03-09 19:01:07 +00:00
}
2018-03-01 20:16:03 +00:00
use_cron()
{
# 1. running systemd
if [ -x "/bin/systemctl" ] && [ -f /etc/systemd/system/clickhouse-server.service ] && [ -d /run/systemd/system ]; then
return 1
fi
# 2. disabled by config
if [ -z "$CLICKHOUSE_CRONFILE" ]; then
return 2
fi
return 0
}
# returns false if cron disabled (with systemd)
enable_cron()
{
2018-03-01 20:16:03 +00:00
use_cron && sed -i 's/^#*//' "$CLICKHOUSE_CRONFILE"
}
# returns false if cron disabled (with systemd)
disable_cron()
{
2018-03-01 20:16:03 +00:00
use_cron && sed -i 's/^#*/#/' "$CLICKHOUSE_CRONFILE"
}
is_cron_disabled()
{
2018-03-01 20:16:03 +00:00
use_cron || return 0
2017-07-08 16:09:32 +00:00
# Assumes that either no lines are commented or all lines are commented.
# Also please note, that currently cron file for ClickHouse has only one line (but some time ago there was more).
grep -q -E '^#' "$CLICKHOUSE_CRONFILE";
}
main()
{
2017-07-08 16:09:32 +00:00
# See how we were called.
EXIT_STATUS=0
case "$1" in
start)
service_or_func start && enable_cron
2017-07-08 16:09:32 +00:00
;;
stop)
disable_cron
service_or_func stop
2017-07-08 16:09:32 +00:00
;;
restart)
service_or_func restart && enable_cron
2017-07-08 16:09:32 +00:00
;;
forcestop)
disable_cron
forcestop
2017-07-08 16:09:32 +00:00
;;
forcerestart)
forcerestart && enable_cron
;;
reload)
service_or_func restart
2017-07-08 16:09:32 +00:00
;;
condstart)
2020-11-26 04:54:18 +00:00
service_or_func start
2017-07-08 16:09:32 +00:00
;;
condstop)
2020-11-26 04:54:18 +00:00
service_or_func stop
2017-07-08 16:09:32 +00:00
;;
condrestart)
2020-11-26 04:54:18 +00:00
service_or_func restart
2017-07-08 16:09:32 +00:00
;;
condreload)
2020-11-26 04:54:18 +00:00
service_or_func restart
2017-07-08 16:09:32 +00:00
;;
initdb)
initdb
;;
enable_cron)
enable_cron
;;
disable_cron)
disable_cron
;;
2017-07-08 16:09:32 +00:00
*)
echo "Usage: $0 {start|stop|status|restart|forcestop|forcerestart|reload|condstart|condstop|condrestart|condreload|initdb}"
2017-07-08 16:09:32 +00:00
exit 2
;;
esac
exit $EXIT_STATUS
}
status()
{
2020-11-26 04:54:18 +00:00
${CLICKHOUSE_GENERIC_PROGRAM} status --pid-path "${CLICKHOUSE_PIDDIR}"
}
# Running commands without need of locking
case "$1" in
status)
2017-07-08 16:09:32 +00:00
status
exit 0
2017-07-08 16:09:32 +00:00
;;
esac
(
2018-05-16 00:34:56 +00:00
if $FLOCK -n 9; then
2017-07-08 16:09:32 +00:00
main "$@"
else
echo "Init script is already running" && exit 1
fi
) 9> $LOCKFILE