go 运维,检测 web 服务状态

来源:互联网 时间:1970-01-01

延续前面的代码,在运维的时候经常需要检测 web 服务的状态,可以在 web 中专门设置一个测试页面(可以查询下数据库,返回一个指定的值)

package main//使用的包import ( "flag" "fmt" "io/ioutil" "log" "net" "net/http" "regexp")//定义脚本的版本const VERSION = "0.1.0"//使用 flag 来定义命令输入的参数var Input_protocol = flag.String("p", "tcp", "The protocol you want to check")//定义检测 tcp 服务的脚本,用到 net 包func tcp(url string) int { _, err := net.Dial("tcp", url) if err != nil { fmt.Println(err) return 0 } else { return 1 }}//定义检测 udp 服务的脚本,用到 net 包func udp(url string) int { _, err := net.Dial("udp", url) if err != nil { fmt.Println(err) return 0 } else { return 1 }}//定义检测 http 服务的脚本,成功返回1,失败返回0func chttp(url, checkword string) int { res, err := http.Get(url) if err != nil { //如果连接失败返回错误 log.Fatal(err) return 0 } //使用 ioutil 读取得到的响应 robots, err := ioutil.ReadAll(res.Body) //关闭资源 res.Body.Close() //失败返回原因 if err != nil { log.Fatal(err) return 0 } //调用 regexp 函数查找 checkword word, err := regexp.MatchString(checkword, string(robots)) if err != nil { log.Println(err) return 0 } if word { log.Printf("The `%s` find in `%s`", checkword, url) return 1 } log.Printf("The `%s` not find in `%s`", checkword, url) return 0}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 或者 check -p http http://dockerpool.com dockerpool" 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) case p == "http": chttp(flag.Args()[0], flag.Args()[1]) }}

使用示例:

$ check -p http http://dockerpool.com dockerpool2015/01/08 10:38:53 The `dockerpool` find in `http://dockerpool.com`$ check -p tcp www.sina.com:801$ check -p tcp www.sina.com:88dial tcp 61.172.201.20:88: connection refused0

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关阅读:
Top