提交 11e3af22 编写于 作者: P plum-lihui

[Modify the use of sudo commands]

上级 e06f2444
......@@ -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}"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册