post.sh 22.7 KB
Newer Older
wafwerar's avatar
wafwerar 已提交
1 2 3 4 5 6 7 8 9
#!/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

iplist=""
serverFqdn=""

wafwerar's avatar
wafwerar 已提交
10 11
osType=`uname`

wafwerar's avatar
wafwerar 已提交
12 13 14
# Dynamic directory
data_dir="/var/lib/taos"
log_dir="/var/log/taos"
wafwerar's avatar
wafwerar 已提交
15
cfg_install_dir="/etc/taos"
wafwerar's avatar
wafwerar 已提交
16

wafwerar's avatar
wafwerar 已提交
17 18 19 20
if [ "$osType" != "Darwin" ]; then
  script_dir=$(dirname $(readlink -f "$0"))
  verNumber=""
  lib_file_ext="so"
wafwerar's avatar
wafwerar 已提交
21
  lib_file_ext_1="so.1"
wafwerar's avatar
wafwerar 已提交
22 23 24 25 26 27 28 29 30 31 32

  bin_link_dir="/usr/bin"
  lib_link_dir="/usr/lib"
  lib64_link_dir="/usr/lib64"
  inc_link_dir="/usr/include"
  
  install_main_dir="/usr/local/taos"
else
  script_dir=${source_dir}/packaging/tools
  verNumber=`ls tdengine/driver | grep -E "libtaos\.[0-9]\.[0-9]" | sed "s/libtaos.//g" |  sed "s/.dylib//g" | head -n 1`
  lib_file_ext="dylib"
wafwerar's avatar
wafwerar 已提交
33
  lib_file_ext_1="1.dylib"
wafwerar's avatar
wafwerar 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47

  bin_link_dir="/usr/local/bin"
  lib_link_dir="/usr/local/lib"
  lib64_link_dir="/usr/local/lib"
  inc_link_dir="/usr/local/include"

  if [ -d "/usr/local/Cellar/" ];then
    install_main_dir="/usr/local/Cellar/tdengine/${verNumber}"
  elif [ -d "/opt/homebrew/Cellar/" ];then
    install_main_dir="/opt/homebrew/Cellar/tdengine/${verNumber}"
  else
    install_main_dir="/usr/local/taos"
  fi
fi
wafwerar's avatar
wafwerar 已提交
48

wafwerar's avatar
wafwerar 已提交
49 50 51 52 53 54 55 56 57
data_link_dir="${install_main_dir}/data"
log_link_dir="${install_main_dir}/log"

# static directory
cfg_dir="${install_main_dir}/cfg"
bin_dir="${install_main_dir}/bin"
lib_dir="${install_main_dir}/driver"
init_d_dir="${install_main_dir}/init.d"
inc_dir="${install_main_dir}/include"
wafwerar's avatar
wafwerar 已提交
58 59 60 61 62 63 64 65 66 67 68 69

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'

csudo=""
wafwerar's avatar
wafwerar 已提交
70
csudouser=""
wafwerar's avatar
wafwerar 已提交
71 72
if command -v sudo > /dev/null; then
    csudo="sudo "
wafwerar's avatar
wafwerar 已提交
73
    csudouser="sudo -u ${USER} "
wafwerar's avatar
wafwerar 已提交
74 75 76 77
fi

initd_mod=0
service_mod=2
wafwerar's avatar
wafwerar 已提交
78
if ps aux | grep -v grep | grep systemd &> /dev/null; then
wafwerar's avatar
wafwerar 已提交
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
    service_mod=0
elif $(which service &> /dev/null); then
    service_mod=1
    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
else
    service_mod=2
fi
wafwerar's avatar
wafwerar 已提交
95 96 97 98 99 100 101 102
if [ "$osType" = "Darwin" ]; then
  if [ -d "${install_main_dir}" ];then
    ${csudo}rm -rf ${install_main_dir}
  fi
  ${csudo}mkdir -p ${install_main_dir}
  ${csudo}rm -rf ${install_main_dir}
  ${csudo}cp -rf tdengine ${install_main_dir}
fi
wafwerar's avatar
wafwerar 已提交
103 104 105 106 107 108 109 110 111

function kill_taosadapter() {
#  ${csudo}pkill -f taosadapter || :
  pid=$(ps -ef | grep "taosadapter" | grep -v "grep" | awk '{print $2}')
  if [ -n "$pid" ]; then
    ${csudo}kill -9 $pid   || :
  fi
}

112 113 114 115 116 117 118
function kill_taoskeeper() {
  pid=$(ps -ef | grep "taoskeeper" | grep -v "grep" | awk '{print $2}')
  if [ -n "$pid" ]; then
    ${csudo}kill -9 $pid   || :
  fi
}

wafwerar's avatar
wafwerar 已提交
119 120 121 122 123 124 125 126 127
function kill_taosd() {
#  ${csudo}pkill -f taosd || :
  pid=$(ps -ef | grep "taosd" | grep -v "grep" | awk '{print $2}')
  if [ -n "$pid" ]; then
    ${csudo}kill -9 $pid   || :
  fi
}

function install_include() {
wafwerar's avatar
wafwerar 已提交
128
    ${csudo}mkdir -p ${inc_link_dir}
129
    ${csudo}rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taosdef.h ${inc_link_dir}/taoserror.h ${inc_link_dir}/taosudf.h || :
130
    [ -f ${inc_link_dir}/taosws.h ] && ${csudo}rm -f ${inc_link_dir}/taosws.h ||:
131

wafwerar's avatar
wafwerar 已提交
132 133 134
    ${csudo}ln -s ${inc_dir}/taos.h ${inc_link_dir}/taos.h
    ${csudo}ln -s ${inc_dir}/taosdef.h ${inc_link_dir}/taosdef.h
    ${csudo}ln -s ${inc_dir}/taoserror.h ${inc_link_dir}/taoserror.h
135
    ${csudo}ln -s ${inc_dir}/taosudf.h ${inc_link_dir}/taosudf.h
136

137
    [ -f ${inc_dir}/taosws.h ] && ${csudo}ln -sf ${inc_dir}/taosws.h ${inc_link_dir}/taosws.h ||:
wafwerar's avatar
wafwerar 已提交
138 139 140 141 142 143
}

function install_lib() {
    ${csudo}rm -f ${lib_link_dir}/libtaos* || :
    ${csudo}rm -f ${lib64_link_dir}/libtaos* || :

wafwerar's avatar
wafwerar 已提交
144 145
    [ -f ${lib_link_dir}/libtaosws.${lib_file_ext} ] && ${csudo}rm -f ${lib_link_dir}/libtaosws.${lib_file_ext} || :
    [ -f ${lib64_link_dir}/libtaosws.${lib_file_ext} ] && ${csudo}rm -f ${lib64_link_dir}/libtaosws.${lib_file_ext} || :
146

wafwerar's avatar
wafwerar 已提交
147 148
    ${csudo}ln -s ${lib_dir}/libtaos.* ${lib_link_dir}/libtaos.${lib_file_ext_1}
    ${csudo}ln -s ${lib_link_dir}/libtaos.${lib_file_ext_1} ${lib_link_dir}/libtaos.${lib_file_ext}
wafwerar's avatar
wafwerar 已提交
149

wafwerar's avatar
wafwerar 已提交
150
    [ -f ${lib_dir}/libtaosws.${lib_file_ext} ] && ${csudo}ln -sf ${lib_dir}/libtaosws.${lib_file_ext} ${lib_link_dir}/libtaosws.${lib_file_ext} ||:
151

wafwerar's avatar
wafwerar 已提交
152 153 154
    if [[ -d ${lib64_link_dir} && ! -e ${lib64_link_dir}/libtaos.${lib_file_ext} ]]; then
      ${csudo}ln -s ${lib_dir}/libtaos.* ${lib64_link_dir}/libtaos.${lib_file_ext_1}          || :
      ${csudo}ln -s ${lib64_link_dir}/libtaos.${lib_file_ext_1} ${lib64_link_dir}/libtaos.${lib_file_ext}   || :
155

wafwerar's avatar
wafwerar 已提交
156
      [ -f ${lib_dir}/libtaosws.${lib_file_ext} ] && ${csudo}ln -sf ${lib_dir}/libtaosws.${lib_file_ext} ${lib64_link_dir}/libtaosws.${lib_file_ext} || :
wafwerar's avatar
wafwerar 已提交
157 158
    fi

wafwerar's avatar
wafwerar 已提交
159 160 161
    if [ "$osType" != "Darwin" ]; then
      ${csudo}ldconfig
    fi
wafwerar's avatar
wafwerar 已提交
162 163 164 165 166 167
}

function install_bin() {
    # Remove links
    ${csudo}rm -f ${bin_link_dir}/taos     || :
    ${csudo}rm -f ${bin_link_dir}/taosd    || :
wafwerar's avatar
wafwerar 已提交
168
    ${csudo}rm -f ${bin_link_dir}/udfd     || :
wafwerar's avatar
wafwerar 已提交
169 170
    ${csudo}rm -f ${bin_link_dir}/taosadapter     || :
    ${csudo}rm -f ${bin_link_dir}/taosBenchmark || :
171
    ${csudo}rm -f ${bin_link_dir}/taoskeeper || :
wafwerar's avatar
wafwerar 已提交
172 173 174 175 176 177 178 179 180 181
    ${csudo}rm -f ${bin_link_dir}/taosdemo || :
    ${csudo}rm -f ${bin_link_dir}/taosdump || :
    ${csudo}rm -f ${bin_link_dir}/rmtaos   || :
    ${csudo}rm -f ${bin_link_dir}/set_core || :

    ${csudo}chmod 0555 ${bin_dir}/*

    #Make link
    [ -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          || :
wafwerar's avatar
wafwerar 已提交
182
    [ -x ${bin_dir}/udfd ] && ${csudo}ln -s ${bin_dir}/udfd ${bin_link_dir}/udfd             || :
wafwerar's avatar
wafwerar 已提交
183 184
    [ -x ${bin_dir}/taosadapter ] && ${csudo}ln -s ${bin_dir}/taosadapter ${bin_link_dir}/taosadapter             || :
    [ -x ${bin_dir}/taosBenchmark ] && ${csudo}ln -sf ${bin_dir}/taosBenchmark ${bin_link_dir}/taosdemo || :
wafwerar's avatar
wafwerar 已提交
185
    [ -x ${bin_dir}/taosBenchmark ] && ${csudo}ln -sf ${bin_dir}/taosBenchmark ${bin_link_dir}/taosBenchmark || :
wafwerar's avatar
wafwerar 已提交
186 187 188
    [ -x ${bin_dir}/TDinsight.sh ] && ${csudo}ln -sf ${bin_dir}/TDinsight.sh ${bin_link_dir}/TDinsight.sh || :
    [ -x ${bin_dir}/taosdump ] && ${csudo}ln -s ${bin_dir}/taosdump ${bin_link_dir}/taosdump || :
    [ -x ${bin_dir}/set_core.sh ] && ${csudo}ln -s ${bin_dir}/set_core.sh ${bin_link_dir}/set_core || :
wafwerar's avatar
wafwerar 已提交
189
    [ -x ${bin_dir}/remove.sh ] && ${csudo}ln -s ${bin_dir}/remove.sh ${bin_link_dir}/rmtaos || :
190
    [ -x ${bin_dir}/taoskeeper ] && ${csudo}ln -sf ${bin_dir}/taoskeeper ${bin_link_dir}/taoskeeper || :
wafwerar's avatar
wafwerar 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
}

function add_newHostname_to_hosts() {
  localIp="127.0.0.1"
  OLD_IFS="$IFS"
  IFS=" "
  iphost=$(cat /etc/hosts | grep $1 | awk '{print $1}')
  arr=($iphost)
  IFS="$OLD_IFS"
  for s in "${arr[@]}"
  do
    if [[ "$s" == "$localIp" ]]; then
      return
    fi
  done
  ${csudo}echo "127.0.0.1  $1" >> /etc/hosts   ||:
}

function set_hostname() {
  echo -e -n "${GREEN}Please enter one hostname(must not be 'localhost')${NC}:"
  read newHostname
  while true; do
    if [[ ! -z "$newHostname" && "$newHostname" != "localhost" ]]; then
      break
    else
      read -p "Please enter one hostname(must not be 'localhost'):" newHostname
217
      break
wafwerar's avatar
wafwerar 已提交
218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
    fi
  done

  ${csudo}hostname $newHostname ||:
  retval=`echo $?`
  if [[ $retval != 0 ]]; then
   echo
   echo "set hostname fail!"
   return
  fi
  #echo -e -n "$(hostnamectl status --static)"
  #echo -e -n "$(hostnamectl status --transient)"
  #echo -e -n "$(hostnamectl status --pretty)"

  #ubuntu/centos /etc/hostname
  if [[ -e /etc/hostname ]]; then
    ${csudo}echo $newHostname > /etc/hostname   ||:
  fi

  #debian: #HOSTNAME=yourname
  if [[ -e /etc/sysconfig/network ]]; then
    ${csudo}sed -i -r "s/#*\s*(HOSTNAME=\s*).*/\1$newHostname/" /etc/sysconfig/network   ||:
  fi

  ${csudo}sed -i -r "s/#*\s*(fqdn\s*).*/\1$newHostname/" ${cfg_install_dir}/taos.cfg
  serverFqdn=$newHostname

  if [[ -e /etc/hosts ]]; then
    add_newHostname_to_hosts $newHostname
  fi
}

function is_correct_ipaddr() {
  newIp=$1
  OLD_IFS="$IFS"
  IFS=" "
  arr=($iplist)
  IFS="$OLD_IFS"
  for s in "${arr[@]}"
  do
   if [[ "$s" == "$newIp" ]]; then
     return 0
   fi
  done

  return 1
}

function set_ipAsFqdn() {
  iplist=$(ip address |grep inet |grep -v inet6 |grep -v 127.0.0.1 |awk '{print $2}' |awk -F "/" '{print $1}') ||:
  if [ -z "$iplist" ]; then
    iplist=$(ifconfig |grep inet |grep -v inet6 |grep -v 127.0.0.1 |awk '{print $2}' |awk -F ":" '{print $2}') ||:
  fi

  if [ -z "$iplist" ]; then
    echo
    echo -e -n "${GREEN}Unable to get local ip, use 127.0.0.1${NC}"
    localFqdn="127.0.0.1"
    # Write the local FQDN to configuration file
    ${csudo}sed -i -r "s/#*\s*(fqdn\s*).*/\1$localFqdn/" ${cfg_install_dir}/taos.cfg
    serverFqdn=$localFqdn
    echo
    return
  fi

  echo -e -n "${GREEN}Please choose an IP from local IP list${NC}:"
  echo
  echo -e -n "${GREEN}$iplist${NC}"
  echo
  echo
  echo -e -n "${GREEN}Notes: if IP is used as the node name, data can NOT be migrated to other machine directly${NC}:"
  read localFqdn
    while true; do
      if [ ! -z "$localFqdn" ]; then
        # Check if correct ip address
        is_correct_ipaddr $localFqdn
        retval=`echo $?`
        if [[ $retval != 0 ]]; then
          read -p "Please choose an IP from local IP list:" localFqdn
        else
          # Write the local FQDN to configuration file
          ${csudo}sed -i -r "s/#*\s*(fqdn\s*).*/\1$localFqdn/" ${cfg_install_dir}/taos.cfg
          serverFqdn=$localFqdn
          break
        fi
      else
        read -p "Please choose an IP from local IP list:" localFqdn
      fi
    done
}

function local_fqdn_check() {
  #serverFqdn=$(hostname)
  echo
  echo -e -n "System hostname is: ${GREEN}$serverFqdn${NC}"
  echo
  if [[ "$serverFqdn" == "" ]] || [[ "$serverFqdn" == "localhost"  ]]; then
    echo -e -n "${GREEN}It is strongly recommended to configure a hostname for this machine ${NC}"
    echo

    while true
    do
        read -r -p "Set hostname now? [Y/n] " input
        if [ ! -n "$input" ]; then
            set_hostname
            break
        else
            case $input in
            [yY][eE][sS]|[yY])
                set_hostname
                break
                ;;

            [nN][oO]|[nN])
                set_ipAsFqdn
                break
                ;;

            *)
337
                set_ipAsFqdn
wafwerar's avatar
wafwerar 已提交
338
                echo "Invalid input..."
339
                break
wafwerar's avatar
wafwerar 已提交
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
                ;;
            esac
        fi
    done
  fi
}

function install_taosadapter_config() {
    if [ ! -f "${cfg_install_dir}/taosadapter.toml" ]; then
        [ ! -d %{cfg_install_dir} ] &&
            ${csudo}${csudo}mkdir -p ${cfg_install_dir}
        [ -f ${cfg_dir}/taosadapter.toml ] && ${csudo}cp ${cfg_dir}/taosadapter.toml ${cfg_install_dir}
        [ -f ${cfg_install_dir}/taosadapter.toml ] &&
            ${csudo}chmod 644 ${cfg_install_dir}/taosadapter.toml
    fi

    [ -f ${cfg_dir}/taosadapter.toml ] &&
        ${csudo}mv ${cfg_dir}/taosadapter.toml ${cfg_dir}/taosadapter.toml.new

    [ -f ${cfg_install_dir}/taosadapter.toml ] &&
        ${csudo}ln -s ${cfg_install_dir}/taosadapter.toml ${cfg_dir}
}

363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378
function install_taoskeeper_config() {
    if [ ! -f "${cfg_install_dir}/keeper.toml" ]; then
        [ ! -d %{cfg_install_dir} ] &&
            ${csudo}${csudo}mkdir -p ${cfg_install_dir}
        [ -f ${cfg_dir}/keeper.toml ] && ${csudo}cp ${cfg_dir}/keeper.toml ${cfg_install_dir}
        [ -f ${cfg_install_dir}/keeper.toml ] &&
            ${csudo}chmod 644 ${cfg_install_dir}/keeper.toml
    fi

    [ -f ${cfg_dir}/keeper.toml ] &&
        ${csudo}mv ${cfg_dir}/keeper.toml ${cfg_dir}/keeper.toml.new

    [ -f ${cfg_install_dir}/keeper.toml ] &&
        ${csudo}ln -s ${cfg_install_dir}/keeper.toml ${cfg_dir}
}

wafwerar's avatar
wafwerar 已提交
379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452
function install_config() {
    if [ ! -f "${cfg_install_dir}/taos.cfg" ]; then
        ${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}/*
    fi

    # Save standard input to 6 and open / dev / TTY on standard input
    exec 6<&0 0</dev/tty

    local_fqdn_check

    # restore the backup standard input, and turn off 6
    exec 0<&6 6<&-

    ${csudo}mv ${cfg_dir}/taos.cfg ${cfg_dir}/taos.cfg.new
    ${csudo}ln -s ${cfg_install_dir}/taos.cfg ${cfg_dir}
    #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
    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}:"
    #read firstEp
    if exec < /dev/tty; then
        read firstEp;
    fi
    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
  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
    done
}

function clean_service_on_sysvinit() {
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
    #${csudo}sed -i "\|${restart_config_str}|d" /etc/inittab || :

wafwerar's avatar
wafwerar 已提交
453
    if ps aux | grep -v grep | grep taosd &> /dev/null; then
wafwerar's avatar
wafwerar 已提交
454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 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 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537
        ${csudo}service taosd stop || :
    fi

    if ((${initd_mod}==1)); then
        ${csudo}chkconfig --del taosd || :
    elif ((${initd_mod}==2)); then
        ${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 || :
    fi
}

function install_service_on_sysvinit() {
    clean_service_on_sysvinit

    sleep 1

    # Install taosd service
    ${csudo}cp %{init_d_dir}/taosd ${service_config_dir} && ${csudo}chmod a+x ${service_config_dir}/taosd

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

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

    # taosd service already is stoped before install in preinst script
    #if systemctl is-active --quiet taosd; then
    #    echo "TDengine is running, stopping it..."
    #    ${csudo}systemctl stop taosd &> /dev/null || echo &> /dev/null
    #fi
    ${csudo}systemctl disable taosd &> /dev/null || echo &> /dev/null

    ${csudo}rm -f ${taosd_service_config}
}

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

    ${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 'ExecStartPre=/usr/local/taos/bin/startPre.sh'           >> ${taosd_service_config}"
    ${csudo}bash -c "echo 'TimeoutStopSec=1000000s'            >> ${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
}

wafwerar's avatar
wafwerar 已提交
538 539 540 541 542 543
function install_service_on_launchctl() {
  if [ -f ${install_main_dir}/service/com.taosdata.taosd.plist ]; then
    ${csudouser}launchctl unload -w /Library/LaunchDaemons/com.taosdata.taosd.plist > /dev/null 2>&1 || :
    ${csudo}cp ${install_main_dir}/service/com.taosdata.taosd.plist /Library/LaunchDaemons/com.taosdata.taosd.plist || :
    ${csudouser}launchctl load -w /Library/LaunchDaemons/com.taosdata.taosd.plist || :
  fi
wafwerar's avatar
wafwerar 已提交
544 545 546 547 548
  if [ -f ${install_main_dir}/service/com.taosdata.taosadapter.plist ]; then
    ${csudouser}launchctl unload -w /Library/LaunchDaemons/com.taosdata.taosadapter.plist > /dev/null 2>&1 || :
    ${csudo}cp ${install_main_dir}/service/com.taosdata.taosadapter.plist /Library/LaunchDaemons/com.taosdata.taosadapter.plist || :
    ${csudouser}launchctl load -w /Library/LaunchDaemons/com.taosdata.taosadapter.plist || :
  fi
wafwerar's avatar
wafwerar 已提交
549 550
}

wafwerar's avatar
wafwerar 已提交
551 552 553 554 555 556 557 558 559 560
function install_taosadapter_service() {
    if ((${service_mod}==0)); then
        [ -f ${script_dir}/../cfg/taosadapter.service ] &&\
            ${csudo}cp ${script_dir}/../cfg/taosadapter.service \
            ${service_config_dir}/ || :
        ${csudo}systemctl daemon-reload
    fi
}

function install_service() {
wafwerar's avatar
wafwerar 已提交
561
  if [ "$osType" != "Darwin" ]; then
wafwerar's avatar
wafwerar 已提交
562 563 564 565 566 567 568 569 570
    if ((${service_mod}==0)); then
        install_service_on_systemd
    elif ((${service_mod}==1)); then
        install_service_on_sysvinit
    else
        # manual start taosd
        kill_taosadapter
        kill_taosd
    fi
wafwerar's avatar
wafwerar 已提交
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589
  else
    install_service_on_launchctl
  fi
}

function install_app() {
  if [ "$osType" = "Darwin" ]; then
    if [ -f ${install_main_dir}/service/TDengine ]; then
      ${csudo}rm -rf /Applications/TDengine.app &&
        ${csudo}mkdir -p /Applications/TDengine.app/Contents/MacOS/ &&
        ${csudo}cp ${install_main_dir}/service/TDengine /Applications/TDengine.app/Contents/MacOS/ &&
        echo "<plist><dict></dict></plist>" | ${csudo}tee /Applications/TDengine.app/Contents/Info.plist > /dev/null &&
        ${csudo}sips -i ${install_main_dir}/service/logo.png > /dev/null && 
        DeRez -only icns ${install_main_dir}/service/logo.png | ${csudo}tee /Applications/TDengine.app/mac_logo.rsrc > /dev/null &&
        ${csudo}rez -append /Applications/TDengine.app/mac_logo.rsrc -o $'/Applications/TDengine.app/Icon\r' &&
        ${csudo}SetFile -a C /Applications/TDengine.app/ &&
        ${csudo}rm /Applications/TDengine.app/mac_logo.rsrc
    fi
  fi
wafwerar's avatar
wafwerar 已提交
590 591 592 593 594 595 596
}

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

    #install log and data dir , then ln to /usr/local/taos
    ${csudo}mkdir -p ${log_dir} && ${csudo}chmod 777 ${log_dir}
wafwerar's avatar
wafwerar 已提交
597
    ${csudo}mkdir -p ${data_dir} && ${csudo}chmod 777 ${data_dir}
wafwerar's avatar
wafwerar 已提交
598 599 600 601 602 603 604 605 606 607 608 609 610

    ${csudo}rm -rf ${log_link_dir}   || :
    ${csudo}rm -rf ${data_link_dir}  || :

    ${csudo}ln -s ${log_dir} ${log_link_dir}     || :
    ${csudo}ln -s ${data_dir} ${data_link_dir}   || :

    # Install include, lib, binary and service
    install_include
    install_lib
    install_bin
    install_config
    install_taosadapter_config
611
    install_taoskeeper_config
wafwerar's avatar
wafwerar 已提交
612 613
    install_taosadapter_service
    install_service
wafwerar's avatar
wafwerar 已提交
614
    install_app
wafwerar's avatar
wafwerar 已提交
615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658

    # 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"
    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}"
    fi



    if [ ! -z "$firstEp" ]; then
      tmpFqdn=${firstEp%%:*}
      substr=":"
      if [[ $firstEp =~ $substr ]];then
        tmpPort=${firstEp#*:}
      else
        tmpPort=""
      fi
      if [[ "$tmpPort" != "" ]];then
           echo -e "${GREEN_DARK}To access TDengine    ${NC}: taos -h $tmpFqdn -P $tmpPort${GREEN_DARK} to login into cluster, then${NC}"
      else
          echo -e "${GREEN_DARK}To access TDengine    ${NC}: taos -h $tmpFqdn${GREEN_DARK} to login into cluster, then${NC}"
      fi
      echo -e "${GREEN_DARK}execute ${NC}: create dnode 'newDnodeFQDN:port'; ${GREEN_DARK}to add this new node${NC}"
      echo
    elif [ ! -z "$serverFqdn" ]; then
      echo -e "${GREEN_DARK}To access TDengine    ${NC}: taos -h $serverFqdn${GREEN_DARK} to login into TDengine server${NC}"
      echo
    fi
    echo
    echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
}


## ==============================Main program starts from here============================
serverFqdn=$(hostname)
install_TDengine