From 36151b9e54ec605f3094e3e94ba2b641b638a691 Mon Sep 17 00:00:00 2001
From: Alexey Milovidov <milovidov@yandex-team.ru>
Date: Thu, 26 Nov 2020 07:54:18 +0300
Subject: [PATCH] Simplify init script (part 2)

---
 debian/clickhouse-server.init | 52 ++++-------------------------------
 programs/install/Install.cpp  | 17 ++++++++----
 2 files changed, 17 insertions(+), 52 deletions(-)

diff --git a/debian/clickhouse-server.init b/debian/clickhouse-server.init
index 8f10153a682..b7b19e429fa 100755
--- a/debian/clickhouse-server.init
+++ b/debian/clickhouse-server.init
@@ -67,26 +67,6 @@ if uname -mpi | grep -q 'x86_64'; then
 fi
 
 
-is_running()
-{
-    pgrep --pidfile "$CLICKHOUSE_PIDFILE" $(echo "${PROGRAM}" | cut -c1-15) 1> /dev/null 2> /dev/null
-}
-
-
-wait_for_done()
-{
-    timeout=$1
-    attempts=0
-    while is_running; do
-        attempts=$(($attempts + 1))
-        if [ -n "$timeout" ] && [ $attempts -gt $timeout ]; then
-            return 1
-        fi
-        sleep 1
-    done
-}
-
-
 die()
 {
     echo $1 >&2
@@ -171,17 +151,7 @@ restart()
 
 forcestop()
 {
-    local EXIT_STATUS
-    EXIT_STATUS=0
-
-    echo -n "Stop forcefully $PROGRAM service: "
-
-    kill -KILL $(cat "$CLICKHOUSE_PIDFILE")
-
-    wait_for_done
-
-    echo "DONE"
-    return $EXIT_STATUS
+    ${CLICKHOUSE_GENERIC_PROGRAM} stop --force --pid-path "${CLICKHOUSE_PIDDIR}"
 }
 
 
@@ -261,16 +231,16 @@ main()
         service_or_func restart
         ;;
     condstart)
-        is_running || service_or_func start
+        service_or_func start
         ;;
     condstop)
-        is_running && service_or_func stop
+        service_or_func stop
         ;;
     condrestart)
-        is_running && service_or_func restart
+        service_or_func restart
         ;;
     condreload)
-        is_running && service_or_func restart
+        service_or_func restart
         ;;
     initdb)
         initdb
@@ -293,17 +263,7 @@ main()
 
 status()
 {
-    if is_running; then
-        echo "$PROGRAM service is running"
-        exit 0
-    else
-        if is_cron_disabled; then
-            echo "$PROGRAM service is stopped";
-        else
-            echo "$PROGRAM: process unexpectedly terminated"
-        fi
-        exit 3
-    fi
+    ${CLICKHOUSE_GENERIC_PROGRAM} status --pid-path "${CLICKHOUSE_PIDDIR}"
 }
 
 
diff --git a/programs/install/Install.cpp b/programs/install/Install.cpp
index da22452819a..b67fdcbeb19 100644
--- a/programs/install/Install.cpp
+++ b/programs/install/Install.cpp
@@ -783,17 +783,20 @@ namespace
         return pid;
     }
 
-    int stop(const fs::path & pid_file)
+    int stop(const fs::path & pid_file, bool force)
     {
         UInt64 pid = isRunning(pid_file);
 
         if (!pid)
             return 0;
 
-        if (0 == kill(pid, 15)) /// Terminate
-            fmt::print("Sent termination signal.\n", pid);
+        int signal = force ? SIGKILL : SIGTERM;
+        const char * signal_name = force ? "kill" : "terminate";
+
+        if (0 == kill(pid, signal))
+            fmt::print("Sent {} signal to process with pid {}.\n", signal_name, pid);
         else
-            throwFromErrno("Cannot send termination signal", ErrorCodes::SYSTEM_ERROR);
+            throwFromErrno(fmt::format("Cannot send {} signal", signal_name), ErrorCodes::SYSTEM_ERROR);
 
         size_t try_num = 0;
         constexpr size_t num_tries = 60;
@@ -869,6 +872,7 @@ int mainEntryClickHouseStop(int argc, char ** argv)
     desc.add_options()
         ("help,h", "produce help message")
         ("pid-path", po::value<std::string>()->default_value("/var/run/clickhouse-server"), "directory for pid file")
+        ("force", po::value<bool>()->default_value(false), "Stop with KILL signal instead of TERM")
     ;
 
     po::variables_map options;
@@ -887,7 +891,7 @@ int mainEntryClickHouseStop(int argc, char ** argv)
     {
         fs::path pid_file = fs::path(options["pid-path"].as<std::string>()) / "clickhouse-server.pid";
 
-        return stop(pid_file);
+        return stop(pid_file, options["force"].as<bool>());
     }
     catch (...)
     {
@@ -940,6 +944,7 @@ int mainEntryClickHouseRestart(int argc, char ** argv)
         ("config-path", po::value<std::string>()->default_value("/etc/clickhouse-server"), "directory with configs")
         ("pid-path", po::value<std::string>()->default_value("/var/run/clickhouse-server"), "directory for pid file")
         ("user", po::value<std::string>()->default_value("clickhouse"), "clickhouse user")
+        ("force", po::value<bool>()->default_value(false), "Stop with KILL signal instead of TERM")
     ;
 
     po::variables_map options;
@@ -962,7 +967,7 @@ int mainEntryClickHouseRestart(int argc, char ** argv)
         fs::path config = fs::path(options["config-path"].as<std::string>()) / "config.xml";
         fs::path pid_file = fs::path(options["pid-path"].as<std::string>()) / "clickhouse-server.pid";
 
-        if (int res = stop(pid_file))
+        if (int res = stop(pid_file, options["force"].as<bool>()))
             return res;
         return start(user, executable, config, pid_file);
     }