You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
package erasure_coding
import ( "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 ShardBits ShardBits }
func NewEcVolumeInfo(collection string, vid needle.VolumeId, shardBits ShardBits) *EcVolumeInfo { return &EcVolumeInfo{ Collection: collection, VolumeId: vid, ShardBits: shardBits, } }
func (ecInfo *EcVolumeInfo) AddShardId(id ShardId) { ecInfo.ShardBits = ecInfo.ShardBits.AddShardId(id) }
func (ecInfo *EcVolumeInfo) RemoveShardId(id ShardId) { ecInfo.ShardBits = ecInfo.ShardBits.RemoveShardId(id) }
func (ecInfo *EcVolumeInfo) HasShardId(id ShardId) bool { return ecInfo.ShardBits.HasShardId(id) }
func (ecInfo *EcVolumeInfo) ShardIds() (ret []ShardId) { return ecInfo.ShardBits.ShardIds() }
func (ecInfo *EcVolumeInfo) ShardIdCount() (count int) { return ecInfo.ShardBits.ShardIdCount() }
func (ecInfo *EcVolumeInfo) Minus(other *EcVolumeInfo) *EcVolumeInfo { ret := &EcVolumeInfo{ VolumeId: ecInfo.VolumeId, Collection: ecInfo.Collection, ShardBits: ecInfo.ShardBits.Minus(other.ShardBits), }
return ret }
func (ecInfo *EcVolumeInfo) ToVolumeEcShardInformationMessage() (ret *master_pb.VolumeEcShardInformationMessage) { return &master_pb.VolumeEcShardInformationMessage{ Id: uint32(ecInfo.VolumeId), EcIndexBits: uint32(ecInfo.ShardBits), Collection: ecInfo.Collection, } }
type ShardBits uint32 // use bits to indicate the shard id, use 32 bits just for possible future extension
func (b ShardBits) AddShardId(id ShardId) ShardBits { return b | (1 << id) }
func (b ShardBits) RemoveShardId(id ShardId) ShardBits { return b &^ (1 << id) }
func (b ShardBits) HasShardId(id ShardId) bool { return b&(1<<id) > 0 }
func (b ShardBits) ShardIds() (ret []ShardId) { for i := ShardId(0); i < TotalShardsCount; i++ { if b.HasShardId(i) { ret = append(ret, i) } } return }
func (b ShardBits) ShardIdCount() (count int) { for count = 0; b > 0; count++ { b &= b - 1 } return }
func (b ShardBits) Minus(other ShardBits) ShardBits { return b &^ other }
func (b ShardBits) Plus(other ShardBits) ShardBits { return b | other }
|