mirror of https://github.com/dtm-labs/dtm.git
csharpjavadistributed-transactionsdtmgogolangmicroservicenodejsphpdatabasesagaseatatcctransactiontransactionsxapythondistributed
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
985 B
35 lines
985 B
/*
|
|
* Copyright (c) 2021 yedf. All rights reserved.
|
|
* Use of this source code is governed by a BSD-style
|
|
* license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
package dtmgrpc
|
|
|
|
import (
|
|
context "context"
|
|
|
|
"github.com/dtm-labs/dtm/dtmcli/dtmimp"
|
|
"github.com/dtm-labs/dtm/dtmgrpc/dtmgimp"
|
|
"github.com/dtm-labs/dtmdriver"
|
|
grpc "google.golang.org/grpc"
|
|
emptypb "google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
// MustGenGid must gen a gid from grpcServer
|
|
func MustGenGid(grpcServer string) string {
|
|
dc := dtmgimp.MustGetDtmClient(grpcServer)
|
|
r, err := dc.NewGid(context.Background(), &emptypb.Empty{})
|
|
dtmimp.E2P(err)
|
|
return r.Gid
|
|
}
|
|
|
|
// UseDriver use the specified driver to handle grpc urls
|
|
func UseDriver(driverName string) error {
|
|
return dtmdriver.Use(driverName)
|
|
}
|
|
|
|
// AddUnaryInterceptor adds grpc.UnaryClientInterceptor
|
|
func AddUnaryInterceptor(interceptor grpc.UnaryClientInterceptor) {
|
|
dtmgimp.ClientInterceptors = append(dtmgimp.ClientInterceptors, interceptor)
|
|
}
|
|
|