From 11e3af220bcdc78bec9ec7b449b9300b955e7b8f Mon Sep 17 00:00:00 2001 From: plum-lihui <2849823933@qq.com> Date: Fri, 19 Jul 2019 15:40:03 +0800 Subject: [PATCH] [Modify the use of sudo commands] --- packaging/tools/install.sh | 199 ++++++++++++++++++++----------------- 1 file changed, 108 insertions(+), 91 deletions(-) diff --git a/packaging/tools/install.sh b/packaging/tools/install.sh index 1883493..bda72ed 100755 --- a/packaging/tools/install.sh +++ b/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}" -- GitLab