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

set -e
7
#set -x
H
hzcheng 已提交
8

9
# -----------------------Variables definition
H
hzcheng 已提交
10 11
source_dir=$1
binary_dir=$2
S
slguan 已提交
12
osType=$3
H
Hui Li 已提交
13
verNumber=$4
S
slguan 已提交
14 15 16 17 18 19 20

if [ "$osType" != "Darwin" ]; then
    script_dir=$(dirname $(readlink -f "$0"))
else
    script_dir=${source_dir}/packaging/tools
fi

H
hzcheng 已提交
21
# Dynamic directory
22

S
slguan 已提交
23
if [ "$osType" != "Darwin" ]; then
24
    data_dir="/var/lib/taos"
S
slguan 已提交
25
    log_dir="/var/log/taos"
H
hzcheng 已提交
26

27
    cfg_install_dir="/etc/taos"
H
hzcheng 已提交
28

S
slguan 已提交
29 30
    bin_link_dir="/usr/bin"
    lib_link_dir="/usr/lib"
H
Hui Li 已提交
31
    lib64_link_dir="/usr/lib64"
S
slguan 已提交
32
    inc_link_dir="/usr/include"
33 34 35 36

    install_main_dir="/usr/local/taos"

    bin_dir="/usr/local/taos/bin"
37
else
38 39 40 41 42
    data_dir="/usr/local/var/lib/taos"
    log_dir="/usr/local/var/log/taos"

    cfg_install_dir="/usr/local/etc/taos"

43 44 45
    bin_link_dir="/usr/local/bin"
    lib_link_dir="/usr/local/lib"
    inc_link_dir="/usr/local/include"
H
hzcheng 已提交
46

47
    install_main_dir="/usr/local/Cellar/tdengine/${verNumber}"
48
    install_main_2_dir="/usr/local/Cellar/tdengine@${verNumber}/${verNumber}"
49

50
    bin_dir="/usr/local/Cellar/tdengine/${verNumber}/bin"
51
    bin_2_dir="/usr/local/Cellar/tdengine@${verNumber}/${verNumber}/bin"
52
fi
H
hzcheng 已提交
53 54 55 56 57 58 59 60 61 62

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'

P
plum-lihui 已提交
63 64
csudo=""

65 66 67
service_mod=2
os_type=0

S
slguan 已提交
68
if [ "$osType" != "Darwin" ]; then
69 70 71
    if command -v sudo > /dev/null; then
    csudo="sudo"
    fi
S
slguan 已提交
72 73 74 75 76
    initd_mod=0
    if pidof systemd &> /dev/null; then
        service_mod=0
    elif $(which service &> /dev/null); then
        service_mod=1
77
        service_config_dir="/etc/init.d"
S
slguan 已提交
78
        if $(which chkconfig &> /dev/null); then
79
            initd_mod=1
S
slguan 已提交
80 81 82 83 84 85 86
        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
L
lihui 已提交
87 88 89
    else
        service_mod=2
    fi
H
hzcheng 已提交
90

S
slguan 已提交
91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
    # 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
L
lihui 已提交
112 113 114
        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
S
slguan 已提交
115
    fi
L
lihui 已提交
116 117
fi

118 119
function kill_taosadapter() {
    pid=$(ps -ef | grep "taosadapter" | grep -v "grep" | awk '{print $2}')
120 121 122 123 124
    if [ -n "$pid" ]; then
        ${csudo} kill -9 $pid   || :
    fi
}

H
huili 已提交
125
function kill_taosd() {
126 127
    ps -ef | grep "taosd"
    pid=$(ps -ef | grep -w "taosd" | grep -v "grep" | awk '{print $2}')
S
slguan 已提交
128 129 130
    if [ -n "$pid" ]; then
        ${csudo} kill -9 $pid   || :
    fi
H
huili 已提交
131 132
}

H
hzcheng 已提交
133 134
function install_main_path() {
    #create install main dir and all sub dir
S
slguan 已提交
135
    if [ "$osType" != "Darwin" ]; then
136 137 138 139 140 141 142 143
        ${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
S
slguan 已提交
144
        ${csudo} mkdir -p ${install_main_dir}/init.d
145 146 147 148 149 150 151 152 153
    else
        ${csudo} rm -rf ${install_main_dir}             || ${csudo} rm -rf ${install_main_2_dir}             || :
        ${csudo} mkdir -p ${install_main_dir}           || ${csudo} mkdir -p ${install_main_2_dir}
        ${csudo} mkdir -p ${install_main_dir}/cfg       || ${csudo} mkdir -p ${install_main_2_dir}/cfg
        ${csudo} mkdir -p ${install_main_dir}/bin       || ${csudo} mkdir -p ${install_main_2_dir}/bin
        ${csudo} mkdir -p ${install_main_dir}/connector || ${csudo} mkdir -p ${install_main_2_dir}/connector
        ${csudo} mkdir -p ${install_main_dir}/driver    || ${csudo} mkdir -p ${install_main_2_dir}/driver
        ${csudo} mkdir -p ${install_main_dir}/examples  || ${csudo} mkdir -p ${install_main_2_dir}/examples
        ${csudo} mkdir -p ${install_main_dir}/include   || ${csudo} mkdir -p ${install_main_2_dir}/include
S
slguan 已提交
154
    fi
H
hzcheng 已提交
155 156 157 158
}

function install_bin() {
    # Remove links
159 160
    ${csudo} rm -f ${bin_link_dir}/taos     || :
    ${csudo} rm -f ${bin_link_dir}/taosd    || :
161
    ${csudo} rm -f ${bin_link_dir}/taosadapter || :
162 163 164
    ${csudo} rm -f ${bin_link_dir}/taosdemo || :
    ${csudo} rm -f ${bin_link_dir}/taosdump || :

S
slguan 已提交
165
    if [ "$osType" != "Darwin" ]; then
166
        ${csudo} rm -f ${bin_link_dir}/perfMonitor || :
H
Hui Li 已提交
167
        ${csudo} rm -f ${bin_link_dir}/set_core || :
168
        ${csudo} rm -f ${bin_link_dir}/rmtaos   || :
169

170 171
        ${csudo} cp -r ${binary_dir}/build/bin/* ${install_main_dir}/bin
        ${csudo} cp -r ${script_dir}/taosd-dump-cfg.gdb   ${install_main_dir}/bin
172

173
        ${csudo} cp -r ${script_dir}/remove.sh     ${install_main_dir}/bin
H
Hui Li 已提交
174
        ${csudo} cp -r ${script_dir}/set_core.sh   ${install_main_dir}/bin
175
        ${csudo} cp -r ${script_dir}/startPre.sh   ${install_main_dir}/bin
176

177 178 179 180
        ${csudo} chmod 0555 ${install_main_dir}/bin/*
        #Make link
        [ -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   || :
181
        [ -x ${install_main_dir}/bin/taosadapter ]  && ${csudo} ln -s ${install_main_dir}/bin/taosadapter ${bin_link_dir}/taosadapter || :
182 183
        [ -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 || :
184
        [ -x ${install_main_dir}/bin/perfMonitor ]  && ${csudo} ln -s ${install_main_dir}/bin/perfMonitor ${bin_link_dir}/perfMonitor || :
H
Hui Li 已提交
185
        [ -x ${install_main_dir}/set_core.sh ]  && ${csudo} ln -s ${install_main_dir}/bin/set_core.sh ${bin_link_dir}/set_core || :
186 187
        [ -x ${install_main_dir}/bin/remove.sh ] && ${csudo} ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/rmtaos  || :
   else
188

189 190 191 192 193 194 195
        ${csudo} cp -r ${binary_dir}/build/bin/*   ${install_main_dir}/bin || ${csudo} cp -r ${binary_dir}/build/bin/*   ${install_main_2_dir}/bin  || :
        ${csudo} cp -r ${script_dir}/taosd-dump-cfg.gdb   ${install_main_dir}/bin || ${csudo}cp -r ${script_dir}/taosd-dump-cfg.gdb   ${install_main_2_dir}  || :
        ${csudo} cp -r ${script_dir}/remove_client.sh   ${install_main_dir}/bin || ${csudo} cp -r ${script_dir}/remove_client.sh   ${install_main_2_dir}/bin
        ${csudo} chmod 0555 ${install_main_dir}/bin/* || ${csudo} chmod 0555 ${install_main_2_dir}/bin/*
        #Make link
        [ -x ${install_main_dir}/bin/taos ] || [ -x ${install_main_2_dir}/bin/taos ] && ${csudo} ln -s ${install_main_dir}/bin/taos ${bin_link_dir}/taos || ${csudo} ln -s ${install_main_2_dir}/bin/taos  || :
        [ -x ${install_main_dir}/bin/taosd ] || [ -x ${install_main_2_dir}/bin/taosd ] &&  ${csudo} ln -s ${install_main_dir}/bin/taosd ${bin_link_dir}/taosd || ${csudo} ln -s ${install_main_2_dir}/bin/taosd || :
196
        [ -x ${install_main_dir}/bin/taosadapter ] || [ -x ${install_main_2_dir}/bin/taosadapter ] &&  ${csudo} ln -s ${install_main_dir}/bin/taosadapter ${bin_link_dir}/taosadapter || ${csudo} ln -s ${install_main_2_dir}/bin/taosadapter || :
197 198
        [ -x ${install_main_dir}/bin/taosdump ] || [ -x ${install_main_2_dir}/bin/taosdump ] && ${csudo} ln -s ${install_main_dir}/bin/taosdump ${bin_link_dir}/taosdump || ln -s ${install_main_2_dir}/bin/taosdump ${bin_link_dir}/taosdump   || :
        [ -x ${install_main_dir}/bin/taosdemo ] || [ -x ${install_main_2_dir}/bin/taosdemo ] && ${csudo} ln -s ${install_main_dir}/bin/taosdemo ${bin_link_dir}/taosdemo || ln -s ${install_main_2_dir}/bin/taosdemo ${bin_link_dir}/taosdemo   || :
S
slguan 已提交
199
    fi
H
hzcheng 已提交
200
}
201

202 203 204 205
function install_jemalloc() {
    if [ "$osType" != "Darwin" ]; then
        /usr/bin/install -c -d /usr/local/bin

206
        if [ -f "${binary_dir}/build/bin/jemalloc-config" ]; then
207
            ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/bin/jemalloc-config /usr/local/bin
208
        fi
209
        if [ -f "${binary_dir}/build/bin/jemalloc.sh" ]; then
210
            ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/bin/jemalloc.sh /usr/local/bin
211
        fi
212
        if [ -f "${binary_dir}/build/bin/jeprof" ]; then
213
            ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/bin/jeprof /usr/local/bin
214
        fi
215
        if [ -f "${binary_dir}/build/include/jemalloc/jemalloc.h" ]; then
216 217
            ${csudo} /usr/bin/install -c -d /usr/local/include/jemalloc
            ${csudo} /usr/bin/install -c -m 644 ${binary_dir}/build/include/jemalloc/jemalloc.h\
218
                /usr/local/include/jemalloc
219
        fi
220
        if [ -f "${binary_dir}/build/lib/libjemalloc.so.2" ]; then
221 222 223 224
            ${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
225
            [ -f ${binary_dir}/build/lib/libjemalloc.a ] &&
226
                ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/lib/libjemalloc.a /usr/local/lib
227
            [ -f ${binary_dir}/build/lib/libjemalloc_pic.a ] &&
228
                ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/lib/libjemalloc_pic.a /usr/local/lib
229
            if [ -f "${binary_dir}/build/lib/pkgconfig/jemalloc.pc" ]; then
230 231
                ${csudo} /usr/bin/install -c -d /usr/local/lib/pkgconfig
                ${csudo} /usr/bin/install -c -m 644 ${binary_dir}/build/lib/pkgconfig/jemalloc.pc\
232 233 234
                    /usr/local/lib/pkgconfig
            fi
            if [ -d /etc/ld.so.conf.d ]; then
235
                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"
236 237 238
                ${csudo} ldconfig
            else
                echo "/etc/ld.so.conf.d not found!"
239 240
            fi
        fi
241
        if [ -f "${binary_dir}/build/share/doc/jemalloc/jemalloc.html" ]; then
242 243
            ${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\
244
                /usr/local/share/doc/jemalloc
245
        fi
246
        if [ -f "${binary_dir}/build/share/man/man3/jemalloc.3" ]; then
247 248
            ${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\
249
                /usr/local/share/man/man3
250
        fi
251 252 253 254 255
    fi
}

function install_avro() {
    if [ "$osType" != "Darwin" ]; then
256
        if [ -f "${binary_dir}/build/$1/libavro.so.23.0.0" ] && [ -d /usr/local/$1 ]; then
257 258 259 260 261
            ${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
262
            [ -f ${binary_dir}/build/$1/libavro.a ] &&
263
                ${csudo} /usr/bin/install -c -m 755 ${binary_dir}/build/$1/libavro.a /usr/local/$1
264 265

            if [ -d /etc/ld.so.conf.d ]; then
266
                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"
267 268 269 270
                ${csudo} ldconfig
            else
                echo "/etc/ld.so.conf.d not found!"
            fi
271
        fi
272 273
    fi
}
H
hzcheng 已提交
274 275 276

function install_lib() {
    # Remove links
S
slguan 已提交
277
    ${csudo} rm -f ${lib_link_dir}/libtaos.*     || :
278 279 280
    if [ "$osType" != "Darwin" ]; then
      ${csudo} rm -f ${lib64_link_dir}/libtaos.*   || :
    fi
281

S
slguan 已提交
282
    if [ "$osType" != "Darwin" ]; then
283 284 285 286
        ${csudo} cp ${binary_dir}/build/lib/libtaos.so.${verNumber} \
            ${install_main_dir}/driver \
            && ${csudo} chmod 777 ${install_main_dir}/driver/*

287
        ${csudo} ln -sf ${install_main_dir}/driver/libtaos.* ${lib_link_dir}/libtaos.so.1
S
slguan 已提交
288
        ${csudo} ln -sf ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
289

H
Hui Li 已提交
290
        if [ -d "${lib64_link_dir}" ]; then
291
          ${csudo} ln -sf ${install_main_dir}/driver/libtaos.* ${lib64_link_dir}/libtaos.so.1
H
Hui Li 已提交
292 293
          ${csudo} ln -sf ${lib64_link_dir}/libtaos.so.1 ${lib64_link_dir}/libtaos.so
        fi
S
slguan 已提交
294
    else
295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315
        ${csudo} cp -Rf ${binary_dir}/build/lib/libtaos.${verNumber}.dylib \
            ${install_main_dir}/driver \
            || ${csudo} cp -Rf ${binary_dir}/build/lib/libtaos.${verNumber}.dylib \
            ${install_main_2_dir}/driver \
            && ${csudo} chmod 777 ${install_main_dir}/driver/* \
            || ${csudo} chmod 777 ${install_main_2_dir}/driver/*

        ${csudo} ln -sf ${install_main_dir}/driver/libtaos.* \
            ${install_main_dir}/driver/libtaos.1.dylib \
            || ${csudo} ln -sf ${install_main_2_dir}/driver/libtaos.* \
            ${install_main_2_dir}/driver/libtaos.1.dylib   || :

        ${csudo} ln -sf ${install_main_dir}/driver/libtaos.1.dylib \
            ${install_main_dir}/driver/libtaos.dylib \
            || ${csudo} ln -sf ${install_main_2_dir}/driver/libtaos.1.dylib \
            ${install_main_2_dir}/driver/libtaos.dylib   || :

        ${csudo} ln -sf ${install_main_dir}/driver/libtaos.${verNumber}.dylib \
            ${lib_link_dir}/libtaos.1.dylib \
            || ${csudo} ln -sf ${install_main_2_dir}/driver/libtaos.${verNumber}.dylib \
            ${lib_link_dir}/libtaos.1.dylib   || :
316

317
        ${csudo} ln -sf ${lib_link_dir}/libtaos.1.dylib ${lib_link_dir}/libtaos.dylib   || :
S
slguan 已提交
318
    fi
319

320
    install_jemalloc
321 322
    install_avro lib
    install_avro lib64
323

324 325 326
    if [ "$osType" != "Darwin" ]; then
        ${csudo} ldconfig
    fi
H
hzcheng 已提交
327 328 329 330
}

function install_header() {

331
    if [ "$osType" != "Darwin" ]; then
332 333
        ${csudo} rm -f ${inc_link_dir}/taos.h ${inc_link_dir}/taosdef.h ${inc_link_dir}/taoserror.h     || :
        ${csudo} cp -f ${source_dir}/src/inc/taos.h ${source_dir}/src/inc/taosdef.h ${source_dir}/src/inc/taoserror.h \
334
            ${install_main_dir}/include && ${csudo} chmod 644 ${install_main_dir}/include/*
335
        ${csudo} ln -s ${install_main_dir}/include/taos.h ${inc_link_dir}/taos.h
336
        ${csudo} ln -s ${install_main_dir}/include/taosdef.h ${inc_link_dir}/taosdef.h
337
        ${csudo} ln -s ${install_main_dir}/include/taoserror.h ${inc_link_dir}/taoserror.h
338
    else
339
        ${csudo} cp -f ${source_dir}/src/inc/taos.h ${source_dir}/src/inc/taosdef.h ${source_dir}/src/inc/taoserror.h \
340
            ${install_main_dir}/include \
341
            || ${csudo} cp -f ${source_dir}/src/inc/taos.h ${source_dir}/src/inc/taosdef.h ${source_dir}/src/inc/taoserror.h \
342 343 344
            ${install_main_2_dir}/include \
            && ${csudo} chmod 644 ${install_main_dir}/include/* \
            || ${csudo} chmod 644 ${install_main_2_dir}/include/*
345
    fi
H
hzcheng 已提交
346 347 348
}

function install_config() {
349
    if [ ! -f ${cfg_install_dir}/taos.cfg ]; then
P
plum-lihui 已提交
350
        ${csudo} mkdir -p ${cfg_install_dir}
351 352
        [ -f ${script_dir}/../cfg/taos.cfg ] &&
        ${csudo} cp ${script_dir}/../cfg/taos.cfg ${cfg_install_dir}
353
        ${csudo} chmod 644 ${cfg_install_dir}/taos.cfg
354 355 356 357
        ${csudo} cp -f ${script_dir}/../cfg/taos.cfg \
            ${cfg_install_dir}/taos.cfg.${verNumber}
        ${csudo} ln -s ${cfg_install_dir}/taos.cfg \
            ${install_main_dir}/cfg/taos.cfg
358
    else
359 360
        ${csudo} cp -f ${script_dir}/../cfg/taos.cfg \
            ${cfg_install_dir}/taos.cfg.${verNumber}
361
    fi
H
hzcheng 已提交
362 363
}

364 365
function install_taosadapter_config() {
    if [ ! -f "${cfg_install_dir}/taosadapter.toml" ]; then
366
        ${csudo} mkdir -p ${cfg_install_dir}
367 368 369 370 371 372 373 374 375 376
        [ -f ${binary_dir}/test/cfg/taosadapter.toml ] &&
            ${csudo} cp ${binary_dir}/test/cfg/taosadapter.toml ${cfg_install_dir}
        [ -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
377
    else
378 379 380
        if [ -f "${binary_dir}/test/cfg/taosadapter.toml" ]; then
            ${csudo} cp -f ${binary_dir}/test/cfg/taosadapter.toml \
                ${cfg_install_dir}/taosadapter.toml.${verNumber}
381
        fi
382 383 384
    fi
}

385
function install_log() {
386 387
    ${csudo} rm -rf ${log_dir}  || :
    ${csudo} mkdir -p ${log_dir} && ${csudo} chmod 777 ${log_dir}
388 389 390 391 392
    if [ "$osType" != "Darwin" ]; then
        ${csudo} ln -s ${log_dir} ${install_main_dir}/log
    else
        ${csudo} ln -s ${log_dir} ${install_main_dir}/log || ${csudo} ln -s ${log_dir} ${install_main_2_dir}/log
    fi
H
hzcheng 已提交
393 394 395
}

function install_data() {
396
    ${csudo} mkdir -p ${data_dir}
397 398 399 400 401
    if [ "$osType" != "Darwin" ]; then
        ${csudo} ln -s ${data_dir} ${install_main_dir}/data
    else
        ${csudo} ln -s ${data_dir} ${install_main_dir}/data || ${csudo} ln -s ${data_dir} ${install_main_2_dir}/data
    fi
H
hzcheng 已提交
402 403 404
}

function install_connector() {
405 406 407 408 409
    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
410 411 412 413
    if [ "$osType" != "Darwin" ]; then
        ${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
    else
414 415 416
        ${csudo} cp -rf ${source_dir}/src/connector/python ${install_main_dir}/connector || ${csudo} cp -rf ${source_dir}/src/connector/python ${install_main_2_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
        ${csudo} cp ${binary_dir}/build/lib/*.jar ${install_main_2_dir}/connector &> /dev/null && ${csudo} chmod 777 ${install_main_2_dir}/connector/*.jar  || echo &> /dev/null
417
    fi
H
hzcheng 已提交
418 419 420
}

function install_examples() {
421 422 423 424 425
    if [ "$osType" != "Darwin" ]; then
       ${csudo} cp -rf ${source_dir}/tests/examples/* ${install_main_dir}/examples
    else
       ${csudo} cp -rf ${source_dir}/tests/examples/* ${install_main_dir}/examples || ${csudo} cp -rf ${source_dir}/tests/examples/* ${install_main_2_dir}/examples
    fi
H
hzcheng 已提交
426 427 428
}

function clean_service_on_sysvinit() {
L
lihui 已提交
429
    #restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
430 431
    #${csudo} sed -i "\|${restart_config_str}|d" /etc/inittab || :

H
hzcheng 已提交
432
    if pidof taosd &> /dev/null; then
P
plum-lihui 已提交
433
        ${csudo} service taosd stop || :
H
hzcheng 已提交
434
    fi
H
huili 已提交
435 436

    if ((${initd_mod}==1)); then
L
lihui 已提交
437
        ${csudo} chkconfig --del taosd || :
H
huili 已提交
438
    elif ((${initd_mod}==2)); then
L
lihui 已提交
439 440 441 442
        ${csudo} insserv -r taosd || :
    elif ((${initd_mod}==3)); then
        ${csudo} update-rc.d -f taosd remove || :
    fi
443

L
lihui 已提交
444
    ${csudo} rm -f ${service_config_dir}/taosd || :
445

L
lihui 已提交
446 447
    if $(which init &> /dev/null); then
        ${csudo} init q || :
H
huili 已提交
448
    fi
H
hzcheng 已提交
449 450 451 452 453 454 455 456
}

function install_service_on_sysvinit() {
    clean_service_on_sysvinit

    sleep 1

    # Install taosd service
L
lihui 已提交
457
    if ((${os_type}==1)); then
458 459
        ${csudo} cp -f ${script_dir}/../deb/taosd ${install_main_dir}/init.d
        ${csudo} cp    ${script_dir}/../deb/taosd ${service_config_dir} && ${csudo} chmod a+x ${service_config_dir}/taosd
L
lihui 已提交
460
    elif ((${os_type}==2)); then
461 462
        ${csudo} cp -f ${script_dir}/../rpm/taosd ${install_main_dir}/init.d
        ${csudo} cp    ${script_dir}/../rpm/taosd ${service_config_dir} && ${csudo} chmod a+x ${service_config_dir}/taosd
L
lihui 已提交
463
    fi
464

L
lihui 已提交
465 466
    #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"
467

H
huili 已提交
468
    if ((${initd_mod}==1)); then
L
lihui 已提交
469 470
        ${csudo} chkconfig --add taosd || :
        ${csudo} chkconfig --level 2345 taosd on || :
H
huili 已提交
471
    elif ((${initd_mod}==2)); then
L
lihui 已提交
472 473 474
        ${csudo} insserv taosd || :
        ${csudo} insserv -d taosd || :
    elif ((${initd_mod}==3)); then
H
huili 已提交
475 476
        ${csudo} update-rc.d taosd defaults || :
    fi
H
hzcheng 已提交
477 478 479 480 481 482 483
}

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 已提交
484
        ${csudo} systemctl stop taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
485
    fi
P
plum-lihui 已提交
486
    ${csudo} systemctl disable taosd &> /dev/null || echo &> /dev/null
H
hzcheng 已提交
487

P
plum-lihui 已提交
488
    ${csudo} rm -f ${taosd_service_config}
489
}
H
hzcheng 已提交
490 491 492 493 494 495 496 497

# 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 已提交
498 499
    ${csudo} bash -c "echo '[Unit]'                             >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Description=TDengine server service' >> ${taosd_service_config}"
500 501
    ${csudo} bash -c "echo 'After=network-online.target taosadapter.service'        >> ${taosd_service_config}"
    ${csudo} bash -c "echo 'Wants=network-online.target taosadapter.service'        >> ${taosd_service_config}"
P
plum-lihui 已提交
502 503 504 505
    ${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}"
506
    ${csudo} bash -c "echo 'ExecStartPre=/usr/local/taos/bin/startPre.sh'           >> ${taosd_service_config}"
507
    ${csudo} bash -c "echo 'TimeoutStopSec=1000000s'            >> ${taosd_service_config}"
P
plum-lihui 已提交
508 509 510 511 512 513 514 515 516 517 518 519
    ${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
H
hzcheng 已提交
520 521
}

522
function install_taosadapter_service() {
523 524 525
    if ((${service_mod}==0)); then
        [ -f ${binary_dir}/test/cfg/taosadapter.service ] &&\
            ${csudo} cp ${binary_dir}/test/cfg/taosadapter.service\
526 527
            ${service_config_dir}/ || :
        ${csudo} systemctl daemon-reload
528
    fi
529 530
}

H
hzcheng 已提交
531
function install_service() {
H
huili 已提交
532
    if ((${service_mod}==0)); then
H
hzcheng 已提交
533
        install_service_on_systemd
H
huili 已提交
534
    elif ((${service_mod}==1)); then
H
hzcheng 已提交
535
        install_service_on_sysvinit
H
huili 已提交
536
    else
L
lihui 已提交
537
        # must manual stop taosd
H
huili 已提交
538
        kill_taosd
H
hzcheng 已提交
539 540 541 542
    fi
}

function update_TDengine() {
543
    echo -e "${GREEN}Start to update TDengine...${NC}"
H
hzcheng 已提交
544
    # Stop the service if running
S
slguan 已提交
545 546

      if pidof taosd &> /dev/null; then
H
huili 已提交
547
        if ((${service_mod}==0)); then
P
plum-lihui 已提交
548
            ${csudo} systemctl stop taosd || :
H
huili 已提交
549
        elif ((${service_mod}==1)); then
P
plum-lihui 已提交
550
            ${csudo} service taosd stop || :
H
huili 已提交
551
        else
552
            kill_taosadapter
H
huili 已提交
553
            kill_taosd
H
hzcheng 已提交
554 555
        fi
        sleep 1
S
slguan 已提交
556
      fi
557

H
hzcheng 已提交
558 559 560 561 562 563 564
    install_main_path

    install_log
    install_header
    install_lib
    install_connector
    install_examples
H
huili 已提交
565
    install_bin
S
slguan 已提交
566

567 568
    install_service
    install_taosadapter_service
S
slguan 已提交
569

H
huili 已提交
570
    install_config
571
    install_taosadapter_config
H
hzcheng 已提交
572

573 574 575
    echo
    echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
    echo
S
slguan 已提交
576

577 578 579 580 581 582
    echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
    echo -e "${GREEN_DARK}To configure Taos Adapter (if has) ${NC}: edit /etc/taos/taosadapter.toml"
    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} service taosd start${NC}"
S
slguan 已提交
583
    else
584 585
        echo -e "${GREEN_DARK}To start Taos Adapter (if has)${NC}: taosadapter &${NC}"
        echo -e "${GREEN_DARK}To start TDengine     ${NC}: taosd${NC}"
S
slguan 已提交
586
    fi
587 588 589 590

    echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
    echo
    echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
H
hzcheng 已提交
591 592 593 594
}

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

597
    install_main_path
S
slguan 已提交
598

599
    install_data
600
    install_log
H
hzcheng 已提交
601 602 603 604
    install_header
    install_lib
    install_connector
    install_examples
H
huili 已提交
605
    install_bin
606

607 608
    install_service
    install_taosadapter_service
609

610
    install_config
611
    install_taosadapter_config
S
slguan 已提交
612

613 614 615 616 617 618 619 620 621 622
    # 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"
    echo -e "${GREEN_DARK}To configure taosadapter (if has) ${NC}: edit /etc/taos/taosadapter.toml"
    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} service taosd start${NC}"
H
huili 已提交
623
    else
624
        echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
H
huili 已提交
625
    fi
626 627 628 629

    echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
    echo
    echo -e "\033[44;32;1mTDengine is installed successfully!${NC}"
H
hzcheng 已提交
630 631 632 633 634
}

## ==============================Main program starts from here============================
echo source directory: $1
echo binary directory: $2
635 636 637 638 639 640
if [ "$osType" != "Darwin" ]; then
    if [ -x ${bin_dir}/taos ]; then
        update_TDengine
    else
        install_TDengine
    fi
H
hzcheng 已提交
641
else
642 643 644 645 646
    if [ -x ${bin_dir}/taos ] || [ -x ${bin_2_dir}/taos ]; then
        update_TDengine
    else
        install_TDengine
    fi
H
hzcheng 已提交
647
fi