golang select结构经典应用

###应用场景mysql

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

###1、主动golang

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

###2、被动api

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

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

//如检测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")
	}
}

###总结:主动方式要比轮询方式效率高不少,要结合本身的实际状况来选择。code