mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-21 06:57:51 +08:00
dc784bf217
* listing files to convert to parquet * write parquet files * save logs into parquet files * pass by value * compact logs into parquet format * can skip existing files * refactor * refactor * fix compilation * when no partition found * refactor * add untested parquet file read * rename package * refactor * rename files * remove unused * add merged log read func * parquet wants to know the file size * rewind by time * pass in stop ts * add stop ts * adjust log * minor * adjust log * skip .parquet files when reading message logs * skip non message files * Update subscriber_record.go * send messages * skip message data with only ts * skip non log files * update parquet-go package * ensure a valid record type * add new field to a record type * Update read_parquet_to_log.go * fix parquet file name generation * separating reading parquet and logs * add key field * add skipped logs * use in memory cache * refactor * refactor * refactor * refactor, and change compact log * refactor * rename * refactor * fix format * prefix v to version directory
109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package topic
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
"time"
|
|
)
|
|
|
|
const PartitionCount = 4096
|
|
|
|
type Partition struct {
|
|
RangeStart int32
|
|
RangeStop int32 // exclusive
|
|
RingSize int32
|
|
UnixTimeNs int64 // in nanoseconds
|
|
}
|
|
|
|
func NewPartition(rangeStart, rangeStop, ringSize int32, unixTimeNs int64) *Partition {
|
|
return &Partition{
|
|
RangeStart: rangeStart,
|
|
RangeStop: rangeStop,
|
|
RingSize: ringSize,
|
|
UnixTimeNs: unixTimeNs,
|
|
}
|
|
}
|
|
|
|
func (partition Partition) Equals(other Partition) bool {
|
|
if partition.RangeStart != other.RangeStart {
|
|
return false
|
|
}
|
|
if partition.RangeStop != other.RangeStop {
|
|
return false
|
|
}
|
|
if partition.RingSize != other.RingSize {
|
|
return false
|
|
}
|
|
if partition.UnixTimeNs != other.UnixTimeNs {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func FromPbPartition(partition *mq_pb.Partition) Partition {
|
|
return Partition{
|
|
RangeStart: partition.RangeStart,
|
|
RangeStop: partition.RangeStop,
|
|
RingSize: partition.RingSize,
|
|
UnixTimeNs: partition.UnixTimeNs,
|
|
}
|
|
}
|
|
|
|
func SplitPartitions(targetCount int32, ts int64) []*Partition {
|
|
partitions := make([]*Partition, 0, targetCount)
|
|
partitionSize := PartitionCount / targetCount
|
|
for i := int32(0); i < targetCount; i++ {
|
|
partitionStop := (i + 1) * partitionSize
|
|
if i == targetCount-1 {
|
|
partitionStop = PartitionCount
|
|
}
|
|
partitions = append(partitions, &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,
|
|
}
|
|
}
|
|
|
|
func (partition Partition) Overlaps(partition2 Partition) bool {
|
|
if partition.RangeStart >= partition2.RangeStop {
|
|
return false
|
|
}
|
|
if partition.RangeStop <= partition2.RangeStart {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (partition Partition) String() string {
|
|
return fmt.Sprintf("%04d-%04d", partition.RangeStart, partition.RangeStop)
|
|
}
|
|
|
|
func ParseTopicVersion(name string) (t time.Time, err error) {
|
|
return time.Parse(PartitionGenerationFormat, name)
|
|
}
|
|
|
|
func ParsePartitionBoundary(name string) (start, stop int32) {
|
|
_, err := fmt.Sscanf(name, "%04d-%04d", &start, &stop)
|
|
if err != nil {
|
|
return 0, 0
|
|
}
|
|
return start, stop
|
|
}
|
|
|
|
func PartitionDir(t Topic, p Partition) string {
|
|
partitionGeneration := time.Unix(0, p.UnixTimeNs).UTC().Format(PartitionGenerationFormat)
|
|
return fmt.Sprintf("%s/%s/%04d-%04d", t.Dir(), partitionGeneration, p.RangeStart, p.RangeStop)
|
|
}
|