opensource/dtm
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

grpc connections use sync.Map

Browse Source
This commit is contained in:
yedf2 2021-08-17 10:03:36 +08:00
parent 57fa97e942
commit fa98442a7b
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
dtmgrpc/type.go
View File

@ -4,6 +4,7 @@ import (
context "context"
"fmt"
"strings"
sync "sync"
"github.com/yedf/dtm/dtmcli"
grpc "google.golang.org/grpc"
@ -12,20 +13,21 @@ import (
"google.golang.org/protobuf/types/known/emptypb"
)
var clients = map[string]*grpc.ClientConn{}
var clients sync.Map
// GetGrpcConn 1
func GetGrpcConn(grpcServer string) (conn *grpc.ClientConn, rerr error) {
if clients[grpcServer] == nil {
v, ok := clients.Load(grpcServer)
if !ok {
dtmcli.Logf("grpc client connecting %s", grpcServer)
conn, err := grpc.Dial(grpcServer, grpc.WithInsecure(), grpc.WithUnaryInterceptor(GrpcClientLog))
if err == nil {
clients[grpcServer] = conn
conn, rerr := grpc.Dial(grpcServer, grpc.WithInsecure(), grpc.WithUnaryInterceptor(GrpcClientLog))
if rerr == nil {
clients.Store(grpcServer, conn)
v = conn
dtmcli.Logf("grpc client inited for %s", grpcServer)
}
}
conn = clients[grpcServer]
return
return v.(*grpc.ClientConn), rerr
}
// MustGetGrpcConn 1

4
examples/grpc_tcc.go
View File

@ -11,9 +11,7 @@ func init() {
gid := dtmgrpc.MustGenGid(DtmGrpcServer)
err := dtmgrpc.TccGlobalTransaction(DtmGrpcServer, gid, func(tcc *dtmgrpc.TccGrpc) error {
data := dtmcli.MustMarshal(&TransReq{Amount: 30})
r, err := tcc.CallBranch(data, BusiGrpc+"/examples.Busi/TransOutTcc", BusiGrpc+"/examples.Busi/TransOutConfirm", BusiGrpc+"/examples.Busi/TransOutRevert")
dtmcli.LogRedf("callbranch return %v", r)
dtmcli.LogRedf("callbranch return data %s", string(r.BusiData))
_, err := tcc.CallBranch(data, BusiGrpc+"/examples.Busi/TransOutTcc", BusiGrpc+"/examples.Busi/TransOutConfirm", BusiGrpc+"/examples.Busi/TransOutRevert")
if err != nil {
return err
}