diff options
Diffstat (limited to 'weed/messaging/client/client.go')
| -rw-r--r-- | weed/messaging/client/client.go | 27 |
1 files changed, 25 insertions, 2 deletions
diff --git a/weed/messaging/client/client.go b/weed/messaging/client/client.go index 9bf9bc71e..3f6d1ca53 100644 --- a/weed/messaging/client/client.go +++ b/weed/messaging/client/client.go @@ -1,11 +1,34 @@ package client +import ( + "context" + + "google.golang.org/grpc" + + "github.com/chrislusf/seaweedfs/weed/pb" + "github.com/chrislusf/seaweedfs/weed/security" + "github.com/chrislusf/seaweedfs/weed/util" +) + type MessagingClient struct { bootstrapBrokers []string + grpcConnection *grpc.ClientConn } -func NewMessagingClient(bootstrapBrokers []string) *MessagingClient { +func NewMessagingClient(bootstrapBrokers []string) (*MessagingClient, error) { + grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.msg_client") + + grpcConnection, err := pb.GrpcDial(context.Background(), "localhost:17777", grpcDialOption) + if err != nil { + return nil, err + } + return &MessagingClient{ bootstrapBrokers: bootstrapBrokers, - } + grpcConnection: grpcConnection, + }, nil +} + +func (mc *MessagingClient) Shutdown() { + mc.grpcConnection.Close() } |
