make_install.sh 28.6 KB
Newer Older
wafwerar's avatar
wafwerar 已提交
1 2 3 4 5 6
#!/bin/bash
#
# This file is used to install TAOS time-series database on linux systems. The operating system
# is required to use systemd to manage services at boot

set -e
wafwerar's avatar
wafwerar 已提交
7
#set -x
wafwerar's avatar
wafwerar 已提交
8 9 10 11 12 13 14 15

# -----------------------Variables definition
source_dir=$1
binary_dir=$2
osType=$3
verNumber=$4

if [ "$osType" != "Darwin" ]; then
wafwerar's avatar
wafwerar 已提交
16
  script_dir=$(dirname $(readlink -f "$0"))
wafwerar's avatar
wafwerar 已提交
17
else
wafwerar's avatar
wafwerar 已提交
18
  script_dir=${source_dir}/packaging/tools
wafwerar's avatar
wafwerar 已提交
19 20 21
fi

# Dynamic directory
wafwerar's avatar
wafwerar 已提交
22 23 24 25 26 27 28 29 30 31
clientName="taos"
serverName="taosd"
logDir="/var/log/taos"
dataDir="/var/lib/taos"
configDir="/etc/taos"
configFile="taos.cfg"
installDir="/usr/local/taos"
productName="TDengine"
emailName="taosdata.com"
uninstallScript="rmtaos"
wafwerar's avatar
wafwerar 已提交
32

wafwerar's avatar
wafwerar 已提交
33 34 35
data_dir=${dataDir}
log_dir=${logDir}
cfg_install_dir=${configDir}
wafwerar's avatar
wafwerar 已提交
36

wafwerar's avatar
wafwerar 已提交
37
if [ "$osType" != "Darwin" ]; then
wafwerar's avatar
wafwerar 已提交
38 39 40 41
  bin_link_dir="/usr/bin"
  lib_link_dir="/usr/lib"
  lib64_link_dir="/usr/lib64"
  inc_link_dir="/usr/include"
wafwerar's avatar
wafwerar 已提交
42
else
wafwerar's avatar
wafwerar 已提交
43 44 45
  bin_link_dir="/usr/local/bin"
  lib_link_dir="/usr/local/lib"
  inc_link_dir="/usr/local/include"
wafwerar's avatar
wafwerar 已提交
46

wafwerar's avatar
wafwerar 已提交
47 48 49 50 51 52 53
  if [ -d "/usr/local/Cellar/" ];then
    installDir="/usr/local/Cellar/tdengine/${verNumber}"
  elif [ -d "/opt/homebrew/Cellar/" ];then
    installDir="/opt/homebrew/Cellar/tdengine/${verNumber}"
  else
    installDir="/usr/local/taos"
  fi
wafwerar's avatar
wafwerar 已提交
54
fi
wafwerar's avatar
wafwerar 已提交
55 56 57
install_main_dir=${installDir}
bin_dir="${installDir}/bin"
cfg_dir="${installDir}/cfg"
wafwerar's avatar
wafwerar 已提交
58 59 60 61 62 63 64 65 66 67 68

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 已提交
69 70 71 72 73
csudouser=""
if command -v sudo >/dev/null; then
  csudo="sudo "
  csudouser="sudo -u ${USER} "
fi
wafwerar's avatar
wafwerar 已提交
74

wafwerar's avatar
wafwerar 已提交
75 76
service_mod=2
os_type=0
wafwerar's avatar
wafwerar 已提交
77

wafwerar's avatar
wafwerar 已提交
78 79 80 81 82 83 84 85 86 87 88 89 90
if [ "$osType" != "Darwin" ]; then
  initd_mod=0
  if pidof systemd &>/dev/null; then
    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
wafwerar's avatar
wafwerar 已提交
91
    else
wafwerar's avatar
wafwerar 已提交
92
      service_mod=2
wafwerar's avatar
wafwerar 已提交
93
    fi
wafwerar's avatar
wafwerar 已提交
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
  else
    service_mod=2
  fi

  # 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}"
  if echo $osinfo | grep -qwi "ubuntu"; then
    echo "this is ubuntu system"
    os_type=1
  elif echo $osinfo | grep -qwi "debian"; then
    echo "this is debian system"
    os_type=1
  elif echo $osinfo | grep -qwi "Kylin"; then
    echo "this is Kylin system"
    os_type=1
  elif echo $osinfo | grep -qwi "centos"; then
    echo "this is centos system"
    os_type=2
  elif echo $osinfo | grep -qwi "fedora"; then
    echo "this is fedora system"
    os_type=2
  else
    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 ${emailName} for support."
    os_type=1
  fi
wafwerar's avatar
wafwerar 已提交
123 124
fi

wafwerar's avatar
wafwerar 已提交
125 126 127 128 129 130 131
function kill_taosadapter() {
  pid=$(ps -ef | grep "taosadapter" | grep -v "grep" | awk '{print $2}')
  if [ -n "$pid" ]; then
    ${csudo}kill -9 $pid || :
  fi
}

wafwerar's avatar
wafwerar 已提交
132
function kill_taosd() {
wafwerar's avatar
wafwerar 已提交
133 134 135 136 137
  ps -ef | grep ${serverName}
  pid=$(ps -ef | grep -w ${serverName} | grep -v "grep" | awk '{print $2}')
  if [ -n "$pid" ]; then
    ${csudo}kill -9 $pid || :
  fi
wafwerar's avatar
wafwerar 已提交
138 139 140
}

function install_main_path() {
wafwerar's avatar
wafwerar 已提交
141
  #create install main dir and all sub dir
wafwerar's avatar
wafwerar 已提交
142 143 144 145 146 147 148 149 150 151
  ${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}/share
  #    ${csudo}mkdir -p ${install_main_dir}/init.d
wafwerar's avatar
wafwerar 已提交
152 153 154
}

function install_bin() {
wafwerar's avatar
wafwerar 已提交
155 156 157 158
  # Remove links
  ${csudo}rm -f ${bin_link_dir}/${clientName} || :
  ${csudo}rm -f ${bin_link_dir}/${serverName} || :
  ${csudo}rm -f ${bin_link_dir}/taosadapter || :
wafwerar's avatar
wafwerar 已提交
159
  ${csudo}rm -f ${bin_link_dir}/udfd || :
wafwerar's avatar
wafwerar 已提交
160 161
  ${csudo}rm -f ${bin_link_dir}/taosdemo || :
  ${csudo}rm -f ${bin_link_dir}/taosdump || :
162
  ${csudo}rm -f ${bin_link_dir}/taosx || :
wafwerar's avatar
wafwerar 已提交
163
  ${csudo}rm -f ${bin_link_dir}/${uninstallScript} || :
wafwerar's avatar
wafwerar 已提交
164 165 166 167 168 169 170 171 172 173

  if [ "$osType" != "Darwin" ]; then
    ${csudo}rm -f ${bin_link_dir}/perfMonitor || :
    ${csudo}rm -f ${bin_link_dir}/set_core || :

    ${csudo}cp -r ${binary_dir}/build/bin/${clientName} ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/taosBenchmark ] && ${csudo}cp -r ${binary_dir}/build/bin/taosBenchmark ${install_main_dir}/bin || :
    [ -f ${install_main_dir}/bin/taosBenchmark ] && ${csudo}ln -sf ${install_main_dir}/bin/taosBenchmark ${install_main_dir}/bin/taosdemo || :
    [ -f ${binary_dir}/build/bin/taosdump ] && ${csudo}cp -r ${binary_dir}/build/bin/taosdump ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/taosadapter ] && ${csudo}cp -r ${binary_dir}/build/bin/taosadapter ${install_main_dir}/bin || :
wafwerar's avatar
wafwerar 已提交
174
    [ -f ${binary_dir}/build/bin/udfd ] && ${csudo}cp -r ${binary_dir}/build/bin/udfd ${install_main_dir}/bin || :
175
    [ -f ${binary_dir}/build/bin/taosx ] && ${csudo}cp -r ${binary_dir}/build/bin/taosx ${install_main_dir}/bin || :
wafwerar's avatar
wafwerar 已提交
176 177 178 179 180 181 182 183
    ${csudo}cp -r ${binary_dir}/build/bin/${serverName} ${install_main_dir}/bin || :

    ${csudo}cp -r ${script_dir}/taosd-dump-cfg.gdb ${install_main_dir}/bin || :
    ${csudo}cp -r ${script_dir}/remove.sh ${install_main_dir}/bin || :
    ${csudo}cp -r ${script_dir}/set_core.sh ${install_main_dir}/bin || :
    ${csudo}cp -r ${script_dir}/startPre.sh ${install_main_dir}/bin || :

    ${csudo}chmod 0555 ${install_main_dir}/bin/*
wafwerar's avatar
wafwerar 已提交
184
    #Make link
wafwerar's avatar
wafwerar 已提交
185 186 187
    [ -x ${install_main_dir}/bin/${clientName} ] && ${csudo}ln -s ${install_main_dir}/bin/${clientName} ${bin_link_dir}/${clientName} || :
    [ -x ${install_main_dir}/bin/${serverName} ] && ${csudo}ln -s ${install_main_dir}/bin/${serverName} ${bin_link_dir}/${serverName} || :
    [ -x ${install_main_dir}/bin/taosadapter ] && ${csudo}ln -s ${install_main_dir}/bin/taosadapter ${bin_link_dir}/taosadapter || :
wafwerar's avatar
wafwerar 已提交
188
    [ -x ${install_main_dir}/bin/udfd ] && ${csudo}ln -s ${install_main_dir}/bin/udfd ${bin_link_dir}/udfd || :
wafwerar's avatar
wafwerar 已提交
189 190
    [ -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 || :
191
    [ -x ${install_main_dir}/bin/taosx ] && ${csudo}ln -s ${install_main_dir}/bin/taosx ${bin_link_dir}/taosx || :
wafwerar's avatar
wafwerar 已提交
192 193 194 195 196
    [ -x ${install_main_dir}/bin/perfMonitor ] && ${csudo}ln -s ${install_main_dir}/bin/perfMonitor ${bin_link_dir}/perfMonitor || :
    [ -x ${install_main_dir}/set_core.sh ] && ${csudo}ln -s ${install_main_dir}/bin/set_core.sh ${bin_link_dir}/set_core || :
    [ -x ${install_main_dir}/bin/remove.sh ] && ${csudo}ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/${uninstallScript} || :
  else

wafwerar's avatar
wafwerar 已提交
197 198 199 200 201 202 203 204 205 206 207
    ${csudo}cp -r ${binary_dir}/build/bin/${clientName} ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/taosBenchmark ] && ${csudo}cp -r ${binary_dir}/build/bin/taosBenchmark ${install_main_dir}/bin || :
    [ -f ${install_main_dir}/bin/taosBenchmark ] && ${csudo}ln -sf ${install_main_dir}/bin/taosBenchmark ${install_main_dir}/bin/taosdemo || :
    [ -f ${binary_dir}/build/bin/taosdump ] && ${csudo}cp -r ${binary_dir}/build/bin/taosdump ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/taosadapter ] && ${csudo}cp -r ${binary_dir}/build/bin/taosadapter ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/udfd ] && ${csudo}cp -r ${binary_dir}/build/bin/udfd ${install_main_dir}/bin || :
    [ -f ${binary_dir}/build/bin/taosx ] && ${csudo}cp -r ${binary_dir}/build/bin/taosx ${install_main_dir}/bin || :
    ${csudo}cp -r ${binary_dir}/build/bin/${serverName} ${install_main_dir}/bin || :

    ${csudo}cp -r ${script_dir}/remove.sh ${install_main_dir}/bin || :
    ${csudo}chmod 0555 ${install_main_dir}/bin/*
wafwerar's avatar
wafwerar 已提交
208
    #Make link
wafwerar's avatar
wafwerar 已提交
209 210 211 212 213 214 215 216
    [ -x ${install_main_dir}/bin/${clientName} ] && ${csudo}ln -s ${install_main_dir}/bin/${clientName} ${bin_link_dir}/${clientName} || :
    [ -x ${install_main_dir}/bin/${serverName} ] && ${csudo}ln -s ${install_main_dir}/bin/${serverName} ${bin_link_dir}/${serverName} || :
    [ -x ${install_main_dir}/bin/taosadapter ] && ${csudo}ln -s ${install_main_dir}/bin/taosadapter ${bin_link_dir}/taosadapter || :
    [ -x ${install_main_dir}/bin/udfd ] && ${csudo}ln -s ${install_main_dir}/bin/udfd ${bin_link_dir}/udfd || :
    [ -x ${install_main_dir}/bin/taosdump ] && ${csudo}ln -s ${install_main_dir}/bin/taosdump ${bin_link_dir}/taosdump || :
    [ -f ${install_main_dir}/bin/taosBenchmark ] && ${csudo}ln -sf ${install_main_dir}/bin/taosBenchmark ${install_main_dir}/bin/taosdemo || :
    [ -x ${install_main_dir}/bin/taosx ] && ${csudo}ln -s ${install_main_dir}/bin/taosx ${bin_link_dir}/taosx || :
    [ -x ${install_main_dir}/bin/remove.sh ] && ${csudo}ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/${uninstallScript} || :
wafwerar's avatar
wafwerar 已提交
217
  fi
wafwerar's avatar
wafwerar 已提交
218 219 220
}

function install_jemalloc() {
wafwerar's avatar
wafwerar 已提交
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
  if [ "$osType" != "Darwin" ]; then
    /usr/bin/install -c -d /usr/local/bin

    if [ -f "${binary_dir}/build/bin/jemalloc-config" ]; then
      ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/bin/jemalloc-config /usr/local/bin
    fi
    if [ -f "${binary_dir}/build/bin/jemalloc.sh" ]; then
      ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/bin/jemalloc.sh /usr/local/bin
    fi
    if [ -f "${binary_dir}/build/bin/jeprof" ]; then
      ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/bin/jeprof /usr/local/bin
    fi
    if [ -f "${binary_dir}/build/include/jemalloc/jemalloc.h" ]; then
      ${csudo}/usr/bin/install -c -d /usr/local/include/jemalloc
      ${csudo}/usr/bin/install -c -m 644 ${binary_dir}/build/include/jemalloc/jemalloc.h \
        /usr/local/include/jemalloc
    fi
    if [ -f "${binary_dir}/build/lib/libjemalloc.so.2" ]; then
      ${csudo}/usr/bin/install -c -d /usr/local/lib
      ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/lib/libjemalloc.so.2 /usr/local/lib
      ${csudo}ln -sf libjemalloc.so.2 /usr/local/lib/libjemalloc.so
      ${csudo}/usr/bin/install -c -d /usr/local/lib
      [ -f ${binary_dir}/build/lib/libjemalloc.a ] &&
        ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/lib/libjemalloc.a /usr/local/lib
      [ -f ${binary_dir}/build/lib/libjemalloc_pic.a ] &&
        ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/lib/libjemalloc_pic.a /usr/local/lib
      if [ -f "${binary_dir}/build/lib/pkgconfig/jemalloc.pc" ]; then
        ${csudo}/usr/bin/install -c -d /usr/local/lib/pkgconfig
        ${csudo}/usr/bin/install -c -m 644 ${binary_dir}/build/lib/pkgconfig/jemalloc.pc \
          /usr/local/lib/pkgconfig
      fi
      if [ -d /etc/ld.so.conf.d ]; then
        echo "/usr/local/lib" | ${csudo}tee /etc/ld.so.conf.d/jemalloc.conf >/dev/null || echo -e "failed to write /etc/ld.so.conf.d/jemalloc.conf"
        ${csudo}ldconfig
      else
        echo "/etc/ld.so.conf.d not found!"
      fi
wafwerar's avatar
wafwerar 已提交
258
    fi
wafwerar's avatar
wafwerar 已提交
259 260 261 262
    if [ -f "${binary_dir}/build/share/doc/jemalloc/jemalloc.html" ]; then
      ${csudo}/usr/bin/install -c -d /usr/local/share/doc/jemalloc
      ${csudo}/usr/bin/install -c -m 644 ${binary_dir}/build/share/doc/jemalloc/jemalloc.html \
        /usr/local/share/doc/jemalloc
wafwerar's avatar
wafwerar 已提交
263
    fi
wafwerar's avatar
wafwerar 已提交
264 265 266 267
    if [ -f "${binary_dir}/build/share/man/man3/jemalloc.3" ]; then
      ${csudo}/usr/bin/install -c -d /usr/local/share/man/man3
      ${csudo}/usr/bin/install -c -m 644 ${binary_dir}/build/share/man/man3/jemalloc.3 \
        /usr/local/share/man/man3
wafwerar's avatar
wafwerar 已提交
268
    fi
wafwerar's avatar
wafwerar 已提交
269 270
  fi
}
wafwerar's avatar
wafwerar 已提交
271

wafwerar's avatar
wafwerar 已提交
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
function install_avro() {
  if [ "$osType" != "Darwin" ]; then
    if [ -f "${binary_dir}/build/$1/libavro.so.23.0.0" ] && [ -d /usr/local/$1 ]; then
      ${csudo}/usr/bin/install -c -d /usr/local/$1
      ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/$1/libavro.so.23.0.0 /usr/local/$1
      ${csudo}ln -sf libavro.so.23.0.0 /usr/local/$1/libavro.so.23
      ${csudo}ln -sf libavro.so.23 /usr/local/$1/libavro.so
      ${csudo}/usr/bin/install -c -d /usr/local/$1
      [ -f ${binary_dir}/build/$1/libavro.a ] &&
        ${csudo}/usr/bin/install -c -m 755 ${binary_dir}/build/$1/libavro.a /usr/local/$1

      if [ -d /etc/ld.so.conf.d ]; then
        echo "/usr/local/$1" | ${csudo}tee /etc/ld.so.conf.d/libavro.conf >/dev/null || echo -e "failed to write /etc/ld.so.conf.d/libavro.conf"
        ${csudo}ldconfig
      else
        echo "/etc/ld.so.conf.d not found!"
      fi
wafwerar's avatar
wafwerar 已提交
289
    fi
wafwerar's avatar
wafwerar 已提交
290 291 292 293 294 295
  fi
}

function install_lib() {
  # Remove links
  ${csudo}rm -f ${lib_link_dir}/libtaos.* || :
296
  [ -f ${lib_link_dir}/libtaosws.so ] && ${csudo}rm -f ${lib_link_dir}/libtaosws.so || :
wafwerar's avatar
wafwerar 已提交
297 298
  if [ "$osType" != "Darwin" ]; then
    ${csudo}rm -f ${lib64_link_dir}/libtaos.* || :
299
    [ -f ${lib64_link_dir}/libtaosws.so ] && ${csudo}rm -f ${lib64_link_dir}/libtaosws.so || :
wafwerar's avatar
wafwerar 已提交
300 301 302 303 304
  fi

  if [ "$osType" != "Darwin" ]; then
    ${csudo}cp ${binary_dir}/build/lib/libtaos.so.${verNumber} \
      ${install_main_dir}/driver &&
305 306
      ${csudo}chmod 777 ${install_main_dir}/driver/libtaos.so.${verNumber}

wafwerar's avatar
wafwerar 已提交
307 308 309
    ${csudo}ln -sf ${install_main_dir}/driver/libtaos.* ${lib_link_dir}/libtaos.so.1
    ${csudo}ln -sf ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
    if [ -d "${lib64_link_dir}" ]; then
310 311 312 313 314 315 316 317 318 319
        ${csudo}ln -sf ${install_main_dir}/driver/libtaos.* ${lib64_link_dir}/libtaos.so.1
        ${csudo}ln -sf ${lib64_link_dir}/libtaos.so.1 ${lib64_link_dir}/libtaos.so
    fi

    if [ -f ${binary_dir}/build/lib/libtaosws.so ]; then
        ${csudo}cp ${binary_dir}/build/lib/libtaosws.so \
            ${install_main_dir}/driver &&
            ${csudo}chmod 777 ${install_main_dir}/driver/libtaosws.so ||:

        ${csudo}ln -sf ${install_main_dir}/driver/libtaosws.so ${lib_link_dir}/libtaosws.so || :
wafwerar's avatar
wafwerar 已提交
320 321 322
    fi
  else
    ${csudo}cp -Rf ${binary_dir}/build/lib/libtaos.${verNumber}.dylib \
wafwerar's avatar
wafwerar 已提交
323
      ${install_main_dir}/driver && ${csudo}chmod 777 ${install_main_dir}/driver/*
wafwerar's avatar
wafwerar 已提交
324 325

    ${csudo}ln -sf ${install_main_dir}/driver/libtaos.${verNumber}.dylib \
wafwerar's avatar
wafwerar 已提交
326
      ${lib_link_dir}/libtaos.1.dylib || :
wafwerar's avatar
wafwerar 已提交
327 328

    ${csudo}ln -sf ${lib_link_dir}/libtaos.1.dylib ${lib_link_dir}/libtaos.dylib || :
wafwerar's avatar
wafwerar 已提交
329 330 331 332 333 334 335 336

    if [ -f ${binary_dir}/build/lib/libtaosws.dylib ]; then
        ${csudo}cp ${binary_dir}/build/lib/libtaosws.dylib \
            ${install_main_dir}/driver &&
            ${csudo}chmod 777 ${install_main_dir}/driver/libtaosws.dylib ||:

        ${csudo}ln -sf ${install_main_dir}/driver/libtaosws.dylib ${lib_link_dir}/libtaosws.dylib || :
    fi
wafwerar's avatar
wafwerar 已提交
337 338 339
  fi

  install_jemalloc
340 341
  #install_avro lib
  #install_avro lib64
wafwerar's avatar
wafwerar 已提交
342 343 344 345

  if [ "$osType" != "Darwin" ]; then
    ${csudo}ldconfig
  fi
wafwerar's avatar
wafwerar 已提交
346 347 348
}

function install_header() {
wafwerar's avatar
wafwerar 已提交
349 350 351 352
  ${csudo}rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taosdef.h ${inc_link_dir}/taoserror.h ${inc_link_dir}/taosudf.h || :
  [ -f ${inc_link_dir}/taosws.h ] && ${csudo}rm -f ${inc_link_dir}/taosws.h ||:
  ${csudo}cp -f ${source_dir}/include/client/taos.h ${source_dir}/include/common/taosdef.h ${source_dir}/include/util/taoserror.h ${source_dir}/include/libs/function/taosudf.h \
    ${install_main_dir}/include && ${csudo}chmod 644 ${install_main_dir}/include/*
wafwerar's avatar
wafwerar 已提交
353

wafwerar's avatar
wafwerar 已提交
354 355 356 357
  if [ -f ${binary_dir}/build/include/taosws.h ]; then
    ${csudo}cp -f ${binary_dir}/build/include/taosws.h ${install_main_dir}/include && ${csudo}chmod 644 ${install_main_dir}/include/taosws.h ||:
    ${csudo}ln -sf ${install_main_dir}/include/taosws.h ${inc_link_dir}/taosws.h ||:
  fi
358

wafwerar's avatar
wafwerar 已提交
359 360 361 362
  ${csudo}ln -s ${install_main_dir}/include/taos.h ${inc_link_dir}/taos.h
  ${csudo}ln -s ${install_main_dir}/include/taosdef.h ${inc_link_dir}/taosdef.h
  ${csudo}ln -s ${install_main_dir}/include/taoserror.h ${inc_link_dir}/taoserror.h
  ${csudo}ln -s ${install_main_dir}/include/taosudf.h ${inc_link_dir}/taosudf.h
363

wafwerar's avatar
wafwerar 已提交
364
  ${csudo}chmod 644 ${install_main_dir}/include/*
wafwerar's avatar
wafwerar 已提交
365 366 367
}

function install_config() {
wafwerar's avatar
wafwerar 已提交
368 369 370
  if [ ! -f ${cfg_install_dir}/${configFile} ]; then
    ${csudo}mkdir -p ${cfg_install_dir}
    [ -f ${script_dir}/../cfg/${configFile} ] &&
wafwerar's avatar
wafwerar 已提交
371 372
      ${csudo}cp ${script_dir}/../cfg/${configFile} ${cfg_install_dir} &&
      ${csudo}cp ${script_dir}/../cfg/${configFile} ${cfg_dir}
wafwerar's avatar
wafwerar 已提交
373 374
    ${csudo}chmod 644 ${cfg_install_dir}/${configFile}
    ${csudo}cp -f ${script_dir}/../cfg/${configFile} \
wafwerar's avatar
wafwerar 已提交
375
      ${cfg_install_dir}/${configFile}.${verNumber}
wafwerar's avatar
wafwerar 已提交
376 377 378 379
    ${csudo}ln -s ${cfg_install_dir}/${configFile} \
      ${install_main_dir}/cfg/${configFile}
  else
    ${csudo}cp -f ${script_dir}/../cfg/${configFile} \
wafwerar's avatar
wafwerar 已提交
380
      ${cfg_install_dir}/${configFile}.${verNumber}
wafwerar's avatar
wafwerar 已提交
381
    ${csudo}cp -f ${script_dir}/../cfg/${configFile} ${cfg_dir}
wafwerar's avatar
wafwerar 已提交
382 383
  fi
}
wafwerar's avatar
wafwerar 已提交
384

wafwerar's avatar
wafwerar 已提交
385 386 387 388
function install_taosadapter_config() {
  if [ ! -f "${cfg_install_dir}/taosadapter.toml" ]; then
    ${csudo}mkdir -p ${cfg_install_dir} || :
    [ -f ${binary_dir}/test/cfg/taosadapter.toml ] &&
wafwerar's avatar
wafwerar 已提交
389 390
      ${csudo}cp ${binary_dir}/test/cfg/taosadapter.toml ${cfg_install_dir} &&
      ${csudo}cp ${binary_dir}/test/cfg/taosadapter.toml ${cfg_dir} || :
wafwerar's avatar
wafwerar 已提交
391 392 393 394 395 396 397 398 399 400 401 402
    [ -f ${cfg_install_dir}/taosadapter.toml ] &&
      ${csudo}chmod 644 ${cfg_install_dir}/taosadapter.toml || :
    [ -f ${binary_dir}/test/cfg/taosadapter.toml ] &&
      ${csudo}cp -f ${binary_dir}/test/cfg/taosadapter.toml \
        ${cfg_install_dir}/taosadapter.toml.${verNumber} || :
    [ -f ${cfg_install_dir}/taosadapter.toml ] &&
      ${csudo}ln -s ${cfg_install_dir}/taosadapter.toml \
        ${install_main_dir}/cfg/taosadapter.toml || :
  else
    if [ -f "${binary_dir}/test/cfg/taosadapter.toml" ]; then
      ${csudo}cp -f ${binary_dir}/test/cfg/taosadapter.toml \
        ${cfg_install_dir}/taosadapter.toml.${verNumber} || :
wafwerar's avatar
wafwerar 已提交
403
      ${csudo}cp -f ${binary_dir}/test/cfg/taosadapter.toml ${cfg_dir} || :
wafwerar's avatar
wafwerar 已提交
404 405
    fi
  fi
wafwerar's avatar
wafwerar 已提交
406 407 408
}

function install_log() {
wafwerar's avatar
wafwerar 已提交
409 410
  ${csudo}rm -rf ${log_dir} || :
  ${csudo}mkdir -p ${log_dir} && ${csudo}chmod 777 ${log_dir}
wafwerar's avatar
wafwerar 已提交
411
  ${csudo}ln -s ${log_dir} ${install_main_dir}/log
wafwerar's avatar
wafwerar 已提交
412 413 414
}

function install_data() {
wafwerar's avatar
wafwerar 已提交
415 416
  ${csudo}mkdir -p ${data_dir} && ${csudo}chmod 777 ${data_dir}
  ${csudo}ln -s ${data_dir} ${install_main_dir}/data
wafwerar's avatar
wafwerar 已提交
417 418 419
}

function install_connector() {
wafwerar's avatar
wafwerar 已提交
420 421 422 423 424
  if find ${source_dir}/src/connector/go -mindepth 1 -maxdepth 1 | read; then
    ${csudo}cp -r ${source_dir}/src/connector/go ${install_main_dir}/connector || :
  else
    echo "WARNING: go connector not found, please check if want to use it!"
  fi
wafwerar's avatar
wafwerar 已提交
425 426
  ${csudo}cp -rf ${source_dir}/src/connector/python ${install_main_dir}/connector || :
  ${csudo}cp ${binary_dir}/build/lib/*.jar ${install_main_dir}/connector &>/dev/null && ${csudo}chmod 777 ${install_main_dir}/connector/*.jar || echo &>/dev/null || :
wafwerar's avatar
wafwerar 已提交
427 428 429
}

function install_examples() {
wafwerar's avatar
wafwerar 已提交
430
  ${csudo}cp -rf ${source_dir}/examples/* ${install_main_dir}/examples || :
wafwerar's avatar
wafwerar 已提交
431 432
}

433 434
function install_web() {
  if [ -d "${binary_dir}/build/share" ]; then
wafwerar's avatar
wafwerar 已提交
435
    ${csudo}cp -rf ${binary_dir}/build/share/* ${install_main_dir}/share || :
436 437 438
  fi
}

wafwerar's avatar
wafwerar 已提交
439
function clean_service_on_sysvinit() {
wafwerar's avatar
wafwerar 已提交
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456
  if pidof ${serverName} &>/dev/null; then
    ${csudo}service ${serverName} stop || :
  fi

  if ((${initd_mod} == 1)); then
    ${csudo}chkconfig --del ${serverName} || :
  elif ((${initd_mod} == 2)); then
    ${csudo}insserv -r ${serverName} || :
  elif ((${initd_mod} == 3)); then
    ${csudo}update-rc.d -f ${serverName} remove || :
  fi

  ${csudo}rm -f ${service_config_dir}/${serverName} || :

  if $(which init &>/dev/null); then
    ${csudo}init q || :
  fi
wafwerar's avatar
wafwerar 已提交
457 458 459
}

function install_service_on_sysvinit() {
wafwerar's avatar
wafwerar 已提交
460 461 462 463 464 465
  clean_service_on_sysvinit

  sleep 1

  if ((${os_type} == 1)); then
    #    ${csudo}cp -f ${script_dir}/../deb/${serverName} ${install_main_dir}/init.d
wafwerar's avatar
wafwerar 已提交
466
    ${csudo}cp ${script_dir}/../deb/${serverName} ${service_config_dir} && ${csudo}chmod a+x ${service_config_dir}/${serverName}
wafwerar's avatar
wafwerar 已提交
467 468
  elif ((${os_type} == 2)); then
    #    ${csudo}cp -f ${script_dir}/../rpm/${serverName} ${install_main_dir}/init.d
wafwerar's avatar
wafwerar 已提交
469
    ${csudo}cp ${script_dir}/../rpm/${serverName} ${service_config_dir} && ${csudo}chmod a+x ${service_config_dir}/${serverName}
wafwerar's avatar
wafwerar 已提交
470 471 472 473 474 475 476 477 478 479 480
  fi

  if ((${initd_mod} == 1)); then
    ${csudo}chkconfig --add ${serverName} || :
    ${csudo}chkconfig --level 2345 ${serverName} on || :
  elif ((${initd_mod} == 2)); then
    ${csudo}insserv ${serverName} || :
    ${csudo}insserv -d ${serverName} || :
  elif ((${initd_mod} == 3)); then
    ${csudo}update-rc.d ${serverName} defaults || :
  fi
wafwerar's avatar
wafwerar 已提交
481 482 483
}

function clean_service_on_systemd() {
wafwerar's avatar
wafwerar 已提交
484
  taosd_service_config="${service_config_dir}/${serverName}.service"
wafwerar's avatar
wafwerar 已提交
485

wafwerar's avatar
wafwerar 已提交
486 487 488 489 490
  if systemctl is-active --quiet ${serverName}; then
    echo "${productName} is running, stopping it..."
    ${csudo}systemctl stop ${serverName} &>/dev/null || echo &>/dev/null
  fi
  ${csudo}systemctl disable ${serverName} &>/dev/null || echo &>/dev/null
wafwerar's avatar
wafwerar 已提交
491

wafwerar's avatar
wafwerar 已提交
492
  ${csudo}rm -f ${taosd_service_config}
wafwerar's avatar
wafwerar 已提交
493 494 495
}

function install_service_on_systemd() {
wafwerar's avatar
wafwerar 已提交
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
  clean_service_on_systemd

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

  ${csudo}bash -c "echo '[Unit]'                             >> ${taosd_service_config}"
  ${csudo}bash -c "echo 'Description=${productName} 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/${serverName}'           >> ${taosd_service_config}"
  ${csudo}bash -c "echo 'ExecStartPre=${installDir}/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 ${serverName}
}

function install_taosadapter_service() {
  if ((${service_mod} == 0)); then
    [ -f ${binary_dir}/test/cfg/taosadapter.service ] &&
      ${csudo}cp ${binary_dir}/test/cfg/taosadapter.service \
        ${service_config_dir}/ || :
    ${csudo}systemctl daemon-reload
  fi
wafwerar's avatar
wafwerar 已提交
531 532
}

wafwerar's avatar
wafwerar 已提交
533 534 535 536 537 538
function install_service_on_launchctl() {
  ${csudouser}launchctl unload -w /Library/LaunchDaemons/com.taosdata.taosd.plist > /dev/null 2>&1 || :
  ${csudo}cp ${script_dir}/com.taosdata.taosd.plist /Library/LaunchDaemons/com.taosdata.taosd.plist
  ${csudouser}launchctl load -w /Library/LaunchDaemons/com.taosdata.taosd.plist || :
}

wafwerar's avatar
wafwerar 已提交
539
function install_service() {
wafwerar's avatar
wafwerar 已提交
540 541 542 543 544 545 546 547
  if [ "$osType" != "Darwin" ]; then
    if ((${service_mod} == 0)); then
      install_service_on_systemd
    elif ((${service_mod} == 1)); then
      install_service_on_sysvinit
    else
      kill_taosd
    fi
wafwerar's avatar
wafwerar 已提交
548
  else
wafwerar's avatar
wafwerar 已提交
549 550 551 552 553 554 555 556 557 558 559 560 561 562
    install_service_on_launchctl
  fi
}
function install_app() {
  if [ "$osType" = "Darwin" ]; then
    ${csudo}rm -rf /Applications/TDengine.app &&
      ${csudo}mkdir -p /Applications/TDengine.app/Contents/MacOS/ &&
      ${csudo}cp ${script_dir}/TDengine /Applications/TDengine.app/Contents/MacOS/ &&
      echo "<plist><dict></dict></plist>" | ${csudo}tee /Applications/TDengine.app/Contents/Info.plist > /dev/null &&
      ${csudo}sips -i ${script_dir}/logo.png > /dev/null && 
      DeRez -only icns ${script_dir}/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
wafwerar's avatar
wafwerar 已提交
563
  fi
wafwerar's avatar
wafwerar 已提交
564 565 566
}

function update_TDengine() {
wafwerar's avatar
wafwerar 已提交
567 568 569 570 571 572 573 574
  echo -e "${GREEN}Start to update ${productName}...${NC}"
  # Stop the service if running

  if pidof ${serverName} &>/dev/null; then
    if ((${service_mod} == 0)); then
      ${csudo}systemctl stop ${serverName} || :
    elif ((${service_mod} == 1)); then
      ${csudo}service ${serverName} stop || :
wafwerar's avatar
wafwerar 已提交
575
    else
wafwerar's avatar
wafwerar 已提交
576 577
      kill_taosadapter
      kill_taosd
wafwerar's avatar
wafwerar 已提交
578
    fi
wafwerar's avatar
wafwerar 已提交
579 580 581 582 583 584 585 586 587 588
    sleep 1
  fi

  install_main_path

  install_log
  install_header
  install_lib
  #  install_connector
  install_examples
589
  install_web
wafwerar's avatar
wafwerar 已提交
590
  install_bin
wafwerar's avatar
wafwerar 已提交
591
  install_app
wafwerar's avatar
wafwerar 已提交
592 593 594 595 596 597 598 599 600 601 602 603

  install_service
  install_taosadapter_service

  install_config
  install_taosadapter_config

  echo
  echo -e "\033[44;32;1m${productName} is updated successfully!${NC}"
  echo

  echo -e "${GREEN_DARK}To configure ${productName} ${NC}: edit ${configDir}/${configFile}"
604 605
  [ -f ${configDir}/taosadapter.toml ] && [ -f ${installDir}/bin/taosadapter ] && \
    echo -e "${GREEN_DARK}To configure Taos Adapter ${NC}: edit ${configDir}/taosadapter.toml"
wafwerar's avatar
wafwerar 已提交
606 607
  if ((${service_mod} == 0)); then
    echo -e "${GREEN_DARK}To start ${productName}     ${NC}: ${csudo}systemctl start ${serverName}${NC}"
608 609
    [ -f ${service_config_dir}/taosadapter.service ] && [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adatper ${NC}: ${csudo}systemctl start taosadapter ${NC}"
wafwerar's avatar
wafwerar 已提交
610 611
  elif ((${service_mod} == 1)); then
    echo -e "${GREEN_DARK}To start ${productName}     ${NC}: ${csudo}service ${serverName} start${NC}"
612 613
    [ -f ${service_config_dir}/taosadapter.service ] && [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adapter ${NC}: ${csudo}service taosadapter start${NC}"
wafwerar's avatar
wafwerar 已提交
614
  else
wafwerar's avatar
wafwerar 已提交
615 616 617 618 619
    if [ "$osType" != "Darwin" ]; then
      echo -e "${GREEN_DARK}To start ${productName}     ${NC}: ${serverName}${NC}"
    else
      echo -e "${GREEN_DARK}To start service      ${NC}: launchctl start ${serverName}${NC}"
    fi
620 621
    [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adapter ${NC}: taosadapter &${NC}"
wafwerar's avatar
wafwerar 已提交
622 623 624 625 626
  fi

  echo -e "${GREEN_DARK}To access ${productName}    ${NC}: use ${GREEN_UNDERLINE}${clientName}${NC} in shell${NC}"
  echo
  echo -e "\033[44;32;1m${productName} is updated successfully!${NC}"
wafwerar's avatar
wafwerar 已提交
627 628 629
}

function install_TDengine() {
wafwerar's avatar
wafwerar 已提交
630 631 632 633 634 635 636 637 638 639 640 641
  # Start to install
  echo -e "${GREEN}Start to install ${productName}...${NC}"

  install_main_path

  install_data
  install_log
  install_header
  install_lib
  #  install_connector
  install_examples
  install_bin
wafwerar's avatar
wafwerar 已提交
642
  install_app
wafwerar's avatar
wafwerar 已提交
643 644 645 646 647 648 649 650 651 652 653 654

  install_service
  install_taosadapter_service

  install_config
  install_taosadapter_config

  # Ask if to start the service
  echo
  echo -e "\033[44;32;1m${productName} is installed successfully!${NC}"
  echo
  echo -e "${GREEN_DARK}To configure ${productName} ${NC}: edit ${configDir}/${configFile}"
655 656
  [ -f ${configDir}/taosadapter.toml ] && [ -f ${installDir}/bin/taosadapter ] && \
    echo -e "${GREEN_DARK}To configure Taos Adapter ${NC}: edit ${configDir}/taosadapter.toml"
wafwerar's avatar
wafwerar 已提交
657 658
  if ((${service_mod} == 0)); then
    echo -e "${GREEN_DARK}To start ${productName}     ${NC}: ${csudo}systemctl start ${serverName}${NC}"
659 660
    [ -f ${service_config_dir}/taosadapter.service ] && [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adapter ${NC}: ${csudo}systemctl start taosadapter ${NC}"
wafwerar's avatar
wafwerar 已提交
661 662
  elif ((${service_mod} == 1)); then
    echo -e "${GREEN_DARK}To start ${productName}    ${NC}: ${csudo}service ${serverName} start${NC}"
663 664
    [ -f ${service_config_dir}/taosadapter.service ] && [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adapter ${NC}: ${csudo}service taosadapter start${NC}"
wafwerar's avatar
wafwerar 已提交
665
  else
wafwerar's avatar
wafwerar 已提交
666 667 668 669 670
    if [ "$osType" != "Darwin" ]; then
      echo -e "${GREEN_DARK}To start ${productName}     ${NC}: ${serverName}${NC}"
    else
      echo -e "${GREEN_DARK}To start service      ${NC}: launchctl start ${serverName}${NC}"
    fi
671 672
    [ -f ${installDir}/bin/taosadapter ] && \
      echo -e "${GREEN_DARK}To start Taos Adapter ${NC}: taosadapter &${NC}"
wafwerar's avatar
wafwerar 已提交
673 674 675 676 677
  fi

  echo -e "${GREEN_DARK}To access ${productName}    ${NC}: use ${GREEN_UNDERLINE}${clientName}${NC} in shell${NC}"
  echo
  echo -e "\033[44;32;1m${productName} is installed successfully!${NC}"
wafwerar's avatar
wafwerar 已提交
678 679 680 681 682
}

## ==============================Main program starts from here============================
echo source directory: $1
echo binary directory: $2
wafwerar's avatar
wafwerar 已提交
683 684
if [ -x ${data_dir}/dnode/dnodeCfg.json ]; then
  echo -e "\033[44;31;5mThe default data directory ${data_dir} contains old data of tdengine 2.x, please clear it before installing!\033[0m"
wafwerar's avatar
wafwerar 已提交
685
else
wafwerar's avatar
wafwerar 已提交
686
  if [ -x ${bin_dir}/${clientName} ]; then
wafwerar's avatar
wafwerar 已提交
687 688
    update_TDengine
  else
wafwerar's avatar
wafwerar 已提交
689
    install_TDengine
wafwerar's avatar
wafwerar 已提交
690
  fi
wafwerar's avatar
wafwerar 已提交
691
fi