Goのバッチで統計を取得するAPIを用意しておくと便利

golangで以下のようなKillするまでずっと動くバッチがあるとします。

package main

import (
    "fmt"
    "sync"
    "time"
)

// なんかしらの処理
func someprocess() {
    somedata := []string{"a", "b", "c", "d", "e"}

    var wg sync.WaitGroup
    for _, s := range somedata {
        wg.Add(1)
        go func(val string) {
            defer wg.Done()
            fmt.Println(val)
        }(s)
    }
}

func main() {
    // ずっと動き続ける
    for {
        someprocess()
        time.Sleep(3 * time.Second)
    }
}

このとき、この処理の稼働状況を知りたくなるのが人の常なので、以下のようにhttpサーバを立てて、
jsonで統計情報を取得できるようにしておくと便利です。
（統計情報の書き込みでロックが発生しますが、処理量にもよりますがほとんど問題にならないはず。）

package main

import (
    "encoding/json"
    "fmt"
    "log"
    "net/http"
    "os"
    "sync"
    "time"
)

var metrics *Metrics

type Metrics struct {
    sync.RWMutex
    PID          int
    StartedAt    time.Time
    LastWorkedAt time.Time
    LoopCount    int
    DataCount    int
}

func someprocess() {
    somedata := []string{"a", "b", "c", "d", "e"}

    var wg sync.WaitGroup
    for _, s := range somedata {
        wg.Add(1)
        go func(val string) {
            defer wg.Done()
            fmt.Println(val)

            // 詳細な処理の統計を更新
            metrics.Lock()
            defer metrics.Unlock()
            metrics.DataCount++
        }(s)
    }
}

func init() {
    metrics = &Metrics{
        PID:          os.Getpid(),
        StartedAt:    time.Now(),
        LastWorkedAt: time.Now(),
        DataCount:    0,
    }
}

func MetricsHandler(w http.ResponseWriter, r *http.Request) {
    metrics.RLock()
    defer metrics.RUnlock()
    b, err := json.Marshal(metrics)
    if err != nil {
        http.Error(w, http.StatusText(500), 500)
    }

    fmt.Fprintf(w, string(b))
}

func main() {
    go func() {
        for {
            someprocess()

            // 1ループごとの統計を更新
            metrics.Lock()
            metrics.LoopCount++
            metrics.LastWorkedAt = time.Now()
            metrics.Unlock()
            time.Sleep(3 * time.Second)
        }
    }()

    // 実行状況を表示するHTTPサーバを立てる。
    port := 9001
    http.HandleFunc("/metrics", MetricsHandler)
    fmt.Printf("Listening on %d...\n", port)
    log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", port), nil))
}

起動後、http://localhost:9001/metricsにアクセスすると、以下のようなjsonが返ってきます。

{
  "PID": 85568,
  "StartedAt": "2015-08-31T20:41:19.326943122+09:00",
  "LastWorkedAt": "2015-08-31T20:41:43.350825325+09:00",
  "LoopCount": 9,
  "DataCount": 45
}

これを監視するとかしておくとよいのではないでしょうか。