summaryrefslogtreecommitdiff
blob: 20479dc22dab7b8e9726f15e3d550311d84bf3b2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/sbin/openrc-run

extra_started_commands="reload"

depend() {
    use net
    after postgresql
}

get_config() {
    [ -f "${INIFILE}" ] || eend 1 "'${INIFILE}' not found"

    eval echo $(sed -e 's:;.*::' "${INIFILE}" | \
        awk '$1 == "'$1'" { print ($2 == "=" ? $3 : $2) }')
}

PIDFILE="$(get_config pidfile)"
UNIX_SOCKET_DIR="$(get_config unix_socket_dir)"

prep() {
    if [ -n "${UNIX_SOCKET_DIR}" ] ; then
        checkpath -o postgres:postgres -m 0775 -d "${UNIX_SOCKET_DIR}" \
            || return 1
    fi
    checkpath -o pgbouncer:postgres -m 0755 -d "$(dirname ${PIDFILE})" \
        || return 1
    checkpath -o pgbouncer:postgres -m 0644 -f "${PIDFILE}" \
        || return 1
    checkpath -o pgbouncer:postgres -m 0755 -d "$(dirname $(get_config logfile))" \
        || return 1
    checkpath -o pgbouncer:postgres -m 0640 -f "$(get_config logfile)" \
        || return 1

    return 0
}

start() {
    ebegin "Starting PgBouncer"
    prep
    local ret=$?
    if [ $ret -ne 0 ] ; then
        eend $ret
        exit $ret
    fi
    start-stop-daemon --start \
        --pidfile ${PIDFILE} \
        --user pgbouncer \
        --exec /usr/bin/pgbouncer -- -q -d "${INIFILE}"
    eend $?
}

stop() {
    local seconds=$(( ${NICE_TIMEOUT} + ${FORCE_QUIT_TIMEOUT} ))
    ebegin "Stopping PgBouncer (this can take up to ${seconds} seconds)"

    local retries=SIGINT/${NICE_TIMEOUT}

    if [ "${FORCE_QUIT}" = "YES" ] ; then
        einfo "FORCE_QUIT enabled."
        retries="${retries}/SIGTERM/${FORCE_QUIT_TIMEOUT}"
    fi

    # Loops through nice and force quit in one go.
    start-stop-daemon --stop \
        --pidfile ${PIDFILE} \
        --retry ${retries}

    eend $?
}

restart() {
    if [ -n "${UNIX_SOCKET_DIR}" ] ; then
        ebegin "Performing online restart of PgBouncer"
        start-stop-daemon --start \
            --pidfile ${PIDFILE} \
            --user pgbouncer \
            --exec /usr/bin/pgbouncer -- -q -d -R "${INIFILE}"
        eend $?
    else
        stop && start
    fi
}

reload() {
    ebegin "Reloading PgBouncer configuration from '${INIFILE}'"
    start-stop-daemon --signal HUP --pidfile ${PIDFILE}
    eend $?
}