diff options
Diffstat (limited to 'weed/mq/topic/partition.go')
| -rw-r--r-- | weed/mq/topic/partition.go | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/weed/mq/topic/partition.go b/weed/mq/topic/partition.go index 79c830f13..ca34c2390 100644 --- a/weed/mq/topic/partition.go +++ b/weed/mq/topic/partition.go @@ -8,6 +8,7 @@ type Partition struct { RangeStart int32 RangeStop int32 // exclusive RingSize int32 + UnixTimeNs int64 // in nanoseconds } func (partition Partition) Equals(other Partition) bool { @@ -20,6 +21,9 @@ func (partition Partition) Equals(other Partition) bool { if partition.RingSize != other.RingSize { return false } + if partition.UnixTimeNs != other.UnixTimeNs { + return false + } return true } @@ -28,10 +32,11 @@ func FromPbPartition(partition *mq_pb.Partition) Partition { RangeStart: partition.RangeStart, RangeStop: partition.RangeStop, RingSize: partition.RingSize, + UnixTimeNs: partition.UnixTimeNs, } } -func SplitPartitions(targetCount int32) []*Partition { +func SplitPartitions(targetCount int32, ts int64) []*Partition { partitions := make([]*Partition, 0, targetCount) partitionSize := PartitionCount / targetCount for i := int32(0); i < targetCount; i++ { @@ -43,7 +48,17 @@ func SplitPartitions(targetCount int32) []*Partition { RangeStart: i * partitionSize, RangeStop: partitionStop, RingSize: PartitionCount, + UnixTimeNs: ts, }) } return partitions } + +func (partition Partition) ToPbPartition() *mq_pb.Partition { + return &mq_pb.Partition{ + RangeStart: partition.RangeStart, + RangeStop: partition.RangeStop, + RingSize: partition.RingSize, + UnixTimeNs: partition.UnixTimeNs, + } +} |