post.sh 11.1 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6
#!/bin/bash
#
# This file is used to install tdengine rpm package on centos systems. The operating system 
# is required to use systemd to manage services at boot
#set -x
# -----------------------Variables definition---------------------
F
Frozen 已提交
7
script_dir=$(dirname $(readlink -f "$0"))
H
hzcheng 已提交
8 9 10 11 12
# Dynamic directory
data_dir="/var/lib/taos"
log_dir="/var/log/taos"
data_link_dir="/usr/local/taos/data"
log_link_dir="/usr/local/taos/log"
H
Hui Li 已提交
13
install_main_dir="/usr/local/taos"
H
hzcheng 已提交
14 15 16 17 18 19 20 21 22 23 24

# static directory
cfg_dir="/usr/local/taos/cfg"
bin_dir="/usr/local/taos/bin"
lib_dir="/usr/local/taos/driver"
init_d_dir="/usr/local/taos/init.d"
inc_dir="/usr/local/taos/include"

cfg_install_dir="/etc/taos"
bin_link_dir="/usr/bin"
lib_link_dir="/usr/lib"
H
Hui Li 已提交
25
lib64_link_dir="/usr/lib64"
H
hzcheng 已提交
26 27 28 29 30 31 32 33 34 35 36 37
inc_link_dir="/usr/include"

service_config_dir="/etc/systemd/system"


# Color setting
RED='\033[0;31m'
GREEN='\033[1;32m'
GREEN_DARK='\033[0;32m'
GREEN_UNDERLINE='\033[4;32m'
NC='\033[0m'

P
plum-lihui 已提交
38 39 40 41 42
csudo=""
if command -v sudo > /dev/null; then
    csudo="sudo"
fi

H
[NONE]  
huili 已提交
43
initd_mod=0
P
plum-lihui 已提交
44 45 46
service_mod=2
if pidof systemd &> /dev/null; then
    service_mod=0
L
lihui 已提交
47
elif $(which service &> /dev/null); then    
H
[NONE]  
huili 已提交
48
    service_mod=1
L
lihui 已提交
49 50 51 52 53 54 55 56 57 58
    service_config_dir="/etc/init.d" 
    if $(which chkconfig &> /dev/null); then
         initd_mod=1 
    elif $(which insserv &> /dev/null); then
        initd_mod=2
    elif $(which update-rc.d &> /dev/null); then
        initd_mod=3
    else
        service_mod=2
    fi
P
plum-lihui 已提交
59 60 61 62 63
else 
    service_mod=2
fi

function kill_taosd() {
H
[NONE]  
huili 已提交
64 65
#  ${csudo} pkill -f taosd || :
  pid=$(ps -ef | grep "taosd" | grep -v "grep" | awk '{print $2}')
L
lihui 已提交
66 67 68
  if [ -n "$pid" ]; then
    ${csudo} kill -9 $pid   || :
  fi
H
hzcheng 已提交
69 70 71
}

function install_include() {
L
lihui 已提交
72
    ${csudo} rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taoserror.h|| :
P
plum-lihui 已提交
73
    ${csudo} ln -s ${inc_dir}/taos.h ${inc_link_dir}/taos.h  
L
lihui 已提交
74
    ${csudo} ln -s ${inc_dir}/taoserror.h ${inc_link_dir}/taoserror.h  
H
hzcheng 已提交
75 76 77
}

function install_lib() {
H
huili 已提交
78
    ${csudo} rm -f ${lib_link_dir}/libtaos* || :
H
Hui Li 已提交
79
    ${csudo} rm -f ${lib64_link_dir}/libtaos* || :
H
huili 已提交
80
    
P
plum-lihui 已提交
81 82
    ${csudo} ln -s ${lib_dir}/libtaos.* ${lib_link_dir}/libtaos.so.1
    ${csudo} ln -s ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
H
Hui Li 已提交
83
    
H
Hui Li 已提交
84 85
    ${csudo} ln -s ${lib_dir}/libtaos.* ${lib64_link_dir}/libtaos.so.1           || :
    ${csudo} ln -s ${lib64_link_dir}/libtaos.so.1 ${lib64_link_dir}/libtaos.so   || :
H
hzcheng 已提交
86 87 88 89
}

function install_bin() {
    # Remove links
P
plum-lihui 已提交
90 91
    ${csudo} rm -f ${bin_link_dir}/taos     || :
    ${csudo} rm -f ${bin_link_dir}/taosd    || :
S
slguan 已提交
92
    ${csudo} rm -f ${bin_link_dir}/taosdemo || :
P
plum-lihui 已提交
93
    ${csudo} rm -f ${bin_link_dir}/rmtaos   || :
H
Hui Li 已提交
94
    ${csudo} rm -f ${bin_link_dir}/set_core || :
H
hzcheng 已提交
95

P
plum-lihui 已提交
96
    ${csudo} chmod 0555 ${bin_dir}/*
H
hzcheng 已提交
97 98

    #Make link
P
plum-lihui 已提交
99 100
    [ -x ${bin_dir}/taos ] && ${csudo} ln -s ${bin_dir}/taos ${bin_link_dir}/taos             || :
    [ -x ${bin_dir}/taosd ] && ${csudo} ln -s ${bin_dir}/taosd ${bin_link_dir}/taosd          || :
S
slguan 已提交
101
    [ -x ${bin_dir}/taosdemo ] && ${csudo} ln -s ${bin_dir}/taosdemo ${bin_link_dir}/taosdemo || :
H
Hui Li 已提交
102
    [ -x ${bin_dir}/set_core.sh ] && ${csudo} ln -s ${bin_dir}/set_core.sh ${bin_link_dir}/set_core || :
H
hzcheng 已提交
103 104 105 106
}

function install_config() {
    if [ ! -f ${cfg_install_dir}/taos.cfg ]; then
P
plum-lihui 已提交
107 108 109
        ${csudo} ${csudo} mkdir -p ${cfg_install_dir}
        [ -f ${cfg_dir}/taos.cfg ] && ${csudo} cp ${cfg_dir}/taos.cfg ${cfg_install_dir}
        ${csudo} chmod 644 ${cfg_install_dir}/*
H
hzcheng 已提交
110 111
    fi

P
plum-lihui 已提交
112 113
    ${csudo} mv ${cfg_dir}/taos.cfg ${cfg_dir}/taos.cfg.org
    ${csudo} ln -s ${cfg_install_dir}/taos.cfg ${cfg_dir}
H
Hui Li 已提交
114 115 116 117 118 119 120
    #FQDN_FORMAT="(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)"
    #FQDN_FORMAT="(:[1-6][0-9][0-9][0-9][0-9]$)"
    #PORT_FORMAT="(/[1-6][0-9][0-9][0-9][0-9]?/)"
    #FQDN_PATTERN=":[0-9]{1,5}$"

    # first full-qualified domain name (FQDN) for TDengine cluster system
    echo
H
Hui Li 已提交
121 122 123
    echo -e -n "${GREEN}Enter FQDN:port (like h1.taosdata.com:6030) of an existing TDengine cluster node to join${NC}"
    echo
    echo -e -n "${GREEN}OR leave it blank to build one${NC}:"
H
Hui Li 已提交
124 125 126 127 128 129 130 131 132 133 134 135 136 137
    read firstEp
    while true; do
        if [ ! -z "$firstEp" ]; then
            # check the format of the firstEp
            #if [[ $firstEp == $FQDN_PATTERN ]]; then
                # Write the first FQDN to configuration file                    
                ${csudo} sed -i -r "s/#*\s*(firstEp\s*).*/\1$firstEp/" ${cfg_install_dir}/taos.cfg    
                break
            #else
            #    read -p "Please enter the correct FQDN:port: " firstEp
            #fi
        else
            break
        fi
H
Hui Li 已提交
138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
    done		

    # user email 
    #EMAIL_PATTERN='^[A-Za-z0-9\u4e00-\u9fa5]+@[a-zA-Z0-9_-]+(\.[a-zA-Z0-9_-]+)+$'
    #EMAIL_PATTERN='^[\w-]+(\.[\w-]+)*@[\w-]+(\.[\w-]+)+$'
    #EMAIL_PATTERN="^[\w-]+(\.[\w-]+)*@[\w-]+(\.[\w-]+)+$"
    echo
    echo -e -n "${GREEN}Enter your email address for priority support or enter empty to skip${NC}: "
    read emailAddr
    while true; do
        if [ ! -z "$emailAddr" ]; then
            # check the format of the emailAddr
            #if [[ "$emailAddr" =~ $EMAIL_PATTERN ]]; then
                # Write the email address to temp file                    
                email_file="${install_main_dir}/email" 
                ${csudo} bash -c "echo $emailAddr > ${email_file}"
                break         
            #else
            #    read -p "Please enter the correct email address: " emailAddr   
            #fi
        else
            break
        fi
H
Hui Li 已提交
161
    done	
H
hzcheng 已提交
162 163 164
}

function clean_service_on_sysvinit() {
L
lihui 已提交
165 166 167
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
    #${csudo} sed -i "\|${restart_config_str}|d" /etc/inittab || :    
    
H
[NONE]  
huili 已提交
168 169 170
    if pidof taosd &> /dev/null; then
        ${csudo} service taosd stop || :
    fi
L
lihui 已提交
171
    
H
[NONE]  
huili 已提交
172
    if ((${initd_mod}==1)); then
L
lihui 已提交
173
        ${csudo} chkconfig --del taosd || :
H
[NONE]  
huili 已提交
174
    elif ((${initd_mod}==2)); then
L
lihui 已提交
175 176 177 178 179 180 181 182 183
        ${csudo} insserv -r taosd || :
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d -f taosd remove || :
    fi
    
    ${csudo} rm -f ${service_config_dir}/taosd || :
    
    if $(which init &> /dev/null); then
        ${csudo} init q || :
H
[NONE]  
huili 已提交
184
    fi
H
hzcheng 已提交
185 186 187 188 189 190 191 192
}

function install_service_on_sysvinit() {
    clean_service_on_sysvinit

    sleep 1

    # Install taosd service  
P
plum-lihui 已提交
193
    ${csudo} cp %{init_d_dir}/taosd ${service_config_dir} && ${csudo} chmod a+x ${service_config_dir}/taosd
H
hzcheng 已提交
194

L
lihui 已提交
195 196 197 198 199 200 201 202 203 204 205 206
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
    #${csudo} grep -q -F "$restart_config_str" /etc/inittab || ${csudo} bash -c "echo '${restart_config_str}' >> /etc/inittab"
    
    if ((${initd_mod}==1)); then
        ${csudo} chkconfig --add taosd || :
        ${csudo} chkconfig --level 2345 taosd on || :
    elif ((${initd_mod}==2)); then
        ${csudo} insserv taosd || :
        ${csudo} insserv -d taosd || :
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d taosd defaults || :
    fi
H
hzcheng 已提交
207 208 209 210 211
}

function clean_service_on_systemd() {
    taosd_service_config="${service_config_dir}/taosd.service"

L
lihui 已提交
212
    # taosd service already is stoped before install in preinst script
H
hzcheng 已提交
213 214
    #if systemctl is-active --quiet taosd; then
    #    echo "TDengine is running, stopping it..."
P
plum-lihui 已提交
215
    #    ${csudo} systemctl stop taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
216
    #fi
P
plum-lihui 已提交
217
    ${csudo} systemctl disable taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
218

P
plum-lihui 已提交
219
    ${csudo} rm -f ${taosd_service_config}
H
hzcheng 已提交
220 221 222 223 224 225 226 227 228
}

# taos:2345:respawn:/etc/init.d/taosd start

function install_service_on_systemd() {
    clean_service_on_systemd

    taosd_service_config="${service_config_dir}/taosd.service"

P
plum-lihui 已提交
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248
    ${csudo} bash -c "echo '[Unit]'                             >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Description=TDengine server service' >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'After=network-online.target'        >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Wants=network-online.target'        >> ${taosd_service_config}"
    ${csudo} bash -c "echo                                      >> ${taosd_service_config}"
    ${csudo} bash -c "echo '[Service]'                          >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Type=simple'                        >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'ExecStart=/usr/bin/taosd'           >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'LimitNOFILE=infinity'               >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'LimitNPROC=infinity'                >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'LimitCORE=infinity'                 >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'TimeoutStartSec=0'                  >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'StandardOutput=null'                >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Restart=always'                     >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'StartLimitBurst=3'                  >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'StartLimitInterval=60s'             >> ${taosd_service_config}"
    ${csudo} bash -c "echo                                      >> ${taosd_service_config}"
    ${csudo} bash -c "echo '[Install]'                          >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'WantedBy=multi-user.target'         >> ${taosd_service_config}"
    ${csudo} systemctl enable taosd
H
hzcheng 已提交
249 250 251
}

function install_service() {
P
plum-lihui 已提交
252
    if ((${service_mod}==0)); then
H
hzcheng 已提交
253
        install_service_on_systemd
P
plum-lihui 已提交
254
    elif ((${service_mod}==1)); then
H
hzcheng 已提交
255
        install_service_on_sysvinit
P
plum-lihui 已提交
256 257 258
    else
        # manual start taosd
        kill_taosd
H
hzcheng 已提交
259 260 261 262 263 264 265
    fi
}

function install_TDengine() {
    echo -e "${GREEN}Start to install TDEngine...${NC}"

    #install log and data dir , then ln to /usr/local/taos
P
plum-lihui 已提交
266 267
    ${csudo} mkdir -p ${log_dir} && ${csudo} chmod 777 ${log_dir}
    ${csudo} mkdir -p ${data_dir} 
H
hzcheng 已提交
268
    
P
plum-lihui 已提交
269 270
    ${csudo} rm -rf ${log_link_dir}   || :
    ${csudo} rm -rf ${data_link_dir}  || :
H
hzcheng 已提交
271
    
P
plum-lihui 已提交
272 273
    ${csudo} ln -s ${log_dir} ${log_link_dir}     || :
    ${csudo} ln -s ${data_dir} ${data_link_dir}   || :
H
hzcheng 已提交
274 275 276 277 278 279 280 281 282
    
    # Install include, lib, binary and service
    install_include
    install_lib
    install_bin
    install_service
    install_config	

    # Ask if to start the service
H
Hui Li 已提交
283 284
    #echo
    #echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
H
hzcheng 已提交
285 286
    echo
    echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
P
plum-lihui 已提交
287 288 289 290 291 292 293
    if ((${service_mod}==0)); then
        echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} systemctl start taosd${NC}"
    elif ((${service_mod}==1)); then
        echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} update-rc.d taosd default  ${RED} for the first time${NC}"
        echo -e "                      : ${csudo} service taosd start ${RED} after${NC}"
    else 
        echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
H
hzcheng 已提交
294 295 296
    fi

    echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
H
Hui Li 已提交
297
    
H
Hui Li 已提交
298 299 300 301 302
    if [ ! -z "$firstEp" ]; then
        echo		    
	echo -e "${GREEN_DARK}Please run${NC}: taos -h $firstEp${GREEN_DARK} to login into cluster, then${NC}"
	echo -e "${GREEN_DARK}execute ${NC}: create dnode 'newDnodeFQDN:port'; ${GREEN_DARK}to add this new node${NC}"
        echo
H
Hui Li 已提交
303
    fi
H
hzcheng 已提交
304 305 306 307 308 309 310
    echo
    echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
}


## ==============================Main program starts from here============================
install_TDengine