install.sh 27.7 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 25 26 27 28 29 30 31 32
# 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"
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 已提交
33 34 35
# v1.5 jar dir
v15_java_app_dir="/usr/local/lib/taos"

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

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

S
slguan 已提交
52 53
update_flag=0

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

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

# 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
L
lihui 已提交
82
  echo "This is ubuntu system"
L
lihui 已提交
83 84
  os_type=1
elif echo $osinfo | grep -qwi "debian" ; then
L
lihui 已提交
85
  echo "This is debian system"
L
lihui 已提交
86 87
  os_type=1
elif echo $osinfo | grep -qwi "Kylin" ; then
L
lihui 已提交
88
  echo "This is Kylin system"
L
lihui 已提交
89 90
  os_type=1
elif  echo $osinfo | grep -qwi "centos" ; then
L
lihui 已提交
91
  echo "This is centos system"
L
lihui 已提交
92 93
  os_type=2
elif echo $osinfo | grep -qwi "fedora" ; then
L
lihui 已提交
94
  echo "This is fedora system"
L
lihui 已提交
95 96
  os_type=2
else
L
lihui 已提交
97 98 99
  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 已提交
100 101
fi

H
Hui Li 已提交
102 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 139

# =============================  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

echo "verType=${verType} interactiveFqdn=${interactiveFqdn}"

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

H
hzcheng 已提交
147 148
function install_main_path() {
    #create install main dir and all sub dir
P
plum-lihui 已提交
149 150 151 152 153 154 155 156 157
    ${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 已提交
158 159 160
    if [ "$verMode" == "cluster" ]; then
        ${csudo} mkdir -p ${nginx_dir}
    fi
H
hzcheng 已提交
161 162 163 164
}

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

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

    #Make link
H
Hui Li 已提交
175 176 177 178 179 180
    [ -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/taosdump ] && ${csudo} ln -s ${install_main_dir}/bin/taosdump ${bin_link_dir}/taosdump          || :
    [ -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          || :
    [ -x ${install_main_dir}/bin/tarbitrator ] && ${csudo} ln -s ${install_main_dir}/bin/tarbitrator ${bin_link_dir}/tarbitrator || :
S
slguan 已提交
181 182 183 184 185 186

    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 已提交
187 188 189 190
}

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

P
plum-lihui 已提交
194
    ${csudo} cp -rf ${script_dir}/driver/* ${install_main_dir}/driver && ${csudo} chmod 777 ${install_main_dir}/driver/*  
H
hzcheng 已提交
195
    
P
plum-lihui 已提交
196 197
    ${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 已提交
198 199 200 201 202 203 204
    
	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
hzcheng 已提交
205 206 207
}

function install_header() {
L
lihui 已提交
208
    ${csudo} rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taoserror.h    || :
P
plum-lihui 已提交
209 210
    ${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 已提交
211
    ${csudo} ln -s ${install_main_dir}/include/taoserror.h ${inc_link_dir}/taoserror.h
H
hzcheng 已提交
212 213 214
}

function install_config() {
P
plum-lihui 已提交
215
    #${csudo} rm -f ${install_main_dir}/cfg/taos.cfg     || :
H
hzcheng 已提交
216
    
L
lihui 已提交
217
    if [ ! -f ${cfg_install_dir}/taos.cfg ]; then
P
plum-lihui 已提交
218 219 220
        ${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 已提交
221 222
    fi 
    
P
plum-lihui 已提交
223
    ${csudo} cp -f ${script_dir}/cfg/taos.cfg ${install_main_dir}/cfg/taos.cfg.org
L
lihui 已提交
224
    ${csudo} ln -s ${cfg_install_dir}/taos.cfg ${install_main_dir}/cfg
S
slguan 已提交
225 226 227 228 229 230 231

    if [ "$verMode" == "cluster" ]; then
        [ ! -z $1 ] && return 0 || : # only install client
    
        if ((${update_flag}==1)); then
            return 0
        fi
H
Hui Li 已提交
232 233 234 235
        
        if [ "$interactiveFqdn" == "no" ]; then
            return 0
        fi
S
slguan 已提交
236

H
Hui Li 已提交
237 238 239 240
        #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}$"
S
slguan 已提交
241

H
Hui Li 已提交
242
        # first full-qualified domain name (FQDN) for TDengine cluster system
S
slguan 已提交
243
        echo
H
Hui Li 已提交
244 245
        echo -e -n "${GREEN}Enter the FQDN of an existing TDengine cluster node to join${NC} OR ${GREEN}leave it blank to build one${NC} :"
        read firstFqdn
S
slguan 已提交
246
        while true; do
H
Hui Li 已提交
247 248 249 250 251
            if [ ! -z "$firstFqdn" ]; then
                # check the format of the firstFqdn
                #if [[ $firstFqdn == $FQDN_PATTERN ]]; then
                    # Write the first FQDN to configuration file
                    ${csudo} sed -i -r "s/#*\s*(first\s*).*/\1$firstFqdn/" ${cfg_install_dir}/taos.cfg
S
slguan 已提交
252

H
Hui Li 已提交
253
                    # Get the second FQDN
S
slguan 已提交
254
                    echo
H
Hui Li 已提交
255 256
                    echo -e -n "${GREEN}Enter the FQDN of another node in cluster${NC} OR ${GREEN}leave it blank to skip${NC}: "
                    read secondFqdn
S
slguan 已提交
257
                    while true; do
H
Hui Li 已提交
258 259 260 261
                        if [ ! -z "$secondFqdn" ]; then
                            #if [[ $secondFqdn == $FQDN_PATTERN ]]; then
                                # Write the second FQDN to configuration file
                                ${csudo} sed -i -r "s/#*\s*(second\s*).*/\1$secondFqdn/" ${cfg_install_dir}/taos.cfg
S
slguan 已提交
262
                                break
H
Hui Li 已提交
263 264 265
                            #else
                            #    read -p "Please enter the correct FQDN: " secondFqdn
                            #fi
S
slguan 已提交
266 267 268 269 270 271
                        else
                            break
                        fi
                    done
    
                    break
H
Hui Li 已提交
272 273 274
                #else
                #    read -p "Please enter the correct FQDN: " firstFqdn
                #fi
S
slguan 已提交
275 276 277 278 279
            else
                break
            fi
        done
	
H
Hui Li 已提交
280
	  fi
H
hzcheng 已提交
281 282
}

L
lihui 已提交
283

H
hzcheng 已提交
284
function install_log() {
P
plum-lihui 已提交
285 286
    ${csudo} rm -rf ${log_dir}  || :
    ${csudo} mkdir -p ${log_dir} && ${csudo} chmod 777 ${log_dir}
H
hzcheng 已提交
287
    
P
plum-lihui 已提交
288
    ${csudo} ln -s ${log_dir} ${install_main_dir}/log
H
hzcheng 已提交
289 290 291
}

function install_data() {
P
plum-lihui 已提交
292
    ${csudo} mkdir -p ${data_dir}
H
hzcheng 已提交
293
    
P
plum-lihui 已提交
294
    ${csudo} ln -s ${data_dir} ${install_main_dir}/data    
H
hzcheng 已提交
295 296 297
}

function install_connector() {
P
plum-lihui 已提交
298
    ${csudo} cp -rf ${script_dir}/connector/* ${install_main_dir}/connector
H
hzcheng 已提交
299 300 301
}

function install_examples() {
S
slguan 已提交
302 303 304
    if [ -d ${script_dir}/examples ]; then
        ${csudo} cp -rf ${script_dir}/examples/* ${install_main_dir}/examples
    fi
H
hzcheng 已提交
305 306 307
}

function clean_service_on_sysvinit() {
L
lihui 已提交
308 309 310
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
    #${csudo} sed -i "\|${restart_config_str}|d" /etc/inittab || :    
    
H
hzcheng 已提交
311
    if pidof taosd &> /dev/null; then
P
plum-lihui 已提交
312
        ${csudo} service taosd stop || :
H
hzcheng 已提交
313
    fi
H
Hui Li 已提交
314 315 316 317
    
    if pidof tarbitrator &> /dev/null; then
        ${csudo} service tarbitratord stop || :
    fi
L
lihui 已提交
318 319 320

    if ((${initd_mod}==1)); then
        ${csudo} chkconfig --del taosd || :
H
Hui Li 已提交
321
        ${csudo} chkconfig --del tarbitratord || :
L
lihui 已提交
322 323
    elif ((${initd_mod}==2)); then
        ${csudo} insserv -r taosd || :
H
Hui Li 已提交
324
        ${csudo} insserv -r tarbitratord || :
L
lihui 已提交
325 326
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d -f taosd remove || :
H
Hui Li 已提交
327
        ${csudo} update-rc.d -f tarbitratord remove || :
L
lihui 已提交
328 329
    fi
    
P
plum-lihui 已提交
330
    ${csudo} rm -f ${service_config_dir}/taosd || :
H
Hui Li 已提交
331
    ${csudo} rm -f ${service_config_dir}/tarbitratord || :
L
lihui 已提交
332 333 334 335
    
    if $(which init &> /dev/null); then
        ${csudo} init q || :
    fi
H
hzcheng 已提交
336 337 338 339 340 341 342 343
}

function install_service_on_sysvinit() {
    clean_service_on_sysvinit

    sleep 1

    # Install taosd service
L
lihui 已提交
344 345 346 347

    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 已提交
348 349
        ${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 已提交
350 351 352
    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 已提交
353 354
        ${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 已提交
355 356 357 358 359 360 361 362
    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 已提交
363 364
        ${csudo} chkconfig --add tarbitratord || :
        ${csudo} chkconfig --level 2345 tarbitratord on || :
L
lihui 已提交
365 366 367
    elif ((${initd_mod}==2)); then
        ${csudo} insserv taosd || :
        ${csudo} insserv -d taosd || :
H
Hui Li 已提交
368 369
        ${csudo} insserv tarbitratord || :
        ${csudo} insserv -d tarbitratord || :
L
lihui 已提交
370 371
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d taosd defaults || :
H
Hui Li 已提交
372
        ${csudo} update-rc.d tarbitratord defaults || :
L
lihui 已提交
373
    fi
H
hzcheng 已提交
374 375 376 377 378 379 380
}

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 已提交
381
        ${csudo} systemctl stop taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
382
    fi
P
plum-lihui 已提交
383
    ${csudo} systemctl disable taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
384

P
plum-lihui 已提交
385
    ${csudo} rm -f ${taosd_service_config}
S
slguan 已提交
386 387 388 389 390 391 392 393 394 395

    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 已提交
396 397 398 399 400 401 402 403 404 405 406
        ${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 已提交
407
}
H
hzcheng 已提交
408 409 410 411 412 413 414 415

# 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 已提交
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435
    ${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 已提交
436 437

    if [ "$verMode" == "cluster" ]; then		
H
Hui Li 已提交
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461
    
        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}"
        ${csudo} systemctl enable tarbitratord  
    
S
slguan 已提交
462
        nginx_service_config="${service_config_dir}/nginxd.service"
H
Hui Li 已提交
463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
        ${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 已提交
484 485 486 487 488 489
        if ! ${csudo} systemctl enable nginxd &> /dev/null; then
            ${csudo} systemctl daemon-reexec
            ${csudo} systemctl enable nginxd
        fi
        ${csudo} systemctl start nginxd
	fi
H
hzcheng 已提交
490 491 492
}

function install_service() {
P
plum-lihui 已提交
493
    if ((${service_mod}==0)); then
H
hzcheng 已提交
494
        install_service_on_systemd
P
plum-lihui 已提交
495
    elif ((${service_mod}==1)); then
H
hzcheng 已提交
496
        install_service_on_sysvinit
P
plum-lihui 已提交
497
    else
L
lihui 已提交
498
        # must manual stop taosd
P
plum-lihui 已提交
499
        kill_taosd
H
hzcheng 已提交
500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533
    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 已提交
534
    curr_version=$(${bin_dir}/taosd -V | head -1 | cut -d ' ' -f 3)
H
hzcheng 已提交
535

L
lihui 已提交
536
    min_compatible_version=$(${script_dir}/bin/taosd -V | head -1 | cut -d ' ' -f 5)
H
hzcheng 已提交
537 538 539 540 541 542 543 544 545 546 547 548 549 550 551

    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 已提交
552
    tar -zxf taos.tar.gz
H
hzcheng 已提交
553 554 555 556 557 558 559

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

L
lihui 已提交
560
    echo -e "${GREEN}Start to update TDengine...${NC}"
H
hzcheng 已提交
561 562
    # Stop the service if running
    if pidof taosd &> /dev/null; then
P
plum-lihui 已提交
563 564 565 566
        if ((${service_mod}==0)); then
            ${csudo} systemctl stop taosd || :
        elif ((${service_mod}==1)); then
            ${csudo} service taosd stop || :
H
hzcheng 已提交
567
        else
P
plum-lihui 已提交
568
            kill_taosd
H
hzcheng 已提交
569 570 571 572 573 574 575 576 577
        fi
        sleep 1
    fi
    
    install_main_path

    install_log
    install_header
    install_lib
L
lihui 已提交
578 579 580
    if [ "$pagMode" != "lite" ]; then
      install_connector
    fi
H
hzcheng 已提交
581 582 583 584 585
    install_examples
    if [ -z $1 ]; then
        install_bin
        install_service
        install_config
S
slguan 已提交
586 587 588 589 590 591 592 593 594 595 596 597 598 599 600
		
		if [ "$verMode" == "cluster" ]; then    
            # Check if openresty is installed
            openresty_work=false

            # 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
		fi 
H
hzcheng 已提交
601 602 603 604 605

        echo
        echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
        echo
        echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
P
plum-lihui 已提交
606 607 608
        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 已提交
609
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} service taosd start${NC}"
H
hzcheng 已提交
610
        else
P
plum-lihui 已提交
611
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
H
hzcheng 已提交
612 613
        fi

S
slguan 已提交
614 615 616 617 618 619 620 621 622
        if [ "$verMode" == "cluster" ]; then  
            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}"
            fi
        else
		    echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
        fi
H
hzcheng 已提交
623 624 625
        echo
        echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
    else
L
lihui 已提交
626
        install_bin
H
hzcheng 已提交
627 628 629 630 631 632 633 634 635 636 637 638 639 640 641
        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 已提交
642
    tar -zxf taos.tar.gz
H
hzcheng 已提交
643

L
lihui 已提交
644
    echo -e "${GREEN}Start to install TDengine...${NC}"
H
hzcheng 已提交
645 646 647 648 649 650 651 652 653 654
    
	  install_main_path
	   
    if [ -z $1 ]; then
        install_data
    fi 
    
    install_log 
    install_header
    install_lib
L
lihui 已提交
655 656 657
    if [ "$pagMode" != "lite" ]; then
      install_connector
    fi
H
hzcheng 已提交
658 659 660 661 662 663
    install_examples

    if [ -z $1 ]; then # install service and client
        # For installing new
        install_bin
        install_service
S
slguan 已提交
664 665 666 667 668 669 670 671 672 673 674 675 676 677

        if [ "$verMode" == "cluster" ]; then  
            openresty_work=false
            # 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 已提交
678 679 680 681 682 683 684
        install_config	

        # Ask if to start the service
        echo
        echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
        echo
        echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
P
plum-lihui 已提交
685 686 687
        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 已提交
688
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ${csudo} service taosd start${NC}"
H
hzcheng 已提交
689
        else
L
lihui 已提交
690
            echo -e "${GREEN_DARK}To start TDengine     ${NC}: taosd${NC}"
H
hzcheng 已提交
691
        fi
S
slguan 已提交
692 693 694 695 696 697
		
        if [ "$verMode" == "cluster" ]; then  
           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
Hui Li 已提交
698 699
           fi
		    else
S
slguan 已提交
700 701 702
            echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
        fi
		
H
hzcheng 已提交
703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
        echo
        echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
    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 已提交
718
if [ "$verType" == "server" ]; then
H
hzcheng 已提交
719 720
    # Install server and client
    if [ -x ${bin_dir}/taosd ]; then
S
slguan 已提交
721
        update_flag=1
H
hzcheng 已提交
722 723 724 725
        update_TDengine
    else
        install_TDengine
    fi
H
Hui Li 已提交
726 727
elif [ "$verType" == "client" ]; then
    interactiveFqdn=no
H
hzcheng 已提交
728 729
    # Only install client
    if [ -x ${bin_dir}/taos ]; then
S
slguan 已提交
730
        update_flag=1
H
hzcheng 已提交
731 732 733 734
        update_TDengine client
    else
        install_TDengine client
    fi
H
Hui Li 已提交
735 736
else 
    echo  "please input correct verType"   
H
hzcheng 已提交
737
fi