golang select结构经典应用

原创
2017/05/25 18:10
阅读数 478

###应用场景

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


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

展开阅读全文
Go
打赏
1
0 收藏
分享
加载中
很棒棒哦
2017/05/26 08:31
回复
举报
更多评论
打赏
1 评论
0 收藏
1
分享
返回顶部
顶部