golang select结构经典应用

前端之家收集整理的这篇文章主要介绍了golang select结构经典应用前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

###应用场景

前几天领导交给我一项任务,让我监控某项服务是否启动成功,当时我想就没想就接下了,回来之后考虑了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://www.f2er.com/go/188492.html

猜你在找的Go相关文章