- 代碼實現
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
2019-09-24改
func CheckPort(port int) error {
checkStatement := fmt.Sprintf("netstat -anp | grep -q %d ", port)
output, err := exec.Command("sh", "-c", checkStatement).CombinedOutput()
if err != nil {
return nil // err != nil 說明端口沒被占用
}
if len(output) > 0 {
return fmt.Errorf("端口%d已被占用", port)
}
return nil
}
注：以上方法的缺陷是目標操作系統上有可能沒有netstat命令
2019-09-29改
func CheckPort(port string) error {
var err error
tcpAddress, err := net.ResolveTCPAddr("tcp4", ":" + port)
if err != nil {
return err
}
for i := 0; i < 3; i++ {
listener, err := net.ListenTCP("tcp", tcpAddress)
if err != nil {
time.Sleep(time.Duration(100) * time.Millisecond)
if i == 3 {
return err
}
continue
} else {
listener.Close()
break
}
}
return nil
}