[[384253]]

本文轉(zhuǎn)載自微信公眾號「AirPython」，作者星安果。轉(zhuǎn)載本文請聯(lián)系A(chǔ)irPython公眾號。   

1. 前言

大家好，我是安果!

在 Web 開發(fā)中，經(jīng)常會遇到「端口被占用」的場景

常規(guī)解決方案是：

• 使用 lsof -i 命令查詢占用端口的進(jìn)程 PID
• 利用 kill -9 PID 干掉目標(biāo)進(jìn)程

雖然只有 2 步，我也覺得很繁瑣，即：既要記住命令，又要輸入兩次

本篇文章將編寫 Shell 腳本，來實(shí)現(xiàn)一鍵干掉端口占用

2. 實(shí)現(xiàn)

相比 Python，Shell 腳本常用于處理偏操作系統(tǒng)底層的業(yè)務(wù)，簡單、開發(fā)效率高

以 Mac OSX 為例，在本地創(chuàng)建一個 .sh 文件，實(shí)現(xiàn)步驟如下：

2-1 定義端口號及過濾內(nèi)容參數(shù)

將要處理的端口號及命令行過濾內(nèi)容參數(shù)化，然后在 Shell 腳本中賦值給兩個變量

• $1

執(zhí)行腳本的第一個參數(shù)，即：端口號

• $2

執(zhí)行腳本的第二個參數(shù)，即：過濾進(jìn)程命令內(nèi)容

# 要干掉的端口號 
port_be_kill=$1 
 
# 過濾內(nèi)容 
filter_content=$2 

2-2 獲取端口占用進(jìn)程返回值

使用端口號組成 lsof -i 命令，執(zhí)行這條命令，將返回值賦值給變量 ip_status

# 獲取Shell返回值 
ip_cmd='lsof -i tcp:'$port_be_kill 
 
echo "獲取端口號對應(yīng)的進(jìn)程命令："$ip_cmd 
echo "過濾命令字符串為:"$filter_content 
 
# 執(zhí)行命令 
ip_status=`$ip_cmd` 

注意：ip_status 數(shù)據(jù)類型為字符串

2-3 轉(zhuǎn)為數(shù)組

由于字符串不便于 PID 過濾，我們需要將上面的 ip_status 變量按「空格」分割成一個數(shù)組

# 以空格來分隔，轉(zhuǎn)為一個數(shù)組變量 
array=(${ip_status// / }) 

2-4 遍歷數(shù)組，過濾 PID

首先，遍歷上面的數(shù)組，提取每一個元素

然后，過濾出所有類型為 number、并且上一個值包含過濾內(nèi)容的數(shù)據(jù)

# 判斷數(shù)據(jù)的類型 
function check(){ 
    local a="$1" 
    printf "%d" "$a" &>/dev/null && echo "integer" && return 
    printf "%d" "$(echo $a|sed 's/^[+-]\?0\+//')" &>/dev/null && echo "integer" && return 
    printf "%f" "$a" &>/dev/null && echo "number" && return 
    [ ${#a} -eq 1 ] && echo "char" && return 
    echo "string" 
} 

最后，使用 kill -9 PID 命令處理對應(yīng)的進(jìn)程

# 遍歷數(shù)組 
for i in "${!array[@]}"; do 
    # 注意：賦值等號=前后不能有空格 
    item="${array[i]}" 
 
    # 注意：過濾十六進(jìn)制字符串 
    # 先轉(zhuǎn)為字符串，然后判斷是否以0x開頭 
    # echo $item 
    if [[ $item != 0x* ]] 
    then 
        # 非十六進(jìn)制數(shù)據(jù)，即：PID 
        if [ $(check $item) = "integer" ] 
        then 
            # 判斷上一個元素是否包含關(guān)鍵字 
            # 命令行是否包含關(guān)鍵字 
            item_pre="${array[i-1]}" 
            # echo $item_pre 
            # echo $filter_content 
            if [[ $item_pre =~ $filter_content ]] 
            then 
                # echo $item 
                # 調(diào)用kill-9 pid命令干掉進(jìn)程 
                kill_cmd="kill -9 "$item 
                echo $kill_cmd 
                # 執(zhí)行命令，干掉進(jìn)程 
                $kill_cmd 
            fi 
        fi 
    fi   
done 

2-5 設(shè)置 Alias

為了一鍵運(yùn)行 Shell 腳本，我們使用 Alias 給命令設(shè)置一個別名

修改「.bash_profile」文件，將 Shell 腳本文件的完整路徑及執(zhí)行命令寫入到一個自定義的函數(shù)中

# vim .bash_profile 
# alias定義 
kill_port() { 
  cd /Users/xingag/Desktop/work 
  ./kill_port_with_args.sh $1 $2 
} 
alias kp=kill_port 

2-6 實(shí)戰(zhàn)一下

使用 source .bash_profile 命令刷新配置文件，讓 Alias 配置立即生效

假如現(xiàn)在 8000 被占用，我們只需要打開終端輸入「kp 8000 python」命令即可以快速干掉目標(biāo)進(jìn)程

運(yùn)行截圖如下：

4. 最后

需要指出的是，Linux 下需要預(yù)先安裝 lsof 命令，以 CentOS 為例

# Centos安裝lsof 
yum install lsof 

如果是 Windows，處理端口占用的 Shell 腳本不一樣;它需要使用 netstat/tasklist/taskkill 命令去改寫

另外，PC 端執(zhí)行 Shell 腳本建議使用 Git Bash

# Win處理端口占用 
# 1、打開cmd終端 
cmd 
 
# 2、查找端口占用的進(jìn)程及PID 
netstat -aon|findstr PORT 
 
# 3、根據(jù)PID查詢進(jìn)程名稱 
tasklist|findstr PID 
 
# 4、使用taskkill命令或在任務(wù)管理器中關(guān)掉進(jìn)程