mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package erasure_coding
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/pb/master_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/storage/needle"
|
|
)
|
|
|
|
// data structure used in master
|
|
type EcVolumeInfo struct {
|
|
VolumeId needle.VolumeId
|
|
Collection string
|
|
shardIds uint16 // use bits to indicate the shard id
|
|
}
|
|
|
|
func NewEcVolumeInfo(collection string, vid needle.VolumeId) *EcVolumeInfo {
|
|
return &EcVolumeInfo{
|
|
Collection: collection,
|
|
VolumeId: vid,
|
|
}
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) AddShardId(id ShardId) {
|
|
ecInfo.shardIds |= (1 << id)
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) RemoveShardId(id ShardId) {
|
|
ecInfo.shardIds &^= (1 << id)
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) HasShardId(id ShardId) bool {
|
|
return ecInfo.shardIds&(1<<id) > 0
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) ShardIds() (ret []ShardId) {
|
|
for i := ShardId(0); i < DataShardsCount+ParityShardsCount; i++ {
|
|
if ecInfo.HasShardId(i) {
|
|
ret = append(ret, i)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) Minus(other *EcVolumeInfo) (*EcVolumeInfo) {
|
|
ret := &EcVolumeInfo{
|
|
VolumeId: ecInfo.VolumeId,
|
|
Collection: ecInfo.Collection,
|
|
shardIds: ecInfo.shardIds &^ other.shardIds,
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) ToVolumeEcShardInformationMessage() (ret []*master_pb.VolumeEcShardInformationMessage) {
|
|
for _, shard := range ecInfo.ShardIds() {
|
|
ret = append(ret, &master_pb.VolumeEcShardInformationMessage{
|
|
Id: uint32(ecInfo.VolumeId),
|
|
EcIndex: uint32(shard),
|
|
Collection: ecInfo.Collection,
|
|
})
|
|
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ecInfo *EcVolumeInfo) String() string {
|
|
return fmt.Sprintf("id:%d shard:%v collection:%v", ecInfo.VolumeId, ecInfo.ShardIds(), ecInfo.Collection)
|
|
}
|