golang select结构经典应用

2019独角兽企业重金招聘Python工程师标准>>>

###应用场景

前几天领导交给我一项任务，让我监控某项服务是否启动成功，当时我想就没想就接下了，回来之后考虑了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")
	}
}

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

转载于:https://my.oschina.net/u/1455335/blog/909155