捕獲Ctrl + C中斷 優雅的退出程序 golang

捕獲Ctrl + C中斷 優雅的退出程序

Gracefully terminate a program in Go

• os/signal 來捕獲系統中斷等信號

// Notify方法將signal發送到channel，
func Notify(c chan<- os.Signal, sig ...os.Signal)
// 初始化一個接受os.Signal的通道
c := make(chan os.Signal)
// 調用Notify方法，綁定signal到channel，一旦有信號到達，signal會發送到channel中
signal.Notify(c, os.Interrupt)

1. 代碼實如今命令行每隔200ms打印字符"#"，使用time.Ticker實現每隔2s打印完整一次
   
2. task有一個close的channel，用來接受中止信號
3. 實例化一個os.Signal來接受諸如ctrl+c的中止信號
4. 使用sync.WaitGroup來組織多個goroutine的調度

package main

import (
    "fmt"
    "os"
    "os/signal"
    "sync"
    "time"
)

type Task struct {
    closed chan struct{}
    wg     sync.WaitGroup
    ticker *time.Ticker
}

func (t *Task) Run() {
    for {
        select {
        case <-t.closed:
            return
        case <-t.ticker.C:
            handle()
        }
    }
}

func (t *Task) Stop() {
    close(t.closed)
}

func handle() {
    for i := 0; i < 5; i++ {
        fmt.Print("#")
        time.Sleep(time.Millisecond * 200)
    }
    fmt.Println()
}

func main() {
    task := &Task{
        closed: make(chan struct{}),
        ticker: time.NewTicker(time.Second * 2),
    }

    c := make(chan os.Signal)
    signal.Notify(c, os.Interrupt)

    task.wg.Add(1)
    go func() { defer task.wg.Done(); task.Run() }()

    select {
    case sig := <-c:
        fmt.Printf("Got %s signal. Aborting...\n", sig)
        task.Stop()
    }
  task.wg.Wait()
}