GO-Grpc微服务开发五 服务调用优化

使用连接池

    func init() {
    so := sync.Once{}
    so.Do(func() {
        //初始化client
        var err error
        base := &fr_common.Base{}
        if base.GetEnv() == "prod" {
            consul_addr = fr_common.CONSUL_ADDR_PROD
            concul_token = fr_common.CONSUL_TOKEN_PROD
        } else {
            consul_addr = fr_common.CONSUL_ADDR_DEV
            concul_token = fr_common.CONSUL_TOKEN_DEV
        }
        config := consul.Config{
            Address:consul_addr,
            Token:concul_token,
        }
        client, err = consul.NewClient(&config)//非默认情况下需要设置实际的参数
        if err != nil {
            client = nil
            fmt.Printf("did not connect: %v", err.Error())
        }

        //初始化连接池
        pool = grpcp.New(func(addr string) (*grpc.ClientConn, error) {
            return grpc.Dial(
                addr,
                grpc.WithInsecure(),
            )
        })
    })
}

/**
 * 获取服务
 * @param name string 服务名称
 * @return service 服务实例
 */
func GetSrvHandel(name string) (conn *grpc.ClientConn, err error) {
    defer func() {
        if err := recover(); err != nil {
            fmt.Println(err)
        }
    }()
    service, _, err = client.Catalog().Service(name, "", nil)
    if err != nil {
        fmt.Printf("can not find services: %v", err.Error())
        return nil, err
    }
    //多服务端时遍历链接直到成功
    for _, v := range service  {
        addr := fmt.Sprintf("%s:%d", v.ServiceAddress, v.ServicePort)
        conn, err = pool.GetConn(addr)
    }
    return conn, err
}

源码：go-grpc-getway