install.sh 27.1 KB
Newer Older
H
hzcheng 已提交
1 2
#!/bin/bash
#
L
lihui 已提交
3
# This file is used to install database on linux systems. The operating system
H
hzcheng 已提交
4 5 6 7 8
# is required to use systemd to manage services at boot

set -e
#set -x

L
lihui 已提交
9 10
verMode=edge
pagMode=full
S
slguan 已提交
11

H
hzcheng 已提交
12
# -----------------------Variables definition---------------------
F
Frozen 已提交
13
script_dir=$(dirname $(readlink -f "$0"))
H
hzcheng 已提交
14 15 16 17 18 19 20 21 22 23 24
# 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"

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
inc_link_dir="/usr/include"

#install main path
install_main_dir="/usr/local/taos"

# old bin dir
bin_dir="/usr/local/taos/bin"

S
slguan 已提交
34
# v1.5 jar dir
H
Hui Li 已提交
35
#v15_java_app_dir="/usr/local/lib/taos"
S
slguan 已提交
36

H
hzcheng 已提交
37
service_config_dir="/etc/systemd/system"
S
slguan 已提交
38 39
nginx_port=6060
nginx_dir="/usr/local/nginxd"
H
hzcheng 已提交
40 41 42 43 44 45 46 47

# 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 已提交
48 49 50 51
csudo=""
if command -v sudo > /dev/null; then
    csudo="sudo"
fi
H
hzcheng 已提交
52

S
slguan 已提交
53 54
update_flag=0

L
lihui 已提交
55
initd_mod=0
P
plum-lihui 已提交
56 57 58
service_mod=2
if pidof systemd &> /dev/null; then
    service_mod=0
L
lihui 已提交
59
elif $(which service &> /dev/null); then    
P
plum-lihui 已提交
60
    service_mod=1
L
lihui 已提交
61 62 63 64 65 66 67 68 69 70
    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 已提交
71 72
else 
    service_mod=2
H
hzcheng 已提交
73 74
fi

L
lihui 已提交
75 76 77 78 79 80 81 82

# get the operating system type for using the corresponding init file
# ubuntu/debian(deb), centos/fedora(rpm), others: opensuse, redhat, ..., no verification
#osinfo=$(awk -F= '/^NAME/{print $2}' /etc/os-release)
osinfo=$(cat /etc/os-release | grep "NAME" | cut -d '"' -f2)
#echo "osinfo: ${osinfo}"
os_type=0
if echo $osinfo | grep -qwi "ubuntu" ; then
H
Hui Li 已提交
83
#  echo "This is ubuntu system"
L
lihui 已提交
84 85
  os_type=1
elif echo $osinfo | grep -qwi "debian" ; then
H
Hui Li 已提交
86
#  echo "This is debian system"
L
lihui 已提交
87 88
  os_type=1
elif echo $osinfo | grep -qwi "Kylin" ; then
H
Hui Li 已提交
89
#  echo "This is Kylin system"
L
lihui 已提交
90 91
  os_type=1
elif  echo $osinfo | grep -qwi "centos" ; then
H
Hui Li 已提交
92
#  echo "This is centos system"
L
lihui 已提交
93 94
  os_type=2
elif echo $osinfo | grep -qwi "fedora" ; then
H
Hui Li 已提交
95
#  echo "This is fedora system"
L
lihui 已提交
96 97
  os_type=2
else
L
lihui 已提交
98 99 100
  echo "${osinfo}: This is an officially unverified linux system, If there are any problems with the installation and operation, "
  echo "please feel free to contact taosdata.com for support."
  os_type=1
L
lihui 已提交
101 102
fi

H
Hui Li 已提交
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138

# =============================  get input parameters =================================================

# install.sh -v [server | client]  -e [yes | no] -i [systemd | service | ...]

# set parameters by default value
interactiveFqdn=yes   # [yes | no]
verType=server        # [server | client]
initType=systemd      # [systemd | service | ...]

while getopts "hv:e:i:" arg
do
  case $arg in
    e)
      #echo "interactiveFqdn=$OPTARG"
      interactiveFqdn=$( echo $OPTARG )
      ;;
    v)
      #echo "verType=$OPTARG"
      verType=$(echo $OPTARG)
      ;;
    i)
      #echo "initType=$OPTARG"
      initType=$(echo $OPTARG)
      ;;
    h)
      echo "Usage: `basename $0` -v [server | client]  -e [yes | no]"
      exit 0
      ;;
    ?) #unknow option 
      echo "unkonw argument"
      exit 1
      ;;
  esac
done

H
Hui Li 已提交
139
#echo "verType=${verType} interactiveFqdn=${interactiveFqdn}"
H
Hui Li 已提交
140

P
plum-lihui 已提交
141 142
function kill_taosd() {
  pid=$(ps -ef | grep "taosd" | grep -v "grep" | awk '{print $2}')
L
lihui 已提交
143 144 145
  if [ -n "$pid" ]; then
    ${csudo} kill -9 $pid   || :
  fi
P
plum-lihui 已提交
146 147
}

H
hzcheng 已提交
148 149
function install_main_path() {
    #create install main dir and all sub dir
P
plum-lihui 已提交
150 151 152 153 154 155 156 157 158
    ${csudo} rm -rf ${install_main_dir}    || :
    ${csudo} mkdir -p ${install_main_dir}  
    ${csudo} mkdir -p ${install_main_dir}/cfg
    ${csudo} mkdir -p ${install_main_dir}/bin    
    ${csudo} mkdir -p ${install_main_dir}/connector
    ${csudo} mkdir -p ${install_main_dir}/driver
    ${csudo} mkdir -p ${install_main_dir}/examples
    ${csudo} mkdir -p ${install_main_dir}/include
    ${csudo} mkdir -p ${install_main_dir}/init.d
S
slguan 已提交
159 160 161
    if [ "$verMode" == "cluster" ]; then
        ${csudo} mkdir -p ${nginx_dir}
    fi
H
hzcheng 已提交
162 163 164 165
}

function install_bin() {
    # Remove links
P
plum-lihui 已提交
166 167
    ${csudo} rm -f ${bin_link_dir}/taos     || :
    ${csudo} rm -f ${bin_link_dir}/taosd    || :
S
slguan 已提交
168
    ${csudo} rm -f ${bin_link_dir}/taosdemo || :
P
plum-lihui 已提交
169
    ${csudo} rm -f ${bin_link_dir}/rmtaos   || :
H
Hui Li 已提交
170
    ${csudo} rm -f ${bin_link_dir}/tarbitrator   || :
H
Hui Li 已提交
171
    ${csudo} rm -f ${bin_link_dir}/set_core   || :
H
hzcheng 已提交
172

P
plum-lihui 已提交
173
    ${csudo} cp -r ${script_dir}/bin/* ${install_main_dir}/bin && ${csudo} chmod 0555 ${install_main_dir}/bin/*
H
hzcheng 已提交
174 175

    #Make link
H
Hui Li 已提交
176 177 178 179
    [ -x ${install_main_dir}/bin/taos ] && ${csudo} ln -s ${install_main_dir}/bin/taos ${bin_link_dir}/taos                      || :
    [ -x ${install_main_dir}/bin/taosd ] && ${csudo} ln -s ${install_main_dir}/bin/taosd ${bin_link_dir}/taosd                   || :
    [ -x ${install_main_dir}/bin/taosdemo ] && ${csudo} ln -s ${install_main_dir}/bin/taosdemo ${bin_link_dir}/taosdemo          || :
    [ -x ${install_main_dir}/bin/remove.sh ] && ${csudo} ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/rmtaos          || :
H
Hui Li 已提交
180
    [ -x ${install_main_dir}/bin/set_core.sh ] && ${csudo} ln -s ${install_main_dir}/bin/set_core.sh ${bin_link_dir}/set_core    || :
H
Hui Li 已提交
181
    [ -x ${install_main_dir}/bin/tarbitrator ] && ${csudo} ln -s ${install_main_dir}/bin/tarbitrator ${bin_link_dir}/tarbitrator || :
S
slguan 已提交
182 183 184 185 186 187

    if [ "$verMode" == "cluster" ]; then
        ${csudo} cp -r ${script_dir}/nginxd/* ${nginx_dir} && ${csudo} chmod 0555 ${nginx_dir}/*
        ${csudo} mkdir -p ${nginx_dir}/logs
        ${csudo} chmod 777 ${nginx_dir}/sbin/nginx
    fi
H
hzcheng 已提交
188 189 190 191
}

function install_lib() {
    # Remove links
L
lihui 已提交
192
    ${csudo} rm -f ${lib_link_dir}/libtaos.*         || :
H
Hui Li 已提交
193
    ${csudo} rm -f ${lib64_link_dir}/libtaos.*       || :
H
Hui Li 已提交
194
    #${csudo} rm -rf ${v15_java_app_dir}              || :
H
hzcheng 已提交
195

P
plum-lihui 已提交
196
    ${csudo} cp -rf ${script_dir}/driver/* ${install_main_dir}/driver && ${csudo} chmod 777 ${install_main_dir}/driver/*  
H
hzcheng 已提交
197
    
P
plum-lihui 已提交
198 199
    ${csudo} ln -s ${install_main_dir}/driver/libtaos.* ${lib_link_dir}/libtaos.so.1
    ${csudo} ln -s ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
S
slguan 已提交
200
    
H
Hui Li 已提交
201 202 203 204
    if [ -d ${lib64_link_dir} ]; then
      ${csudo} ln -s ${install_main_dir}/driver/libtaos.* ${lib64_link_dir}/libtaos.so.1       || :
      ${csudo} ln -s ${lib64_link_dir}/libtaos.so.1 ${lib64_link_dir}/libtaos.so               || :
    fi
H
Hui Li 已提交
205
        
H
Hui Li 已提交
206 207 208 209 210 211
	  #if [ "$verMode" == "cluster" ]; then
    #    # Compatible with version 1.5
    #    ${csudo} mkdir -p ${v15_java_app_dir}
    #    ${csudo} ln -s ${install_main_dir}/connector/taos-jdbcdriver-1.0.2-dist.jar ${v15_java_app_dir}/JDBCDriver-1.0.2-dist.jar
    #    ${csudo} chmod 777 ${v15_java_app_dir} || :
    #fi
H
Hui Li 已提交
212 213
    
    ${csudo} ldconfig
H
hzcheng 已提交
214 215 216
}

function install_header() {
L
lihui 已提交
217
    ${csudo} rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taoserror.h    || :
P
plum-lihui 已提交
218 219
    ${csudo} cp -f ${script_dir}/inc/* ${install_main_dir}/include && ${csudo} chmod 644 ${install_main_dir}/include/*    
    ${csudo} ln -s ${install_main_dir}/include/taos.h ${inc_link_dir}/taos.h
L
lihui 已提交
220
    ${csudo} ln -s ${install_main_dir}/include/taoserror.h ${inc_link_dir}/taoserror.h
H
hzcheng 已提交
221 222 223
}

function install_config() {
P
plum-lihui 已提交
224
    #${csudo} rm -f ${install_main_dir}/cfg/taos.cfg     || :
H
hzcheng 已提交
225
    
L
lihui 已提交
226
    if [ ! -f ${cfg_install_dir}/taos.cfg ]; then
P
plum-lihui 已提交
227 228 229
        ${csudo} mkdir -p ${cfg_install_dir}
        [ -f ${script_dir}/cfg/taos.cfg ] && ${csudo} cp ${script_dir}/cfg/taos.cfg ${cfg_install_dir}
        ${csudo} chmod 644 ${cfg_install_dir}/*
H
hzcheng 已提交
230 231
    fi 
    
P
plum-lihui 已提交
232
    ${csudo} cp -f ${script_dir}/cfg/taos.cfg ${install_main_dir}/cfg/taos.cfg.org
L
lihui 已提交
233
    ${csudo} ln -s ${cfg_install_dir}/taos.cfg ${install_main_dir}/cfg
S
slguan 已提交
234

H
Hui Li 已提交
235
    [ ! -z $1 ] && return 0 || : # only install client
S
slguan 已提交
236
    
H
Hui Li 已提交
237 238 239 240 241 242 243
    if ((${update_flag}==1)); then
        return 0
    fi
    
    if [ "$interactiveFqdn" == "no" ]; then
        return 0
    fi
S
slguan 已提交
244

H
Hui Li 已提交
245 246 247 248 249 250 251
    #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 已提交
252
    echo -e -n "${GREEN}Enter FQDN:port (like h1.taosdata.com:6030) of an existing TDengine cluster node to join${NC}"
H
Hui Li 已提交
253
    echo
H
Hui Li 已提交
254
    echo -e -n "${GREEN}OR leave it blank to build one${NC}:"
H
Hui Li 已提交
255
    read firstEp
H
Hui Li 已提交
256 257 258 259 260 261
    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    
S
slguan 已提交
262
                break
H
Hui Li 已提交
263 264 265 266 267 268 269
            #else
            #    read -p "Please enter the correct FQDN:port: " firstEp
            #fi
        else
            break
        fi
    done	
H
hzcheng 已提交
270 271
}

L
lihui 已提交
272

H
hzcheng 已提交
273
function install_log() {
P
plum-lihui 已提交
274 275
    ${csudo} rm -rf ${log_dir}  || :
    ${csudo} mkdir -p ${log_dir} && ${csudo} chmod 777 ${log_dir}
H
hzcheng 已提交
276
    
P
plum-lihui 已提交
277
    ${csudo} ln -s ${log_dir} ${install_main_dir}/log
H
hzcheng 已提交
278 279 280
}

function install_data() {
P
plum-lihui 已提交
281
    ${csudo} mkdir -p ${data_dir}
H
hzcheng 已提交
282
    
P
plum-lihui 已提交
283
    ${csudo} ln -s ${data_dir} ${install_main_dir}/data    
H
hzcheng 已提交
284 285 286
}

function install_connector() {
P
plum-lihui 已提交
287
    ${csudo} cp -rf ${script_dir}/connector/* ${install_main_dir}/connector
H
hzcheng 已提交
288 289 290
}

function install_examples() {
S
slguan 已提交
291 292 293
    if [ -d ${script_dir}/examples ]; then
        ${csudo} cp -rf ${script_dir}/examples/* ${install_main_dir}/examples
    fi
H
hzcheng 已提交
294 295 296
}

function clean_service_on_sysvinit() {
L
lihui 已提交
297 298 299
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
    #${csudo} sed -i "\|${restart_config_str}|d" /etc/inittab || :    
    
H
hzcheng 已提交
300
    if pidof taosd &> /dev/null; then
P
plum-lihui 已提交
301
        ${csudo} service taosd stop || :
H
hzcheng 已提交
302
    fi
H
Hui Li 已提交
303 304 305 306
    
    if pidof tarbitrator &> /dev/null; then
        ${csudo} service tarbitratord stop || :
    fi
L
lihui 已提交
307 308 309

    if ((${initd_mod}==1)); then
        ${csudo} chkconfig --del taosd || :
H
Hui Li 已提交
310
        ${csudo} chkconfig --del tarbitratord || :
L
lihui 已提交
311 312
    elif ((${initd_mod}==2)); then
        ${csudo} insserv -r taosd || :
H
Hui Li 已提交
313
        ${csudo} insserv -r tarbitratord || :
L
lihui 已提交
314 315
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d -f taosd remove || :
H
Hui Li 已提交
316
        ${csudo} update-rc.d -f tarbitratord remove || :
L
lihui 已提交
317 318
    fi
    
P
plum-lihui 已提交
319
    ${csudo} rm -f ${service_config_dir}/taosd || :
H
Hui Li 已提交
320
    ${csudo} rm -f ${service_config_dir}/tarbitratord || :
L
lihui 已提交
321 322 323 324
    
    if $(which init &> /dev/null); then
        ${csudo} init q || :
    fi
H
hzcheng 已提交
325 326 327 328 329 330 331 332
}

function install_service_on_sysvinit() {
    clean_service_on_sysvinit

    sleep 1

    # Install taosd service
L
lihui 已提交
333 334 335 336

    if ((${os_type}==1)); then
        ${csudo} cp -f ${script_dir}/init.d/taosd.deb ${install_main_dir}/init.d/taosd
        ${csudo} cp    ${script_dir}/init.d/taosd.deb ${service_config_dir}/taosd && ${csudo} chmod a+x ${service_config_dir}/taosd
H
Hui Li 已提交
337 338
        ${csudo} cp -f ${script_dir}/init.d/tarbitratord.deb ${install_main_dir}/init.d/tarbitratord
        ${csudo} cp    ${script_dir}/init.d/tarbitratord.deb ${service_config_dir}/tarbitratord && ${csudo} chmod a+x ${service_config_dir}/tarbitratord
L
lihui 已提交
339 340 341
    elif ((${os_type}==2)); then
        ${csudo} cp -f ${script_dir}/init.d/taosd.rpm ${install_main_dir}/init.d/taosd
        ${csudo} cp    ${script_dir}/init.d/taosd.rpm ${service_config_dir}/taosd && ${csudo} chmod a+x ${service_config_dir}/taosd
H
Hui Li 已提交
342 343
        ${csudo} cp -f ${script_dir}/init.d/tarbitratord.rpm ${install_main_dir}/init.d/tarbitratord
        ${csudo} cp    ${script_dir}/init.d/tarbitratord.rpm ${service_config_dir}/tarbitratord && ${csudo} chmod a+x ${service_config_dir}/tarbitratord
L
lihui 已提交
344 345 346 347 348 349 350 351
    fi
    
    #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 || :
H
Hui Li 已提交
352 353
        ${csudo} chkconfig --add tarbitratord || :
        ${csudo} chkconfig --level 2345 tarbitratord on || :
L
lihui 已提交
354 355 356
    elif ((${initd_mod}==2)); then
        ${csudo} insserv taosd || :
        ${csudo} insserv -d taosd || :
H
Hui Li 已提交
357 358
        ${csudo} insserv tarbitratord || :
        ${csudo} insserv -d tarbitratord || :
L
lihui 已提交
359 360
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d taosd defaults || :
H
Hui Li 已提交
361
        ${csudo} update-rc.d tarbitratord defaults || :
L
lihui 已提交
362
    fi
H
hzcheng 已提交
363 364 365 366 367 368 369
}

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

    if systemctl is-active --quiet taosd; then
        echo "TDengine is running, stopping it..."
P
plum-lihui 已提交
370
        ${csudo} systemctl stop taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
371
    fi
P
plum-lihui 已提交
372
    ${csudo} systemctl disable taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
373

P
plum-lihui 已提交
374
    ${csudo} rm -f ${taosd_service_config}
S
slguan 已提交
375 376 377 378 379 380 381 382 383 384

    if [ "$verMode" == "cluster" ]; then
        nginx_service_config="${service_config_dir}/nginxd.service"
	
        if systemctl is-active --quiet nginxd; then
            echo "Nginx for TDengine is running, stopping it..."
            ${csudo} systemctl stop nginxd &> /dev/null || echo &> /dev/null
        fi
        ${csudo} systemctl disable nginxd &> /dev/null || echo &> /dev/null
	
H
Hui Li 已提交
385 386 387 388 389 390 391 392 393 394 395
        ${csudo} rm -f ${nginx_service_config}              

        tarbitratord_service_config="${service_config_dir}/tarbitratord.service"
        if systemctl is-active --quiet tarbitratord; then
            echo "tarbitrator is running, stopping it..."
            ${csudo} systemctl stop tarbitratord &> /dev/null || echo &> /dev/null
        fi
        ${csudo} systemctl disable tarbitratord &> /dev/null || echo &> /dev/null

        ${csudo} rm -f ${tarbitratord_service_config}
	  fi
S
slguan 已提交
396
}
H
hzcheng 已提交
397 398 399 400 401 402 403 404

# 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 已提交
405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424
    ${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
S
slguan 已提交
425 426

    if [ "$verMode" == "cluster" ]; then		
H
Hui Li 已提交
427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448
    
        tarbitratord_service_config="${service_config_dir}/tarbitratord.service"

        ${csudo} bash -c "echo '[Unit]'                                  >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'Description=TDengine arbitrator service' >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'After=network-online.target'             >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'Wants=network-online.target'             >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo                                           >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo '[Service]'                               >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'Type=simple'                             >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'ExecStart=/usr/bin/tarbitrator'          >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'LimitNOFILE=infinity'                    >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'LimitNPROC=infinity'                     >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'LimitCORE=infinity'                      >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'TimeoutStartSec=0'                       >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'StandardOutput=null'                     >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'Restart=always'                          >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'StartLimitBurst=3'                       >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'StartLimitInterval=60s'                  >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo                                           >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo '[Install]'                               >> ${tarbitratord_service_config}"
        ${csudo} bash -c "echo 'WantedBy=multi-user.target'              >> ${tarbitratord_service_config}"
H
Hui Li 已提交
449
        # ${csudo} systemctl enable tarbitratord  
H
Hui Li 已提交
450
    
S
slguan 已提交
451
        nginx_service_config="${service_config_dir}/nginxd.service"
H
Hui Li 已提交
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
        ${csudo} bash -c "echo '[Unit]'                                             >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'Description=Nginx For TDengine Service'             >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'After=network-online.target'                        >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'Wants=network-online.target'                        >> ${nginx_service_config}"
        ${csudo} bash -c "echo                                                      >> ${nginx_service_config}"
        ${csudo} bash -c "echo '[Service]'                                          >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'Type=forking'                                       >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'PIDFile=/usr/local/nginxd/logs/nginx.pid'           >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'ExecStart=/usr/local/nginxd/sbin/nginx'             >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'ExecStop=/usr/local/nginxd/sbin/nginx -s stop'      >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'LimitNOFILE=infinity'                               >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'LimitNPROC=infinity'                                >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'LimitCORE=infinity'                                 >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'TimeoutStartSec=0'                                  >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'StandardOutput=null'                                >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'Restart=always'                                     >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'StartLimitBurst=3'                                  >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'StartLimitInterval=60s'                             >> ${nginx_service_config}"
        ${csudo} bash -c "echo                                                      >> ${nginx_service_config}"
        ${csudo} bash -c "echo '[Install]'                                          >> ${nginx_service_config}"
        ${csudo} bash -c "echo 'WantedBy=multi-user.target'                         >> ${nginx_service_config}"
S
slguan 已提交
473 474 475 476 477 478
        if ! ${csudo} systemctl enable nginxd &> /dev/null; then
            ${csudo} systemctl daemon-reexec
            ${csudo} systemctl enable nginxd
        fi
        ${csudo} systemctl start nginxd
	fi
H
hzcheng 已提交
479 480 481
}

function install_service() {
P
plum-lihui 已提交
482
    if ((${service_mod}==0)); then
H
hzcheng 已提交
483
        install_service_on_systemd
P
plum-lihui 已提交
484
    elif ((${service_mod}==1)); then
H
hzcheng 已提交
485
        install_service_on_sysvinit
P
plum-lihui 已提交
486
    else
L
lihui 已提交
487
        # must manual stop taosd
P
plum-lihui 已提交
488
        kill_taosd
H
hzcheng 已提交
489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522
    fi
}

vercomp () {
    if [[ $1 == $2 ]]; then
        return 0
    fi
    local IFS=.
    local i ver1=($1) ver2=($2)
    # fill empty fields in ver1 with zeros
    for ((i=${#ver1[@]}; i<${#ver2[@]}; i++)); do
        ver1[i]=0
    done

    for ((i=0; i<${#ver1[@]}; i++)); do
        if [[ -z ${ver2[i]} ]]
        then
            # fill empty fields in ver2 with zeros
            ver2[i]=0
        fi
        if ((10#${ver1[i]} > 10#${ver2[i]}))
        then
            return 1
        fi
        if ((10#${ver1[i]} < 10#${ver2[i]}))
        then
            return 2
        fi
    done
    return 0
}

function is_version_compatible() {

L
lihui 已提交
523
    curr_version=$(${bin_dir}/taosd -V | head -1 | cut -d ' ' -f 3)
H
hzcheng 已提交
524

L
lihui 已提交
525
    min_compatible_version=$(${script_dir}/bin/taosd -V | head -1 | cut -d ' ' -f 5)
H
hzcheng 已提交
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540

    vercomp $curr_version $min_compatible_version
    case $? in
        0) return 0;;
        1) return 0;;
        2) return 1;;
    esac
}

function update_TDengine() {
    # Start to update
    if [ ! -e taos.tar.gz ]; then
        echo "File taos.tar.gz does not exist"
        exit 1
    fi
L
lihui 已提交
541
    tar -zxf taos.tar.gz
H
hzcheng 已提交
542 543 544 545 546 547 548

    # Check if version compatible
    if ! is_version_compatible; then
        echo -e "${RED}Version incompatible${NC}"
        return 1
    fi

L
lihui 已提交
549
    echo -e "${GREEN}Start to update TDengine...${NC}"
H
hzcheng 已提交
550 551
    # Stop the service if running
    if pidof taosd &> /dev/null; then
P
plum-lihui 已提交
552 553 554 555
        if ((${service_mod}==0)); then
            ${csudo} systemctl stop taosd || :
        elif ((${service_mod}==1)); then
            ${csudo} service taosd stop || :
H
hzcheng 已提交
556
        else
P
plum-lihui 已提交
557
            kill_taosd
H
hzcheng 已提交
558 559 560 561 562 563 564 565 566
        fi
        sleep 1
    fi
    
    install_main_path

    install_log
    install_header
    install_lib
L
lihui 已提交
567 568 569
    if [ "$pagMode" != "lite" ]; then
      install_connector
    fi
H
hzcheng 已提交
570 571 572 573
    install_examples
    if [ -z $1 ]; then
        install_bin
        install_service
H
Hui Li 已提交
574
        install_config		
S
slguan 已提交
575
		
H
Hui Li 已提交
576
        openresty_work=false
H
Hui Li 已提交
577
		    if [ "$verMode" == "cluster" ]; then    
S
slguan 已提交
578 579 580 581 582 583 584 585 586 587
            # Check if openresty is installed
            # Check if nginx is installed successfully
            if type curl &> /dev/null; then
                if curl -sSf http://127.0.0.1:${nginx_port} &> /dev/null; then
                    echo -e "\033[44;32;1mNginx for TDengine is updated successfully!${NC}"
                    openresty_work=true
                else
                    echo -e "\033[44;31;5mNginx for TDengine does not work! Please try again!\033[0m"
                fi
            fi
H
Hui Li 已提交
588
		    fi 
H
hzcheng 已提交
589

H
Hui Li 已提交
590 591
        #echo
        #echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
H
hzcheng 已提交
592 593
        echo
        echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
P
plum-lihui 已提交
594 595 596
        if ((${service_mod}==0)); then
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} systemctl start taosd${NC}"
        elif ((${service_mod}==1)); then
L
lihui 已提交
597
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} service taosd start${NC}"
H
hzcheng 已提交
598
        else
P
plum-lihui 已提交
599
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
H
hzcheng 已提交
600 601
        fi

H
Hui Li 已提交
602 603
        if [ ${openresty_work} = 'true' ]; then
            echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell OR from ${GREEN_UNDERLINE}http://127.0.0.1:${nginx_port}${NC}"
S
slguan 已提交
604
        else
H
Hui Li 已提交
605
            echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
S
slguan 已提交
606
        fi
H
Hui Li 已提交
607
                
H
hzcheng 已提交
608 609 610
        echo
        echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
    else
L
lihui 已提交
611
        install_bin
H
hzcheng 已提交
612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
        install_config

        echo
        echo -e "\033[44;32;1mTDengine client is updated successfully!${NC}"
    fi

    rm -rf $(tar -tf taos.tar.gz)
}

function install_TDengine() {
    # Start to install
    if [ ! -e taos.tar.gz ]; then
        echo "File taos.tar.gz does not exist"
        exit 1
    fi
L
lihui 已提交
627
    tar -zxf taos.tar.gz
H
hzcheng 已提交
628

L
lihui 已提交
629
    echo -e "${GREEN}Start to install TDengine...${NC}"
H
hzcheng 已提交
630 631 632 633 634 635 636 637 638 639
    
	  install_main_path
	   
    if [ -z $1 ]; then
        install_data
    fi 
    
    install_log 
    install_header
    install_lib
L
lihui 已提交
640 641 642
    if [ "$pagMode" != "lite" ]; then
      install_connector
    fi
H
hzcheng 已提交
643 644 645 646 647 648
    install_examples

    if [ -z $1 ]; then # install service and client
        # For installing new
        install_bin
        install_service
S
slguan 已提交
649

H
Hui Li 已提交
650 651
        openresty_work=false
        if [ "$verMode" == "cluster" ]; then
S
slguan 已提交
652 653 654 655 656 657 658 659 660 661 662
            # Check if nginx is installed successfully
            if type curl &> /dev/null; then
                if curl -sSf http://127.0.0.1:${nginx_port} &> /dev/null; then
                    echo -e "\033[44;32;1mNginx for TDengine is installed successfully!${NC}"
                    openresty_work=true
                else
                    echo -e "\033[44;31;5mNginx for TDengine does not work! Please try again!\033[0m"
                fi
            fi
        fi
		
H
hzcheng 已提交
663 664 665
        install_config	

        # Ask if to start the service
H
Hui Li 已提交
666 667
        #echo
        #echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
H
hzcheng 已提交
668 669
        echo
        echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
P
plum-lihui 已提交
670 671 672
        if ((${service_mod}==0)); then
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} systemctl start taosd${NC}"
        elif ((${service_mod}==1)); then
L
lihui 已提交
673
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} service taosd start${NC}"
H
hzcheng 已提交
674
        else
L
lihui 已提交
675
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: taosd${NC}"
H
Hui Li 已提交
676 677 678 679 680 681
        fi		

        if [ ${openresty_work} = 'true' ]; then
             echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell OR from ${GREEN_UNDERLINE}http://127.0.0.1:${nginx_port}${NC}"
        else
             echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
H
hzcheng 已提交
682
        fi
S
slguan 已提交
683
		
H
Hui Li 已提交
684 685 686 687 688
        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
S
slguan 已提交
689
        fi
H
hzcheng 已提交
690
        echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
H
Hui Li 已提交
691
        echo       
H
hzcheng 已提交
692 693 694 695 696 697 698 699 700 701 702 703 704
    else # Only install client
        install_bin
        install_config

        echo
        echo -e "\033[44;32;1mTDengine client is installed successfully!${NC}"
    fi

    rm -rf $(tar -tf taos.tar.gz)
}


## ==============================Main program starts from here============================
H
Hui Li 已提交
705
if [ "$verType" == "server" ]; then
H
hzcheng 已提交
706 707
    # Install server and client
    if [ -x ${bin_dir}/taosd ]; then
S
slguan 已提交
708
        update_flag=1
H
hzcheng 已提交
709 710 711 712
        update_TDengine
    else
        install_TDengine
    fi
H
Hui Li 已提交
713 714
elif [ "$verType" == "client" ]; then
    interactiveFqdn=no
H
hzcheng 已提交
715 716
    # Only install client
    if [ -x ${bin_dir}/taos ]; then
S
slguan 已提交
717
        update_flag=1
H
hzcheng 已提交
718 719 720 721
        update_TDengine client
    else
        install_TDengine client
    fi
H
Hui Li 已提交
722 723
else 
    echo  "please input correct verType"   
H
hzcheng 已提交
724
fi