使用 go 写一个检测 tcp udp 状态的包

因为要跟着 dockerpool 的基友写 Docker 管理的程序，所以今年的目标是学习 go 语言。

看完 go 的入门教程，高深的代码还看不懂，所以决定先尝试写一些平常可以用的东西，mac 底下的 nc 命令可以检测很多东西，但是用着不习惯，所以决定写一个 check tcp udp 的包。

package main

import (
    "flag"
    "fmt"
    "net"
)

const version = "0.1.0"

var Input_protocol = flag.String("p", "tcp", "The protocol you want to check")

func tcp(url string) int {
    _, err := net.Dial("tcp", url)
    if err != nil {
        fmt.Println(err)
        return 0
    } else {
        return 1
    }
}

func udp(url string) int {
    _, err := net.Dial("udp", url)
    if err != nil {
        fmt.Println(err)
        return 0
    } else {
        return 1
    }
}

func main() {
    flag.Parse()
    if flag.NArg() < 1 {
        useage := "使用示例: check -p tcp 192.168.7.26:22 或者 check -p udp 192.168.7.23:123 "
        fmt.Println(useage)
        return
    }
    p := *Input_protocol
    switch {
    case p == "tcp":
        r := tcp(flag.Args()[0])
        fmt.Println(r)
    case p == "udp":
        r := udp(flag.Args()[0])
        fmt.Println(r)
    }
}

go build go install 之后，就可以再命令行中使用了。

$ go build
$ go install
$ check
使用示例: check -p tcp 192.168.7.26:22 或者 check -p udp 192.168.7.23:123
$ check -p tcp www.sina.com:80
1
$ check -p tcp www.sina.com:88
dial tcp 61.172.201.20:88: connection refused
0