filer master start up with default ip address instead of just localhost

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()
 }