以前在調用grpc接口時，總要調用Dial函數來創建一個grpc的ClientConn，如果每次都要創建，那么開銷代價是很大的。

grpc的ClientConn對象可以幫我們實現自動重連的機制，并且是并發安全的，因此可以定義一個全局的ClientConn，然后用到的時候就用這個連接。

下面是簡易實現：

package delayqueueimport ("google.golang.org/grpc""sync""sync/atomic""unsafe" )var (globalClientConn unsafe.Pointerlck sync.Mutex )func GetClient(target string) (DelayQueueClient, error) { //返回需要的clientconn, err := GetConn(target)if err != nil {return (DelayQueueClient)(nil), err}return NewDelayQueueClient(conn), nil //此處調用pb.go文件中生成的創建client的方法 }func GetConn(target string) (*grpc.ClientConn, error) {if atomic.LoadPointer(&globalClientConn) != nil {return (*grpc.ClientConn)(globalClientConn), nil}lck.Lock()defer lck.Unlock()if atomic.LoadPointer(&globalClientConn) != nil { //double checkreturn (*grpc.ClientConn)(globalClientConn), nil}cli, err := newGrpcConn(target)if err != nil {return nil, err}atomic.StorePointer(&globalClientConn, unsafe.Pointer(cli))return cli, nil }func newGrpcConn(target string) (*grpc.ClientConn, error) {conn, err := grpc.Dial(target,grpc.WithInsecure(),)if err != nil {return nil, err}return conn, nil }

?

總結

以上是生活随笔為你收集整理的grpc复用client连接的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。