aboutsummaryrefslogtreecommitdiff
path: root/weed/messaging/client/client.go
blob: 910fd02e9568139f8b98969721a22f93766e5042 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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, error) {
	grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.msg_client")

	grpcConnection, err := pb.GrpcDial(context.Background(), "localhost:17777", grpcDialOption)
	if err != nil {
		return nil, err
	}

	util.OnInterrupt(func() {
		grpcConnection.Close()
	})

	return &MessagingClient{
		bootstrapBrokers: bootstrapBrokers,
		grpcConnection:   grpcConnection,
	}, nil
}