26 lines
538 B
Go
26 lines
538 B
Go
|
package proptools
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
)
|
||
|
|
|
||
|
|
func ShardBySize[T ~[]E, E any](toShard T, shardSize int) []T {
|
||
|
|
if len(toShard) == 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
ret := make([]T, 0, (len(toShard)+shardSize-1)/shardSize)
|
||
|
|
for len(toShard) > shardSize {
|
||
|
|
ret = append(ret, toShard[0:shardSize])
|
||
|
|
toShard = toShard[shardSize:]
|
||
|
|
}
|
||
|
|
if len(toShard) > 0 {
|
||
|
|
ret = append(ret, toShard)
|
||
|
|
}
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
func ShardByCount[T ~[]E, E any](toShard T, shardCount int) []T {
|
||
|
|
return ShardBySize(toShard, int(math.Ceil(float64(len(toShard))/float64(shardCount))))
|
||
|
|
}
|