每日一库：实现多个grpc 连接池shimingyah/pool

1128 阅读 0 评论 745 点赞

我是靠谱客的博主每日一库，这篇文章主要介绍每日一库：实现多个grpc 连接池shimingyah/pool，现在分享给大家，希望可以做个参考。

Go 的 grpc 客户端的连接池，支持连接重用。

池提供附加功能：

Connection reuse由特定的 MaxConcurrentStreams 参数支持。
Failure reconnection由 grpc 的 keepalive 支持。

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package main

import (
   "fmt"
   "github.com/shimingyah/pool"
   "google.golang.org/grpc"
   "log"
   "sync"
)

var (
   grpcOnce sync.Once
   poolConn  pool.Pool
)

func main() {

}

//参考: https://github.com/shimingyah/pool
func GetGrpcClient() *grpc.ClientConn {
   grpcOnce.Do(func() {
      var err error
      //grpc 服务器的地址或者 服务注册中心的地址
      address :=""
      poolConn, err = pool.New(address, pool.DefaultOptions)
      if err != nil {
         log.Fatalf("failed to new pool: %v", err)
      }
   })
   conn, err := poolConn.Get()
   if err != nil {
      log.Fatalf("failed to get conn: %v", err)
   }
   return conn.Value()
}