broker report stats to balancer

2023-09-16 23:00:51 -07:00
parent a1cb49d078
commit a65088d739
11 changed files with 645 additions and 470 deletions
--- a/weed/mq/broker/broker_grpc_balancer.go
+++ b/weed/mq/broker/broker_grpc_balancer.go
@@ -40,9 +40,7 @@ func (broker *MessageQueueBroker) ConnectToBalancer(stream mq_pb.SeaweedMessagin
 			return status.Errorf(codes.Unavailable, "not current broker balancer")
 		}
 		if receivedStats := req.GetStats(); receivedStats != nil {
-			brokerStats.TopicPartitionCount = receivedStats.TopicPartitionCount
-			brokerStats.ConsumerCount = receivedStats.ConsumerCount
-			brokerStats.CpuUsagePercent = receivedStats.CpuUsagePercent
+			brokerStats.UpdateStats(receivedStats)

 			glog.V(3).Infof("broker %s stats: %+v", initMessage.Broker, brokerStats)
 		}