美文网首页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