post.sh 21.6 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 112 113 114 115 116 117 118 119 120

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
}

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() {
121
    ${csudo}rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taosdef.h ${inc_link_dir}/taoserror.h ${inc_link_dir}/taosudf.h || :
122
    [ -f ${inc_link_dir}/taosws.h ] && ${csudo}rm -f ${inc_link_dir}/taosws.h ||:
123

wafwerar's avatar
wafwerar 已提交
124 125 126
    ${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
127
    ${csudo}ln -s ${inc_dir}/taosudf.h ${inc_link_dir}/taosudf.h
128

129
    [ -f ${inc_dir}/taosws.h ] && ${csudo}ln -sf ${inc_dir}/taosws.h ${inc_link_dir}/taosws.h ||:
wafwerar's avatar
wafwerar 已提交
130 131 132 133 134 135
}

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

wafwerar's avatar
wafwerar 已提交
136 137
    [ -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} || :
138

wafwerar's avatar
wafwerar 已提交
139 140
    ${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 已提交
141

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

wafwerar's avatar
wafwerar 已提交
144 145 146
    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}   || :
147

wafwerar's avatar
wafwerar 已提交
148
      [ -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 已提交
149 150
    fi

wafwerar's avatar
wafwerar 已提交
151 152 153
    if [ "$osType" != "Darwin" ]; then
      ${csudo}ldconfig
    fi
wafwerar's avatar
wafwerar 已提交
154 155 156 157 158 159
}

function install_bin() {
    # Remove links
    ${csudo}rm -f ${bin_link_dir}/taos     || :
    ${csudo}rm -f ${bin_link_dir}/taosd    || :
wafwerar's avatar
wafwerar 已提交
160
    ${csudo}rm -f ${bin_link_dir}/udfd     || :
wafwerar's avatar
wafwerar 已提交
161 162 163 164 165 166 167 168 169 170 171 172
    ${csudo}rm -f ${bin_link_dir}/taosadapter     || :
    ${csudo}rm -f ${bin_link_dir}/taosBenchmark || :
    ${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 已提交
173
    [ -x ${bin_dir}/udfd ] && ${csudo}ln -s ${bin_dir}/udfd ${bin_link_dir}/udfd             || :
wafwerar's avatar
wafwerar 已提交
174 175
    [ -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 已提交
176
    [ -x ${bin_dir}/taosBenchmark ] && ${csudo}ln -sf ${bin_dir}/taosBenchmark ${bin_link_dir}/taosBenchmark || :
wafwerar's avatar
wafwerar 已提交
177 178 179
    [ -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 已提交
180
    [ -x ${bin_dir}/remove.sh ] && ${csudo}ln -s ${bin_dir}/remove.sh ${bin_link_dir}/rmtaos || :
wafwerar's avatar
wafwerar 已提交
181 182 183 184 185 186 187 188 189 190 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 217 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 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 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
}

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
    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
                ;;

            *)
                echo "Invalid input..."
                ;;
            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}
}

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 已提交
424
    if ps aux | grep -v grep | grep taosd &> /dev/null; then
wafwerar's avatar
wafwerar 已提交
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 453 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
        ${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 已提交
509 510 511 512 513 514
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 已提交
515 516 517 518 519
  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 已提交
520 521
}

wafwerar's avatar
wafwerar 已提交
522 523 524 525 526 527 528 529 530 531
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 已提交
532
  if [ "$osType" != "Darwin" ]; then
wafwerar's avatar
wafwerar 已提交
533 534 535 536 537 538 539 540 541
    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 已提交
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
  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 已提交
561 562 563 564 565 566 567
}

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 已提交
568
    ${csudo}mkdir -p ${data_dir} && ${csudo}chmod 777 ${data_dir}
wafwerar's avatar
wafwerar 已提交
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583

    ${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
    install_taosadapter_service
    install_service
wafwerar's avatar
wafwerar 已提交
584
    install_app
wafwerar's avatar
wafwerar 已提交
585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628

    # 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