golangでhttpを監視するscriptを書いて結果をslackにpostする

golangで簡単なscriptを書いてみる

ぐぐると、簡単な1ファイルのサンプルスクリプトは出て来るが、もうちょっと拡張性のある、ライブラリ置き場や設定ファイル置き場を分けた様な構成でscriptを書いてみる事にした。

という事で、今回はgolangを使ってhttp監視をしてみる事にする。

ディレクトリ構成

GOPATH以下の構成。基本的にGOPATH/src以下にプロジェクトを作っていく事が推奨らしい。
src以下にprojectというディレクトリを作りその下に各ディレクトリを配置。

src - project - config - secrets.go
              |        |
              |        - targets.go
              |
              - libs - slack - slack.go
              |
              - scripts - check_http.go

config : 設定ファイル置き場。secrets.goにはgit管理したくない物を入れて.gitignoreしておくと吉。

libs : slackにpostする周りのライブラリ置き場

scripts : mainの監視用script

ソースコード

各ファイルの名前と中身、ちょっとしたコメント。

config/secrets.go

slackのincomingのurlを書いておく。
人には知られるとまずい物なので、その他のファイルとは独立させておく。

package config

var (
    IncomingUrl string = "https://hooks.slack.com/services/XXXXXXXXXXXXXXXXXXXXXX"
)

config/targets.go

テスト用に、3つドメインを用意してみた。最初の一つだけ成功して後は失敗するテスト。

package config

import (
)

type HttpConfigSlice struct {
    Http []HttpConfig
}

type HttpConfig struct {
    Name string
    Host string
    Path string
    Proto string
    Domain string
}

func HttpTargets() []HttpConfig {
    return []HttpConfig{
        HttpConfig{Name: "Google", Host: "google.com", Path: "/", Proto: "http", Domain: "google.com"},
        HttpConfig{Name: "Google", Host: "google.com", Path: "/no-such-path", Proto: "http", Domain: "google.com"},
        HttpConfig{Name: "NiseGoogle", Host: "nisegoogle.com", Path: "/", Proto: "http", Domain: "nisegoogle.com"},
    }
}

libs/slack/slack.go

引数で貰ったmsgをそのままslackにPOSTするやつ。実はslackのgo用ライブラリがあったっぽいが書いた後に気づいた。
これ。https://github.com/nlopes/slack

書いちゃったので、今回は自分のを使う。

configをimportする時には、GOPATH/src以下からのpathで読める。
pathの前に名前をつけるとエイリアスになって便利。

package slack

import (
    "log"
    "net/http"
    "net/url"
    "strings"
    "encoding/json"
    "time"

    config "project/config"
)

type CheckConfig struct {
    Http []HttpConfig `json:"http"`
}

type HttpConfig struct {
    Name   string `json:"name"`
    Host   string `json:"host"`
    Path   string `json:"path"`
    Proto  string `json:"proto"`
    Domain string `json:"domain"`
}

type Slack struct {
    Text     string `json:"text"`
    Username string `json:"username"`
}

func Post(msg string) {
    params, _ := json.Marshal(Slack{
        msg,
        "MonitoringBot",
    })

    client := &http.Client{Timeout: 5 * time.Second}

    values := url.Values{"payload": {string(params)}}
    req, err := http.NewRequest("POST", config.IncomingUrl, strings.NewReader(values.Encode()))
    if err != nil {
        log.Print(err)
        return
    }

    req.Header.Add("Content-Type", "application/x-www-form-urlencoded")

    resp, err := client.Do(req)
    if err != nil {
        log.Print(err)
        return
    }
    defer resp.Body.Close()
}

scripts/check_http.go

監視のおおもとのやつ。単純に、ターゲット持ってきて、httpチェックをするだけ。地味にリトライ入れてみた。

package main

import (
    "net/http"
    "crypto/tls"
    "time"
    "fmt"

    slack "project/libs/slack"
    config "project/config"
)

func main() {
    for _, target := range config.HttpTargets() {
        check(target)
    }
}

func check(target config.HttpConfig) {
    errorNum := 0
    checkNum := 0
    fatalNum := 2
    errorMessage := ""
    for checkNum < fatalNum {
        checkNum += 1
        targetPath := target.Proto + "://" + target.Host + target.Path
        tr := &http.Transport{
            TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
        }
        client := &http.Client{Timeout: 5 * time.Second, Transport: tr}
        req, err := http.NewRequest("GET", targetPath, nil)
        if err != nil {
            slack.Post(err.Error())
            return
        }

        req.Header.Add("Host", target.Domain)
        resp, err := client.Do(req)
        if err != nil {
            slack.Post(err.Error())
            return
        }

        if resp.StatusCode != 200 {
            errorNum += 1
            if (errorNum >= fatalNum) {
                errorMessage += targetPath + " [" + target.Name + "] " + "returns " + fmt.Sprint(resp.StatusCode) + "\n"
            }
        } else {
            break
        }

        defer resp.Body.Close()
    }

    if errorMessage != "" {
        slack.Post(errorMessage)
    }
}

slack post結果

実行してみる。

$ go run scripts/check_http.go

期待通り、pathがおかしいやつが404、ドメイン変なやつが no such hostになった。

意外に簡単に書ける、go言語。