aboutsummaryrefslogtreecommitdiff
path: root/weed/messaging/client/client.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/messaging/client/client.go')
-rw-r--r--weed/messaging/client/client.go27
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()
}