This commit is contained in:
chrislu
2023-10-11 21:44:56 -07:00
parent fb450b2a5d
commit 35b5264ab7
8 changed files with 510 additions and 552 deletions

View File

@@ -12,8 +12,8 @@ import (
"time"
)
// BrokerConnectToBalancer connects to the broker balancer and sends stats
func (broker *MessageQueueBroker) BrokerConnectToBalancer(self string) error {
// BrokerConnectToPubCoordinator connects to the broker balancer and sends stats
func (broker *MessageQueueBroker) BrokerConnectToPubCoordinator(self string) error {
// find the lock owner
var brokerBalancer string
err := broker.WithFilerClient(false, func(client filer_pb.SeaweedFilerClient) error {
@@ -35,14 +35,14 @@ func (broker *MessageQueueBroker) BrokerConnectToBalancer(self string) error {
// connect to the lock owner
err = pb.WithBrokerGrpcClient(false, brokerBalancer, broker.grpcDialOption, func(client mq_pb.SeaweedMessagingClient) error {
stream, err := client.ConnectToBalancer(context.Background())
stream, err := client.ConnectToPubCoordinator(context.Background())
if err != nil {
return fmt.Errorf("connect to balancer %v: %v", brokerBalancer, err)
}
defer stream.CloseSend()
err = stream.Send(&mq_pb.ConnectToBalancerRequest{
Message: &mq_pb.ConnectToBalancerRequest_Init{
Init: &mq_pb.ConnectToBalancerRequest_InitMessage{
err = stream.Send(&mq_pb.ConnectToPubCoordinatorRequest{
Message: &mq_pb.ConnectToPubCoordinatorRequest_Init{
Init: &mq_pb.ConnectToPubCoordinatorRequest_InitMessage{
Broker: self,
},
},
@@ -53,8 +53,8 @@ func (broker *MessageQueueBroker) BrokerConnectToBalancer(self string) error {
for {
stats := broker.localTopicManager.CollectStats(time.Second * 5)
err = stream.Send(&mq_pb.ConnectToBalancerRequest{
Message: &mq_pb.ConnectToBalancerRequest_Stats{
err = stream.Send(&mq_pb.ConnectToPubCoordinatorRequest{
Message: &mq_pb.ConnectToPubCoordinatorRequest_Stats{
Stats: stats,
},
})