时间:2023-01-20 09:34:26 | 栏目:Golang | 点击:次
在运维开发的过程中,经常碰到这样的情况:启动某个应用前,需要先检测一下端口是否被其他的应用占用了,若占用了,能否得到占用进程的PID。后续可以根据这个PID,查找是哪个应用占据这端口,然后KILL掉。非常简单的需求,思路是:
利用 netstat 命令获取当前的相关端口号的PID,然后正则表达式过滤出相关的PID即可
netstat -ano | findstr 8099
输出如下:
以windows 系统下为例,linux系统下,修改相关参数即可。代码如下:
// 传入查询的端口号 // 返回端口号对应的进程PID,若没有找到相关进程,返回-1 func portInUse(portNumber int) int { res := -1 var outBytes bytes.Buffer cmdStr := fmt.Sprintf("netstat -ano -p tcp | findstr %d", portNumber) cmd := exec.Command("cmd", "/c", cmdStr) cmd.Stdout = &outBytes cmd.Run() resStr := outBytes.String() r := regexp.MustCompile(`\s\d+\s`).FindAllString(resStr, -1) if len(r) > 0 { pid, err := strconv.Atoi(strings.TrimSpace(r[0])) if err != nil { res = -1 } else { res = pid } } return res }
对端口进行扫描基于连接,在go中我们可以使用net.Dial进行判断,如果返回的结果没有错误,那么这个端口就应该是打开的,如果返回了错误,那么可以说明这个端口是关闭的,代码如下
func ScanPort(protocol string, hostname string, port int) bool { fmt.Printf("scanning port %d \n", port) p := strconv.Itoa(port) addr := net.JoinHostPort(hostname, p) conn, err := net.DialTimeout(protocol, addr, 3*time.Second) if err != nil { return false } defer conn.Close() return true }