mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package topic
|
|
|
|
import (
|
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/util/log_buffer"
|
|
"time"
|
|
)
|
|
|
|
type LocalPartition struct {
|
|
Partition
|
|
isLeader bool
|
|
FollowerBrokers []pb.ServerAddress
|
|
logBuffer *log_buffer.LogBuffer
|
|
}
|
|
|
|
type OnEachMessageFn func(logEntry *filer_pb.LogEntry) error
|
|
|
|
func (p LocalPartition) Publish(message *mq_pb.DataMessage) {
|
|
p.logBuffer.AddToBuffer(message.Key, message.Value, time.Now().UnixNano())
|
|
}
|
|
|
|
func (p LocalPartition) Subscribe(clientName string, startReadTime time.Time, eachMessageFn OnEachMessageFn) {
|
|
p.logBuffer.LoopProcessLogData(clientName, startReadTime, 0, func() bool {
|
|
return true
|
|
}, eachMessageFn)
|
|
}
|
|
|
|
func FromPbBrokerPartitionsAssignment(self pb.ServerAddress, assignment *mq_pb.BrokerPartitionsAssignment) *LocalPartition {
|
|
isLeaer := assignment.LeaderBroker == string(self)
|
|
localPartition := &LocalPartition{
|
|
Partition: Partition{
|
|
RangeStart: assignment.PartitionStart,
|
|
RangeStop: assignment.PartitionStop,
|
|
RingSize: PartitionCount,
|
|
},
|
|
isLeader: isLeaer,
|
|
}
|
|
if !isLeaer {
|
|
return localPartition
|
|
}
|
|
followers := make([]pb.ServerAddress, len(assignment.FollowerBrokers))
|
|
for i, follower := range assignment.FollowerBrokers {
|
|
followers[i] = pb.ServerAddress(follower)
|
|
}
|
|
localPartition.FollowerBrokers = followers
|
|
return localPartition
|
|
}
|