Golang检测Linux服务器端口占用

• 代码实现

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()的区别

func main() {
    str1 := "abcdabceeebdc"
    str2 := "abcd"
    str3 := "abced"
    str4 := "ebdc"
    // `trimright(s, cutset string) string`指从`s`右端开始，删除所有`cutset`中出现的字符
    // 若从右到左的过程中遇到其他字符，则停止删除。  
    fmt.println(strings.trimright(str1, str2))    
    fmt.println("空行：", strings.trimright(str1, str3))   

    // `trimsuffix(s, suffix string)`删除后缀`suffix` ，若不匹配，返回s
    fmt.println(strings.trimsuffix(str1, str4))    
    fmt.println(strings.trimsuffix(str1, str2))    
}

// 输出结果：
// abcdabceee
// 空行：
// abcdabcee
// abcdabceeebdc