gsh-digital-services/seaweedFS
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add follower id info

Browse Source
This commit is contained in:
chrislu
2024-03-16 17:11:18 -07:00
parent b74e8082ba
commit 3f2dd8cc3e
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
weed/mq/broker/broker_grpc_pub_follow.go
View File

@@ -23,7 +23,7 @@ func (b *MessageQueueBroker) PublishFollowMe(c context.Context, request *mq_pb.P
Message: &mq_pb.FollowInMemoryMessagesRequest_Init{ Message: &mq_pb.FollowInMemoryMessagesRequest_Init{
Init: &mq_pb.FollowInMemoryMessagesRequest_InitMessage{ Init: &mq_pb.FollowInMemoryMessagesRequest_InitMessage{
ConsumerGroup: string(b.option.BrokerAddress()), ConsumerGroup: string(b.option.BrokerAddress()),
ConsumerId: fmt.Sprintf("followMe-%d", followerId), ConsumerId: fmt.Sprintf("followMe@%s-%d", b.option.BrokerAddress(), followerId),
FollowerId: followerId, FollowerId: followerId,
Topic: request.Topic, Topic: request.Topic,
PartitionOffset: &mq_pb.PartitionOffset{ PartitionOffset: &mq_pb.PartitionOffset{
@@ -52,7 +52,7 @@ func (b *MessageQueueBroker) PublishFollowMe(c context.Context, request *mq_pb.P
} }
wg.Done() wg.Done()
b.doFollowInMemoryMessage(context.Background(), subscribeClient) b.doFollowInMemoryMessage(context.Background(), followerId, subscribeClient)
return nil return nil
}) })
@@ -60,7 +60,7 @@ func (b *MessageQueueBroker) PublishFollowMe(c context.Context, request *mq_pb.P
return &mq_pb.PublishFollowMeResponse{}, ret return &mq_pb.PublishFollowMeResponse{}, ret
} }
func (b *MessageQueueBroker) doFollowInMemoryMessage(c context.Context, client mq_pb.SeaweedMessaging_FollowInMemoryMessagesClient) { func (b *MessageQueueBroker) doFollowInMemoryMessage(c context.Context, followerId int32, client mq_pb.SeaweedMessaging_FollowInMemoryMessagesClient) {
for { for {
resp, err := client.Recv() resp, err := client.Recv()
if err != nil { if err != nil {
@@ -87,9 +87,11 @@ func (b *MessageQueueBroker) doFollowInMemoryMessage(c context.Context, client m
} }
if m.Ctrl.FollowerChangedToId != 0 { if m.Ctrl.FollowerChangedToId != 0 {
// follower changed // follower changed
glog.V(0).Infof("doFollowInMemoryMessage follower changed to %d", m.Ctrl.FollowerChangedToId) glog.V(0).Infof("doFollowInMemoryMessage follower changed from %d to %d", followerId, m.Ctrl.FollowerChangedToId)
return return
} }
default:
glog.V(0).Infof("doFollowInMemoryMessage unknown message type: %v", m)
} }
} }
} }