golang select结构经典应用

摘要: golang 的 select 的功能和 select, poll, epoll 相似， 就是监听 IO 操作，当 IO 操作发生时，触发相应的动作。

###应用场景

前几天领导交给我一项任务，让我监控某项服务是否启动成功，当时我想就没想就接下了，回来之后考虑了N多种方案。大致分为2类

###一、主动

就是把自身信息主动推上去，如服务启动后把自身信息注册到zookeeper，这样调用zookeeper的api就可以看到

###二、被动

使用轮询方式拉取服务状态,如使用定时器或for循环主动去获取服务器状态，当然这种方式并不是永无止境的，要设置一个临值

使用定时器和for循环这种方式我觉得有点太老套，下面我就结合golang 当中的 select 和goroutine来实现。

//如检测mysql是否链路是否畅通
func () testConn(host, port string) error {
ch := make(chan int, 1)    //定义一个带缓冲通道的goroutine
go func() {
var err error
var c net.Conn
for {
c, err = net.DialTimeout("tcp", "localhost:3306", 20 * time.Second)
//如果err ==nil 就代表链路是畅通的，此时退出就可以了
if err == nil {
c.Close()
ch <- 1//随便向通道里放一个int类型的值就可以
} else {
//如果链路不同，就休眠两秒继续检测
log.Errorf("failed to connect to db, retry after 2 seconds :%v", err)
time.Sleep(2 * time.Second) //休眠两秒
}
}
}()
//select 会监听routine的数据流动，如果有就退出，否则会等待超时(也就是60秒)
select {
case <-ch: //有数据就退出
return nil
case <-time.After(60 * time.Second): //如果等待60秒还没有监听到ch当中有数据流动就抛出异常，程序结束
return errors.New("failed to connect to database after 60 seconds")
}
}

###总结：主动方式要比轮询方式效率高很多，要结合自己的实际情况来选择。