go select使用的一个demo

select是Go中的一个控制结构，类似于用于通信的switch语句。每个case必须是一个通信操作，要么是发送要么是接收。
select随机执行一个可运行的case。如果没有case可运行，它将阻塞，直到有case可运行。一个默认的子句应该总是可运行的。（菜鸟教程）

使用select和chan实现一个启动定时炸弹，然后倒计时过程中可以手动取消的一个小例子

image.png

首先准备要使用到的函数，倒计时，执行爆炸，监听取消动作

package main

import (
    "log"
    "os"
    "strconv"
    "time"
)

//爆炸了
func booooom() {
    writeLog("---------booooom----------")
}

//倒计时
func countdown(litSignal chan int) {
    c := time.Tick(1 * time.Second)
    for countNum := 10; countNum > 0; countNum-- {
        writeLog("---------" + strconv.Itoa(countNum) + "---------")
        <-c
    }
    litSignal <- -1
}

//监听取消动作
func isCancel(cancelSignal chan int) {
    //当有输入信息时，向管道写入数据-1
    var buffer [256]byte
    _, err := os.Stdin.Read(buffer[:])
    if err != nil {
        panic(err)
    }
    writeLog("---------cancel---------")
    cancelSignal <- -1
}

func writeLog(msg string) {
    logFile, err := os.OpenFile("syslog.txt-"+time.Now().Format("2006-01-02"), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766)
    defer logFile.Close()
    if err != nil {
        panic(err)
    }
    infoLog := log.New(logFile, "[INFO]", log.LstdFlags)
    infoLog.Print(msg)
}

使用协程分别启动监听取消函数isCancel和倒计时函数countdown，然后使用select去监听channel，当有取消动作发生时，则终止程序（即相当于终止倒计时，炸弹不爆炸）

func main() {
    writeLog("---------placing bombs---------")
    cancelSignal := make(chan int)
    litSignal := make(chan int)
    go isCancel(cancelSignal)
    go countdown(litSignal)

    select {
    case <-litSignal:
        booooom()
    case <-cancelSignal:
        return
    }
}

打包编译，先执行一遍正常程序

[INFO]2019/04/11 17:43:34 ---------placing bombs---------
[INFO]2019/04/11 17:43:34 ---------10---------
[INFO]2019/04/11 17:43:35 ---------9---------
[INFO]2019/04/11 17:43:36 ---------8---------
[INFO]2019/04/11 17:43:37 ---------7---------
[INFO]2019/04/11 17:43:38 ---------6---------
[INFO]2019/04/11 17:43:39 ---------5---------
[INFO]2019/04/11 17:43:40 ---------4---------
[INFO]2019/04/11 17:43:41 ---------3---------
[INFO]2019/04/11 17:43:42 ---------2---------
[INFO]2019/04/11 17:43:43 ---------1---------
[INFO]2019/04/11 17:43:44 ---------booooom----------

在倒计时的过程中，随便输入点什么，打断倒计时

[INFO]2019/04/11 17:45:30 ---------placing bombs---------
[INFO]2019/04/11 17:45:30 ---------10---------
[INFO]2019/04/11 17:45:31 ---------9---------
[INFO]2019/04/11 17:45:32 ---------8---------
[INFO]2019/04/11 17:45:33 ---------7---------
[INFO]2019/04/11 17:45:34 ---------6---------
[INFO]2019/04/11 17:45:34 ---------cancel---------

OK！
(参考文章：https://www.cnblogs.com/piperck/p/6435512.html)

最后编辑于：2019.04.24 12:21:29