[Modify the use of sudo commands]

packaging/tools/install.sh

@@ -36,106 +36,114 @@ GREEN_DARK='\033[0;32m'
 GREEN_UNDERLINE='\033[4;32m'
 NC='\033[0m'
 
-function is_using_systemd() {
-    if pidof systemd &> /dev/null; then
-        return 0
-    else
-        return 1
-    fi
-}
+csudo=""
+if command -v sudo > /dev/null; then
+    csudo="sudo"
+fi
 
-if ! is_using_systemd; then
+service_mod=2
+if pidof systemd &> /dev/null; then
+    service_mod=0
+elif $(which update-rc.d &> /dev/null); then
+    service_mod=1
     service_config_dir="/etc/init.d"
+else 
+    service_mod=2
 fi
 
+function kill_taosd() {
+  pid=$(ps -ef | grep "taosd" | grep -v "grep" | awk '{print $2}')
+  ${csudo} kill -9 pid   || :
+}
+
 function install_main_path() {
     #create install main dir and all sub dir
-    sudo rm -rf ${install_main_dir}    || :
-    sudo mkdir -p ${install_main_dir}  
-    sudo mkdir -p ${install_main_dir}/cfg
-    sudo mkdir -p ${install_main_dir}/bin    
-    sudo mkdir -p ${install_main_dir}/connector
-    sudo mkdir -p ${install_main_dir}/driver
-    sudo mkdir -p ${install_main_dir}/examples
-    sudo mkdir -p ${install_main_dir}/include
-    sudo mkdir -p ${install_main_dir}/init.d
+    ${csudo} rm -rf ${install_main_dir}    || :
+    ${csudo} mkdir -p ${install_main_dir}  
+    ${csudo} mkdir -p ${install_main_dir}/cfg
+    ${csudo} mkdir -p ${install_main_dir}/bin    
+    ${csudo} mkdir -p ${install_main_dir}/connector
+    ${csudo} mkdir -p ${install_main_dir}/driver
+    ${csudo} mkdir -p ${install_main_dir}/examples
+    ${csudo} mkdir -p ${install_main_dir}/include
+    ${csudo} mkdir -p ${install_main_dir}/init.d
 }
 
 function install_bin() {
     # Remove links
-    sudo rm -f ${bin_link_dir}/taos     || :
-    sudo rm -f ${bin_link_dir}/taosd    || :
-    sudo rm -f ${bin_link_dir}/taosdump || :
-    sudo rm -f ${bin_link_dir}/rmtaos   || :
+    ${csudo} rm -f ${bin_link_dir}/taos     || :
+    ${csudo} rm -f ${bin_link_dir}/taosd    || :
+    ${csudo} rm -f ${bin_link_dir}/taosdump || :
+    ${csudo} rm -f ${bin_link_dir}/rmtaos   || :
 
-    sudo cp -r ${script_dir}/bin/* ${install_main_dir}/bin && sudo chmod 0555 ${install_main_dir}/bin/*
+    ${csudo} cp -r ${script_dir}/bin/* ${install_main_dir}/bin && ${csudo} chmod 0555 ${install_main_dir}/bin/*
 
     #Make link
-    [ -x ${install_main_dir}/bin/taos ] && sudo ln -s ${install_main_dir}/bin/taos ${bin_link_dir}/taos             || :
-    [ -x ${install_main_dir}/bin/taosd ] && sudo ln -s ${install_main_dir}/bin/taosd ${bin_link_dir}/taosd          || :   
-    [ -x ${install_main_dir}/bin/taosdump ] && sudo ln -s ${install_main_dir}/bin/taosdump ${bin_link_dir}/taosdump || :
-    [ -x ${install_main_dir}/bin/remove.sh ] && sudo ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/rmtaos || :
+    [ -x ${install_main_dir}/bin/taos ] && ${csudo} ln -s ${install_main_dir}/bin/taos ${bin_link_dir}/taos             || :
+    [ -x ${install_main_dir}/bin/taosd ] && ${csudo} ln -s ${install_main_dir}/bin/taosd ${bin_link_dir}/taosd          || :   
+    [ -x ${install_main_dir}/bin/taosdump ] && ${csudo} ln -s ${install_main_dir}/bin/taosdump ${bin_link_dir}/taosdump || :
+    [ -x ${install_main_dir}/bin/remove.sh ] && ${csudo} ln -s ${install_main_dir}/bin/remove.sh ${bin_link_dir}/rmtaos || :
 }
 
 function install_lib() {
     # Remove links
-    sudo rm -f ${lib_link_dir}/libtaos.*     || :    
+    ${csudo} rm -f ${lib_link_dir}/libtaos.*     || :    
 
-    sudo cp -rf ${script_dir}/driver/* ${install_main_dir}/driver && sudo chmod 777 ${install_main_dir}/driver/*  
+    ${csudo} cp -rf ${script_dir}/driver/* ${install_main_dir}/driver && ${csudo} chmod 777 ${install_main_dir}/driver/*  
     
-    sudo ln -s ${install_main_dir}/driver/libtaos.* ${lib_link_dir}/libtaos.so.1
-    sudo ln -s ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
+    ${csudo} ln -s ${install_main_dir}/driver/libtaos.* ${lib_link_dir}/libtaos.so.1
+    ${csudo} ln -s ${lib_link_dir}/libtaos.so.1 ${lib_link_dir}/libtaos.so
 }
 
 function install_header() {
-    sudo rm -f ${inc_link_dir}/taos.h     || :    
-    sudo cp -f ${script_dir}/inc/* ${install_main_dir}/include && sudo chmod 644 ${install_main_dir}/include/*    
-    sudo ln -s ${install_main_dir}/include/taos.h ${inc_link_dir}/taos.h
+    ${csudo} rm -f ${inc_link_dir}/taos.h     || :    
+    ${csudo} cp -f ${script_dir}/inc/* ${install_main_dir}/include && ${csudo} chmod 644 ${install_main_dir}/include/*    
+    ${csudo} ln -s ${install_main_dir}/include/taos.h ${inc_link_dir}/taos.h
 }
 
 function install_config() {
-    #sudo rm -f ${install_main_dir}/cfg/taos.cfg     || :
+    #${csudo} rm -f ${install_main_dir}/cfg/taos.cfg     || :
     
     if [ ! -f ${cfg_install_dir}/taos.cfg ]; then  
-        sudo sudo mkdir -p ${cfg_install_dir}
-        [ -f ${script_dir}/cfg/taos.cfg ] && sudo cp ${script_dir}/cfg/taos.cfg ${cfg_install_dir}
-        sudo chmod 644 ${cfg_install_dir}/*
+        ${csudo} mkdir -p ${cfg_install_dir}
+        [ -f ${script_dir}/cfg/taos.cfg ] && ${csudo} cp ${script_dir}/cfg/taos.cfg ${cfg_install_dir}
+        ${csudo} chmod 644 ${cfg_install_dir}/*
     fi 
     
-    sudo cp -f ${script_dir}/cfg/taos.cfg ${install_main_dir}/cfg/taos.cfg.org
-    sudo ln -s ${cfg_install_dir}/taos.cfg ${install_main_dir}/cfg   
+    ${csudo} cp -f ${script_dir}/cfg/taos.cfg ${install_main_dir}/cfg/taos.cfg.org
+    ${csudo} ln -s ${cfg_install_dir}/taos.cfg ${install_main_dir}/cfg   
 }
 
 function install_log() {
-    sudo rm -rf ${log_dir}  || :
-    sudo mkdir -p ${log_dir} && sudo chmod 777 ${log_dir}
+    ${csudo} rm -rf ${log_dir}  || :
+    ${csudo} mkdir -p ${log_dir} && ${csudo} chmod 777 ${log_dir}
     
-    sudo ln -s ${log_dir} ${install_main_dir}/log
+    ${csudo} ln -s ${log_dir} ${install_main_dir}/log
 }
 
 function install_data() {
-    sudo mkdir -p ${data_dir}
+    ${csudo} mkdir -p ${data_dir}
     
-    sudo ln -s ${data_dir} ${install_main_dir}/data    
+    ${csudo} ln -s ${data_dir} ${install_main_dir}/data    
 }
 
 function install_connector() {
-    sudo cp -rf ${script_dir}/connector/* ${install_main_dir}/connector
+    ${csudo} cp -rf ${script_dir}/connector/* ${install_main_dir}/connector
 }
 
 function install_examples() {
-    sudo cp -rf ${script_dir}/examples/* ${install_main_dir}/examples
+    ${csudo} cp -rf ${script_dir}/examples/* ${install_main_dir}/examples
 }
 
 function clean_service_on_sysvinit() {
     restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
     if pidof taosd &> /dev/null; then
-        sudo service taosd stop || :
+        ${csudo} service taosd stop || :
     fi
-    sudo sed -i "\|${restart_config_str}|d" /etc/inittab || :
-    sudo rm -f ${service_config_dir}/taosd || :
-    sudo update-rc.d -f taosd remove || :
-    sudo init q || :
+    ${csudo} sed -i "\|${restart_config_str}|d" /etc/inittab || :
+    ${csudo} rm -f ${service_config_dir}/taosd || :
+    ${csudo} update-rc.d -f taosd remove || :
+    ${csudo} init q || :
 }
 
 function install_service_on_sysvinit() {
@@ -144,13 +152,13 @@ function install_service_on_sysvinit() {
     sleep 1
 
     # Install taosd service
-    sudo cp -f ${script_dir}/init.d/taosd ${install_main_dir}/init.d
-    sudo cp ${script_dir}/init.d/taosd ${service_config_dir} && sudo chmod a+x ${service_config_dir}/taosd
+    ${csudo} cp -f ${script_dir}/init.d/taosd ${install_main_dir}/init.d
+    ${csudo} cp ${script_dir}/init.d/taosd ${service_config_dir} && ${csudo} chmod a+x ${service_config_dir}/taosd
     restart_config_str="taos:2345:respawn:${service_config_dir}/taosd start"
 
-    sudo grep -q -F "$restart_config_str" /etc/inittab || sudo bash -c "echo '${restart_config_str}' >> /etc/inittab"
+    ${csudo} grep -q -F "$restart_config_str" /etc/inittab || ${csudo} bash -c "echo '${restart_config_str}' >> /etc/inittab"
     # TODO: for centos, change here
-    sudo update-rc.d taosd defaults
+    ${csudo} update-rc.d taosd defaults
     # chkconfig mysqld on
 }
 
@@ -159,11 +167,11 @@ function clean_service_on_systemd() {
 
     if systemctl is-active --quiet taosd; then
         echo "TDengine is running, stopping it..."
-        sudo systemctl stop taosd &> /dev/null || echo &> /dev/null
+        ${csudo} systemctl stop taosd &> /dev/null || echo &> /dev/null
     fi
-    sudo systemctl disable taosd &> /dev/null || echo &> /dev/null
+    ${csudo} systemctl disable taosd &> /dev/null || echo &> /dev/null
 
-    sudo rm -f ${taosd_service_config}
+    ${csudo} rm -f ${taosd_service_config}
 }   
 
 # taos:2345:respawn:/etc/init.d/taosd start
@@ -173,33 +181,36 @@ function install_service_on_systemd() {
 
     taosd_service_config="${service_config_dir}/taosd.service"
 
-    sudo bash -c "echo '[Unit]'                             >> ${taosd_service_config}"
-    sudo bash -c "echo 'Description=TDengine server service' >> ${taosd_service_config}"
-    sudo bash -c "echo 'After=network-online.target'        >> ${taosd_service_config}"
-    sudo bash -c "echo 'Wants=network-online.target'        >> ${taosd_service_config}"
-    sudo bash -c "echo                                      >> ${taosd_service_config}"
-    sudo bash -c "echo '[Service]'                          >> ${taosd_service_config}"
-    sudo bash -c "echo 'Type=simple'                        >> ${taosd_service_config}"
-    sudo bash -c "echo 'ExecStart=/usr/bin/taosd'           >> ${taosd_service_config}"
-    sudo bash -c "echo 'LimitNOFILE=infinity'               >> ${taosd_service_config}"
-    sudo bash -c "echo 'LimitNPROC=infinity'                >> ${taosd_service_config}"
-    sudo bash -c "echo 'LimitCORE=infinity'                 >> ${taosd_service_config}"
-    sudo bash -c "echo 'TimeoutStartSec=0'                  >> ${taosd_service_config}"
-    sudo bash -c "echo 'StandardOutput=null'                >> ${taosd_service_config}"
-    sudo bash -c "echo 'Restart=always'                     >> ${taosd_service_config}"
-    sudo bash -c "echo 'StartLimitBurst=3'                  >> ${taosd_service_config}"
-    sudo bash -c "echo 'StartLimitInterval=60s'             >> ${taosd_service_config}"
-    sudo bash -c "echo                                      >> ${taosd_service_config}"
-    sudo bash -c "echo '[Install]'                          >> ${taosd_service_config}"
-    sudo bash -c "echo 'WantedBy=multi-user.target'         >> ${taosd_service_config}"
-    sudo systemctl enable taosd
+    ${csudo} bash -c "echo '[Unit]'                             >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'Description=TDengine server service' >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'After=network-online.target'        >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'Wants=network-online.target'        >> ${taosd_service_config}"
+    ${csudo} bash -c "echo                                      >> ${taosd_service_config}"
+    ${csudo} bash -c "echo '[Service]'                          >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'Type=simple'                        >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'ExecStart=/usr/bin/taosd'           >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'LimitNOFILE=infinity'               >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'LimitNPROC=infinity'                >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'LimitCORE=infinity'                 >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'TimeoutStartSec=0'                  >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'StandardOutput=null'                >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'Restart=always'                     >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'StartLimitBurst=3'                  >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'StartLimitInterval=60s'             >> ${taosd_service_config}"
+    ${csudo} bash -c "echo                                      >> ${taosd_service_config}"
+    ${csudo} bash -c "echo '[Install]'                          >> ${taosd_service_config}"
+    ${csudo} bash -c "echo 'WantedBy=multi-user.target'         >> ${taosd_service_config}"
+    ${csudo} systemctl enable taosd
 }
 
 function install_service() {
-    if is_using_systemd; then
+    if ((${service_mod}==0)); then
         install_service_on_systemd
-    else
+    elif ((${service_mod}==1)); then
         install_service_on_sysvinit
+    else
+        # must manual start taosd
+        kill_taosd
     fi
 }
 
@@ -263,10 +274,12 @@ function update_TDengine() {
     echo -e "${GREEN}Start to update TDEngine...${NC}"
     # Stop the service if running
     if pidof taosd &> /dev/null; then
-        if is_using_systemd; then
-            sudo systemctl stop taosd || :
+        if ((${service_mod}==0)); then
+            ${csudo} systemctl stop taosd || :
+        elif ((${service_mod}==1)); then
+            ${csudo} service taosd stop || :
         else
-            sudo service taosd stop || :
+            kill_taosd
         fi
         sleep 1
     fi
@@ -287,11 +300,13 @@ function update_TDengine() {
         echo -e "\033[44;32;1mTDengine is updated successfully!${NC}"
         echo
         echo -e "${GREEN_DARK}To configure TDengine ${NC}: edit /etc/taos/taos.cfg"
-        if is_using_systemd; then
-            echo -e "${GREEN_DARK}To start TDengine     ${NC}: sudo systemctl start taosd${NC}"
+        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}: sudo update-rc.d taosd default  ${RED} for the first time${NC}"
-            echo -e "                      : sudo service taosd start ${RED} after${NC}"
+            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
         fi
 
         echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"
@@ -341,11 +356,13 @@ function install_TDengine() {
         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 is_using_systemd; then
-            echo -e "${GREEN_DARK}To start TDengine     ${NC}: sudo systemctl start taosd${NC}"
+        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}: sudo update-rc.d taosd default  ${RED} for the first time${NC}"
-            echo -e "                      : sudo service taosd start ${RED} after${NC}"
+            echo -e "${GREEN_DARK}To start TDengine     ${NC}: ./taosd${NC}"
         fi
 
         echo -e "${GREEN_DARK}To access TDengine    ${NC}: use ${GREEN_UNDERLINE}taos${NC} in shell${NC}"