美文网首页Go
go select使用的一个demo

go select使用的一个demo

作者: 嘉磊 | 来源:发表于2019-04-11 17:46 被阅读10次

    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)

    相关文章

      网友评论

        本文标题:go select使用的一个demo

        本文链接:https://www.haomeiwen.com/subject/jxdxwqtx.html