Golang检测Linux服务器端口占用

Posted 2022-09-27 gaiheilukamei

• 代码实现

func CheckPort(port int) error 
    checkStatement := fmt.Sprintf(`netstat -anp | grep -q %d ; echo $?`, port)
    output, err := exec.Command("sh", "-c", checkStatement).CombinedOutput()
    if err != nil 
        return err
    
    // log.Println(output, string(output)) ==> [48 10] 0 或 [49 10] 1
    result, err := strconv.Atoi(strings.TrimSuffix(string(output), "\n"))
    if err != nil 
        return err
    
    if result == 0 
        return fmt.Errorf("端口%d已被占用", port)
    

    return nil

• 纠错思路

  • 第8行原代码为：result, err := strconv.Atoi(string(output)),
    报错为: strconv.Atoi: parsing "0\n": invalid syntax
  • 经排查，CombinedOutput()函数的执行结果0\n(1\n)包含换行符\n,
    利用strings.TrimSuffix()函数删除换行符\n

• 延伸知识

  • strings.TrimSuffix()和strings.TrimRight()的区别：

    • TrimRight(s string, cutset string) string指从s右端开始，删除所有cutset中出现的字符，若从右到左的过程中遇到其他字符，则停止删除。

    • TrimSuffix(s, suffix string) string删除后缀suffix

    • eg：strings.TrimRight("abcdabcdaaaaeac", "abcd")返回"abcdabcdaaaae"，
      而strings.TrimRight("abcdaaaaa", "abcd")返回""