K8S 开发环境操作工具

ゝ一纸荒年。 2022-11-29 11:52 167阅读 0赞

#! /bin/bash
    
    NAMESPACE=dev
    
    function undeployAll()
    { 
    	undeploy svc $1
    	undeploy hpa $1
    	undeploy deploy $1
    	undeploy ingress $1
    	echo "卸载完成"
    }
    
    function undeploy()
    { 
      echo "卸载$1..."
      CNT=`kubectl get $1 -n ${ NAMESPACE} |grep "^$2"|wc -l`
      if [ $CNT -gt 0 ]; then
        kubectl get $1 -n ${NAMESPACE} |grep "^$2"|cut -d ' ' -f 1|xargs kubectl delete $1 -n ${NAMESPACE} 
      fi
    }
    
    
    function readLog(){ 
            SERVICENAME=$1
            CNT=`kubectl get pod -n ${ NAMESPACE} |grep "^$SERVICENAME"| grep 'Running'|wc -l`
            if [ $CNT -gt 1 ]; then
              echo "该服务名称前缀对应多个pod，无法定位"
              return        
            fi
            if [ $CNT -gt 0 ]; then
                podName=$(kubectl get pod -n ${ NAMESPACE} | grep "^$SERVICENAME"| grep 'Running' |cut -d ' ' -f 1)
                echo "打印pod:"$podName
                kubectl  -n ${NAMESPACE}  exec  $podName  -- sh -c 'cd /data/applog/slp/$CONTAINER_NAME && ls -lt| grep "\.log"| grep -v 'zipkin'| grep -v 'brave'| grep -v 'docker'| grep -v 'metrics'|grep -v 'WARN'| grep -v 'gz' | sed -n '1p' | sed "s/ */ /g" | cut -d " " -f 9 | xargs tail -200'
                return
            fi
    }
    
    #更新POD为host network
    function updateToHostNetwork()
    { 
            SERVICENAME=$1
            CNT=`kubectl get deploy -n ${ NAMESPACE} |grep "^$SERVICENAME"|wc -l`
            if [ $CNT -gt 0 ]; then
                    deployments=$(kubectl get deploy -n ${ NAMESPACE} | grep "^$SERVICENAME"|cut -d ' ' -f 1)
                    for element in ${deployments[@]}
                    #也可以写成for element in ${array[*]}
                    do
                      hostNetwork=$(kubectl -n ${ NAMESPACE} get deploy $element -o yaml | grep 'hostNetwork: true')
                      if  [ ! -n "$hostNetwork" ] ;then
                        echo "*******************开始更新$element的网络***********************"
                        kubectl -n ${NAMESPACE}   patch deploy $element -p '{"spec":{"template":{"spec":{"hostNetwork":true}}}}'
                      else
                        echo "$element已是宿主机网络"
                      fi
                    done
            fi
            
    }
    
    #环境瘦身
    function Slimming()
    { 
                    deployments=$(kubectl get deploy -n ${ NAMESPACE} | grep -v 'NAME' | cut -d ' ' -f 1)
                    for element in ${deployments[@]}
                    do
                      containerName=$(kubectl -n ${ NAMESPACE} get deploy $element -owide | grep -v 'NAME' | awk '{print $7}')
                      #${containerName}在json中是字符串类型，json串中必须使用双引号应用
                      #不能直接包含在双引号中，否者shell不会替换为参数值
                      #通过前后字符串来拼接，注意不要前后字符串都别缺少双引号
                      kubectl -n ${NAMESPACE}  patch deploy ${element} -p '{"spec":{"template":{"spec":{"containers":[{"name": "'${containerName}'","resources":{"limits":{"cpu":"1","memory":"1Gi"},"requests":{"cpu":"1","memory":"1Gi"}}}]}}}}'
                      kubectl -n ${NAMESPACE}  patch hpa  ${element}  -p '{"spec":{"minReplicas":1,"maxReplicas":1}}'
                    done
    
    }
    
    # 转移POD到目标NODE
    # 参数1：POD名前缀
    # 参数2：目标NODE名
    function movePod(){ 
       
       SERVICENAME=$1
       NODEHOSTNAME=$2
       CNT=`kubectl get deploy -n ${ NAMESPACE} |grep "^$SERVICENAME"|wc -l`
       if [ $CNT -lt 0 ]; then echo "目标服务不存在";return 1;fi
       DEPLOYNAME=$(kubectl get deploy -n ${ NAMESPACE} | grep -v 'NAME'| grep $SERVICENAME | cut -d ' ' -f 1)
       #给目标node打上标签
       kubectl patch node $NODEHOSTNAME -p '{"metadata":{"labels":{"gw":"yes"}}}'
       kubectl patch node $NODEHOSTNAME -p '{"metadata":{"labels":{"slpcore":"yes"}}}'
       kubectl -n ${NAMESPACE}  patch deploy $DEPLOYNAME -p '{"spec":{"template":{"spec":{"nodeSelector":{"kubernetes.io/hostname":"'${NODEHOSTNAME}'"}}}}}'
    
    }
    
    while :
    do    
    	clear
    	echo "========================================================================"
    	echo "This tool provides batch uninstallation of service deployment."
    	echo "========================================================================"
    	echo ""
            echo "1. 批量卸载应用"
            echo "2. 更新应用网络为主机网络"
            echo "3. 查看日志"
            echo "4. 扩容缩容：修改副本数"
            echo "6. 容器环境瘦身"
            echo "7. 转移POD"
            echo "8. 禁用NODE:上面的POD自动迁移到其他NODE"
            echo "9. 恢复NODE:可重新被使用"
       	echo "999. 退出"
    	echo ""
            read -e -p "请选择: " OPTION
            case $OPTION in 
    		1) 	echo "进行批量卸载的应用需要输入服务前缀"
    			echo "例如：aim 或者 psc"
    			read -e -p "请输入服务前缀: " SERVICENAME
                            if [ -z "$SERVICENAME" ]; then
                                echo "服务名为空"
                            else
                                undeployAll $SERVICENAME
                            fi 
    			            ;;
                    2)      echo "更新应用网络为主机网络需要输入服务前缀"
                            echo "例如：aim"
                            read -e -p "请输入服务前缀: " SERVICENAME
                            updateToHostNetwork $SERVICENAME
                            ;;
                    3)      echo "查看服务日志需要输入服务前缀"
                            echo "例如：customer-gateway"
                            read -e -p "请输入服务前缀: " SERVICENAME
                            readLog $SERVICENAME
                            ;;
                    4)      read -e -p "请输入目标服务名称前缀: " SERVICENAME
                            read -e -p "请输入目标副本数: " REPLICAS
                            if [ -z $SERVICENAME ]; then echo '目标服务名称前缀为空' continue; fi
                            if [ -z $REPLICAS ]; then echo '目标副本数为空' continue ; fi
                            kubectl -n ${NAMESPACE}  get hpa | grep $SERVICENAME| awk { 'print $1'} | xargs kubectl -n ${NAMESPACE}  patch hpa -p '{"spec": {"maxReplicas": '${REPLICAS}', "minReplicas": '${REPLICAS}'}}'
                            echo "如果看到horizontalpodautoscaler **** patched 则执行成功，请等待30s"
                            ;;
                    6)      echo "确定缩容么？"
                            Slimming
                            ;;
                    7)      read -e -p "请输入服务前缀: " SERVICENAME
                            read -e -p "请输入目标NODE的HostName，比如 k8snode11: " NODEHOSTNAME
                            movePod $SERVICENAME $NODEHOSTNAME
                            ;;
                    8)      read -e -p "请输入目标NODE的HostName，比如 k8snode11: " NODEHOSTNAME
                            nohup kubectl drain  ${NODEHOSTNAME} --ignore-daemonsets=true --delete-local-data=true --force=true > std.log 2>&1 &
                            ;;
                    9)      read -e -p "请输入目标NODE的HostName，比如 k8snode11: " NODEHOSTNAME
                            nohup kubectl uncordon  ${NODEHOSTNAME}  > std.log  2>&1  &
                            ;;
    				999)	echo "Bye, see you next time"
    						exit 0
    						;;
           	        *) 	echo "请输入正确选项!!"
    			;;
            esac
    	echo ""
    	read -p "press any key to continue..." -n 1 anyKey            
    done