本記事は、アドベントカレンダー Go4 の1日目の記事です。
これは何?
みなさんは、どんなdaemon処理を書いていますか？
ここでは、Go の daemon 処理サンプルを紹介します.
daemon とは、バックグラウンドで動作するプロセス実行し続けるアレで、typoに注意なやつです(いつもdeamonと書いてしまう).
Goではその言語特性から、様々な処理をシンプルに記述しやすいと感じており、daemonの実装例を通してGoらしさを学ぶことにも役立つと思います.
他にも役立つレシピがあれば、教えて下さい 
まずは基本から
単純に無限実行させる
The Go Playground で実行してみる
func main() {
timeout := 5 * time.Second
ctx, cancel := context.WithTimeout(context.Background(), timeout)
defer cancel()
f := func() error { // some task
return nil
}
simple(ctx, f)
}
func simple(ctx context.Context, task func() error) {
for {
err := task()
if err != nil {
log.Printf("[ERROR] err: %v", err)
}
time.Sleep(500 * time.Millisecond)
}
}
context を使ったタイムアウト
The Go Playgroundで実行してみる
func withTimeout(ctx context.Context, task func() error) {
child, childCancel := context.WithCancel(ctx)
defer childCancel()
for {
err := task()
if err != nil {
log.Printf("[ERROR] err: %v", err)
}
select {
case <-child.Done():
log.Printf("[DEBUG] timeout")
return
default:
time.Sleep(500 * time.Millisecond)
}
}
}
応用編
特定時刻にだけ処理をする
The Go Playground で実行してみる
func runDaemon(ctx context.Context, f func(context.Context) error) {
daemonHour := "7-10"
waitDuration := 500 * time.Millisecond
for {
now := time.Now()
if len(daemonHour) != 0 { //起動時刻の指定があったら
isExec := isExecHour(now, daemonHour)
log.Printf("[DEBUG] daemon起動時間かどうかの判定 now:%v, daemonHour:%s, isExec:%v", now, daemonHour, isExec)
if !isExec {
time.Sleep(1 * time.Minute)
continue
}
}
err := f(ctx)
if err != nil {
log.Printf("[ERRROR] err:%v", err)
}
time.Sleep(waitDuration)
}
}
func isExecHour(now time.Time, dHour string) bool {
delimitor := "-"
dh := strings.Split(dHour, delimitor)
if len(dh) <= 1 {
return false
}
start, err := strconv.Atoi(dh[0])
if err != nil {
return false
}
end, err := strconv.Atoi(dh[1])
if err != nil {
return false
}
h := now.Hour()
if start <= h && h <= end {
return true
}
return false
}
一定周期ごとに(前の処理が終わっていなくても)処理を実行する
The Go Playground で実行してみる
func timeTicker(ctx context.Context, task func(context.Context) error) {
counter := 0
waitTime := 1 * time.Second
ticker := time.NewTicker(waitTime)
defer ticker.Stop()
child, childCancel := context.WithCancel(ctx)
defer childCancel()
for { // deamon化するため無限実行
select {
case t := <-ticker.C:
counter++
requestID := counter
log.Println("[DEBUG] START taskNo=", requestID, "t=", t)
errCh := make(chan error, 1)
go func() { // 登録したタスクをブロックせずに実行
errCh <- task(ctx)
}()
go func() {
// error channelにリクエストの結果が返ってくるのを待つ
select {
case err := <-errCh:
if err != nil {
// Deamonの強制終了
log.Println("[ERROR] ", err)
}
log.Println("[DEBUG] END requestNo=", requestID)
}
}()
case <-child.Done():
return
}
}
}
みなさんも、よき daemon Life をお過ごしください!