Apache/Nginx 自動再起動付き監視デーモンを golang で作ろう

はじめに

モダンな監視ツールを導入できない環境向けに簡単な監視デーモンを golang で簡単に実装してみます。
クロスコンパイルでほとんどの環境で動作するデーモンを作れるのでこういう時に golang は本当に頼りになります。

やってみよう

必用なもの

• golang

仕様

監視対象URLにポーリング → エラー回数が上限を超え → 任意のコマンドを発行
上記のデーモンを golang で実装する
監視対象URLやエラー時に発行するコマンドなどを環境変数にから読み込む

実装

まずは必要な変数を定義します。

main.go

var (
	CheckUrl         string // 監視対象URL
	CheckTimeout     int64  // 監視のタイムアウト
	CheckInterval    int64  // 監視間隔（秒）
	RetryCount       int64  // リトライ最大回数
	StopCmd          string // エラー検知時のApache/Nginx停止コマンド
	StartCmd         string // Apache/Nginx起動コマンド
	WaitAfterStop    int64  // StopCmd実行後のWait（秒）
	WaitAfterRestart int64  // リスタート後のWait（秒）
	ErrorCounter     int64  // エラーカウンター
)

次に http.DefaultClient を使用して指定のURLにリクエストする func を作成します。

main.go

type Check struct {
	client *http.Client
}

func (c *Check) request(ctx context.Context, req *http.Request) (*http.Response, error) {
	req = req.WithContext(ctx)
	resCh := make(chan *http.Response)
	errCh := make(chan error)
	go func() {
		res, err := c.client.Do(req)
		if err != nil {
			errCh <- err
			return
		}
		resCh <- res
	}()
	select {
	case res := <-resCh:
		return res, nil
	case err := <-errCh:
		return nil, err
	case <-ctx.Done():
		return nil, errors.New("HTTP request cancelled")
	}
}

指定URLにリクエストして、エラーの場合はApache/Nginxを再起動する func を作成します。

main.go

func check() {
	c := Check{
		client: http.DefaultClient, // DefaultClient を使い回し
	}
	req, err := http.NewRequest(http.MethodGet, CheckUrl, nil)
	if err != nil {
		log.Fatal(err)
		return
	}
	defer func() {
		if ErrorCounter < RetryCount {
			return
		}
		// Apache/Nginxリスタート
		err := c.restart()
		if err != nil {
			log.Fatal(err)
		}
		ErrorCounter = 0
		time.Sleep(time.Second * time.Duration(WaitAfterRestart))
	}()
	res, err := c.request(context.Background(), req)
	if err != nil {
		log.Println(err)
		ErrorCounter++
		return
	}
	// ステータスコードが400番台以降ならエラーと判定してエラーカウンタをインクリメント
	if res.StatusCode >= 400 {
		log.Println(fmt.Sprintf("bad response status code %d", res.StatusCode))
		ErrorCounter++
		return
	}
	log.Println(fmt.Sprintf("%s %d", CheckUrl, res.StatusCode))
	ErrorCounter = 0
}

最後にStopCmdとStartCmdの実行 func を作成します。

main.go

func (c *Check) restart() error {
	var cmd []string
	var err error
	log.Println(`restart...`)
	cmd = strings.Fields(StopCmd)
	stop := exec.Command(cmd[0], cmd[1:]...)
	log.Println(fmt.Sprintf(`exec stop command. [%#v]`, cmd))
	err = stop.Run()
	if err != nil {
		return err
	}
	time.Sleep(time.Second * time.Duration(WaitAfterStop))
	cmd = strings.Fields(StartCmd)
	start := exec.Command(cmd[0], cmd[1:]...)
	log.Println(fmt.Sprintf(`exec start command. [%#v]`, cmd))
	err = start.Run()
	if err != nil {
		return err
	}
	log.Println(`done...`)
	return nil
}

func main() {
	t := time.NewTicker(time.Second * time.Duration(CheckInterval))
	for {
		select {
		case <-t.C:
			check()
		}
	}
}

今回のコード一式は以下にあります。
やっぱツール作るなら golang がお気に入りです。
https://github.com/noridas80/watch-localhost