go營運，檢測web服務狀態

go營運，檢測web服務狀態

延續前面的代碼，在營運的時候經常需要檢測 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