scylla/scylla_lib.sh at master · ajithdaniel/scylla · GitHub

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
89
90
91
92
93
94
95
96
97
98
99
#
#  Copyright (C) 2016 ScyllaDB

is_debian_variant() {
    [ -f /etc/debian_version ]
}

is_redhat_variant() {
    [ -f /etc/redhat-release ]
}

is_gentoo_variant() {
    [ -f /etc/gentoo-release ]
}

is_systemd() {
    grep -q '^systemd$' /proc/1/comm
}

is_ec2() {
    [ -f /sys/hypervisor/uuid ] && [ "$(head -c 3 /sys/hypervisor/uuid)" = "ec2" ]
}

is_selinux_enabled() {
    STATUS=`getenforce`
    if [ "$STATUS" = "Disabled" ]; then
        return 0
    else
        return 1
    fi
}

ec2_is_supported_instance_type() {
    TYPE=`curl -s http://169.254.169.254/latest/meta-data/instance-type|cut -d . -f 1`
    case $TYPE in
           "m3"|"c3"|"i2"|"i3") echo 1;;
            *) echo 0;;
    esac
}

#
# check_cpuset_conf <NIC name>
#
get_tune_mode() {
    local nic=$1

    # if cpuset.conf doesn't exist use the default mode
    [[ ! -e '/etc/scylla.d/cpuset.conf' ]] && return

    local cur_cpuset=`cat /etc/scylla.d/cpuset.conf | cut -d "\"" -f2- | cut -d" " -f2`
    local mq_cpuset=`/usr/lib/scylla/perftune.py --tune net --nic "$nic" --mode mq --get-cpu-mask | /usr/lib/scylla/hex2list.py`
    local sq_cpuset=`/usr/lib/scylla/perftune.py --tune net --nic "$nic" --mode sq --get-cpu-mask | /usr/lib/scylla/hex2list.py`
    local sq_split_cpuset=`/usr/lib/scylla/perftune.py --tune net --nic "$nic" --mode sq_split --get-cpu-mask | /usr/lib/scylla/hex2list.py`
    local tune_mode=""

    case "$cur_cpuset" in
        "$mq_cpuset")
            tune_mode="--mode mq"
            ;;
        "$sq_cpuset")
            tune_mode="--mode sq"
            ;;
        "$sq_split_cpuset")
            tune_mode="--mode sq_split"
            ;;
    esac

    # if cpuset is something different from what we expect - use the default mode
    echo "$tune_mode"
}

#
# create_perftune_conf [<NIC name>]
#
create_perftune_conf() {
    local nic=$1
    [[ -z "$nic" ]] && nic='eth0'

    # if exists - do nothing
    [[ -e '/etc/scylla.d/perftune.yaml' ]] && return

    local mode=`get_tune_mode "$nic"`
    /usr/lib/scylla/perftune.py --tune net --nic "$nic" $mode --dump-options-file > /etc/scylla.d/perftune.yaml
}

. /etc/os-release
if is_debian_variant || is_gentoo_variant; then
    SYSCONFIG=/etc/default
else
    SYSCONFIG=/etc/sysconfig
fi
. $SYSCONFIG/scylla-server

for i in /etc/scylla.d/*.conf; do
    if [ "$i" = "/etc/scylla.d/*.conf" ]; then
        break
    fi
    . "$i"
done