lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-08-01 17:46:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
chrislu 2024-05-17 18:21:21 -07:00
parent 94742b8ace
commit 8d5bb7420d
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
weed/mq/topic/local_partition.go
View File

@ -28,7 +28,7 @@ type LocalPartition struct {
Publishers *LocalPartitionPublishers Publishers *LocalPartitionPublishers
Subscribers *LocalPartitionSubscribers Subscribers *LocalPartitionSubscribers
followerStream mq_pb.SeaweedMessaging_PublishFollowMeClient publishFolloweMeStream mq_pb.SeaweedMessaging_PublishFollowMeClient
followerGrpcConnection *grpc.ClientConn followerGrpcConnection *grpc.ClientConn
follower string follower string
} }
@ -55,9 +55,9 @@ func (p *LocalPartition) Publish(message *mq_pb.DataMessage) error {
p.LogBuffer.AddToBuffer(message) p.LogBuffer.AddToBuffer(message)
// maybe send to the follower // maybe send to the follower
if p.followerStream != nil { if p.publishFolloweMeStream != nil {
// println("recv", string(message.Key), message.TsNs) // println("recv", string(message.Key), message.TsNs)
if followErr := p.followerStream.Send(&mq_pb.PublishFollowMeRequest{ if followErr := p.publishFolloweMeStream.Send(&mq_pb.PublishFollowMeRequest{
Message: &mq_pb.PublishFollowMeRequest_Data{ Message: &mq_pb.PublishFollowMeRequest_Data{
Data: message, Data: message,
}, },
@ -134,7 +134,7 @@ func (p *LocalPartition) WaitUntilNoPublishers() {
} }
func (p *LocalPartition) MaybeConnectToFollowers(initMessage *mq_pb.PublishMessageRequest_InitMessage, grpcDialOption grpc.DialOption) (err error) { func (p *LocalPartition) MaybeConnectToFollowers(initMessage *mq_pb.PublishMessageRequest_InitMessage, grpcDialOption grpc.DialOption) (err error) {
if p.followerStream != nil { if p.publishFolloweMeStream != nil {
return nil return nil
} }
if len(initMessage.FollowerBrokers) == 0 { if len(initMessage.FollowerBrokers) == 0 {
@ -148,11 +148,11 @@ func (p *LocalPartition) MaybeConnectToFollowers(initMessage *mq_pb.PublishMessa
return fmt.Errorf("fail to dial %s: %v", p.follower, err) return fmt.Errorf("fail to dial %s: %v", p.follower, err)
} }
followerClient := mq_pb.NewSeaweedMessagingClient(p.followerGrpcConnection) followerClient := mq_pb.NewSeaweedMessagingClient(p.followerGrpcConnection)
p.followerStream, err = followerClient.PublishFollowMe(ctx) p.publishFolloweMeStream, err = followerClient.PublishFollowMe(ctx)
if err != nil { if err != nil {
return fmt.Errorf("fail to create publish client: %v", err) return fmt.Errorf("fail to create publish client: %v", err)
} }
if err = p.followerStream.Send(&mq_pb.PublishFollowMeRequest{ if err = p.publishFolloweMeStream.Send(&mq_pb.PublishFollowMeRequest{
Message: &mq_pb.PublishFollowMeRequest_Init{ Message: &mq_pb.PublishFollowMeRequest_Init{
Init: &mq_pb.PublishFollowMeRequest_InitMessage{ Init: &mq_pb.PublishFollowMeRequest_InitMessage{
Topic: initMessage.Topic, Topic: initMessage.Topic,
@ -170,7 +170,7 @@ func (p *LocalPartition) MaybeConnectToFollowers(initMessage *mq_pb.PublishMessa
}() }()
for { for {
ack, err := p.followerStream.Recv() ack, err := p.publishFolloweMeStream.Recv()
if err != nil { if err != nil {
e, _ := status.FromError(err) e, _ := status.FromError(err)
if e.Code() == codes.Canceled { if e.Code() == codes.Canceled {
@ -194,9 +194,9 @@ func (p *LocalPartition) MaybeShutdownLocalPartition() (hasShutdown bool) {
for !p.LogBuffer.IsAllFlushed() { for !p.LogBuffer.IsAllFlushed() {
time.Sleep(113 * time.Millisecond) time.Sleep(113 * time.Millisecond)
} }
if p.followerStream != nil { if p.publishFolloweMeStream != nil {
// send close to the follower // send close to the follower
if followErr := p.followerStream.Send(&mq_pb.PublishFollowMeRequest{ if followErr := p.publishFolloweMeStream.Send(&mq_pb.PublishFollowMeRequest{
Message: &mq_pb.PublishFollowMeRequest_Close{ Message: &mq_pb.PublishFollowMeRequest_Close{
Close: &mq_pb.PublishFollowMeRequest_CloseMessage{}, Close: &mq_pb.PublishFollowMeRequest_CloseMessage{},
}, },
@ -205,7 +205,7 @@ func (p *LocalPartition) MaybeShutdownLocalPartition() (hasShutdown bool) {
} }
glog.V(4).Infof("closing grpcConnection to follower") glog.V(4).Infof("closing grpcConnection to follower")
p.followerGrpcConnection.Close() p.followerGrpcConnection.Close()
p.followerStream = nil p.publishFolloweMeStream = nil
p.follower = "" p.follower = ""
} }
@ -224,8 +224,8 @@ func (p *LocalPartition) Shutdown() {
} }
func (p *LocalPartition) NotifyLogFlushed(flushTsNs int64) { func (p *LocalPartition) NotifyLogFlushed(flushTsNs int64) {
if p.followerStream != nil { if p.publishFolloweMeStream != nil {
if followErr := p.followerStream.Send(&mq_pb.PublishFollowMeRequest{ if followErr := p.publishFolloweMeStream.Send(&mq_pb.PublishFollowMeRequest{
Message: &mq_pb.PublishFollowMeRequest_Flush{ Message: &mq_pb.PublishFollowMeRequest_Flush{
Flush: &mq_pb.PublishFollowMeRequest_FlushMessage{ Flush: &mq_pb.PublishFollowMeRequest_FlushMessage{
TsNs: flushTsNs, TsNs: flushTsNs,