lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-09-24 02:13:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

sub broker wait for partition

Browse Source
This commit is contained in:
chrislu
2024-01-22 11:56:55 -08:00
parent 2beaa2d0b3
commit 3ff6b31d94
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
weed/mq/broker/broker_grpc_sub.go
View File

@@ -21,6 +21,7 @@ func (b *MessageQueueBroker) SubscribeMessage(req *mq_pb.SubscribeMessageRequest
glog.V(0).Infof("Subscriber %s on %v %v connected", req.GetInit().ConsumerId, t, partition) glog.V(0).Infof("Subscriber %s on %v %v connected", req.GetInit().ConsumerId, t, partition)
waitIntervalCount := 0
var localTopicPartition *topic.LocalPartition var localTopicPartition *topic.LocalPartition
for localTopicPartition == nil { for localTopicPartition == nil {
localTopicPartition = b.localTopicManager.GetTopicPartition(t, partition) localTopicPartition = b.localTopicManager.GetTopicPartition(t, partition)
@@ -32,7 +33,11 @@ func (b *MessageQueueBroker) SubscribeMessage(req *mq_pb.SubscribeMessageRequest
if localTopicPartition != nil { if localTopicPartition != nil {
break break
} }
time.Sleep(337 * time.Millisecond) waitIntervalCount++
if waitIntervalCount > 10 {
waitIntervalCount = 10
}
time.Sleep(time.Duration(waitIntervalCount) * 337 * time.Millisecond)
// Check if the client has disconnected by monitoring the context // Check if the client has disconnected by monitoring the context
select { select {
case <-ctx.Done(): case <-ctx.Done():