forked from TrueCloudLab/frostfs-api-go
97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package storagegroup
|
|
|
|
import (
|
|
"bytes"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neofs-api-go/hash"
|
|
"github.com/nspcc-dev/neofs-api-go/refs"
|
|
)
|
|
|
|
type (
|
|
// Hash is alias of hash.Hash for proto definition.
|
|
Hash = hash.Hash
|
|
|
|
// Provider is an interface that defines storage group instance.
|
|
// There was different storage group implementation. Right now it
|
|
// is implemented as extended header in the object.
|
|
Provider interface {
|
|
// Group returns list of object ids of the storage group.
|
|
// This list **should be** sorted.
|
|
Group() []refs.ObjectID
|
|
|
|
IDInfo() *IdentificationInfo
|
|
Zones() []ZoneInfo
|
|
}
|
|
|
|
// ZoneInfo provides validation information of storage group.
|
|
ZoneInfo struct {
|
|
hash.Hash
|
|
Size uint64
|
|
}
|
|
|
|
// IdentificationInfo provides meta information about storage group.
|
|
IdentificationInfo struct {
|
|
CID refs.CID
|
|
SGID refs.SGID
|
|
OwnerID refs.OwnerID
|
|
}
|
|
|
|
// IDList is a slice of object ids, that can be sorted.
|
|
IDList []refs.ObjectID
|
|
)
|
|
|
|
var _ proto.Message = (*StorageGroup)(nil)
|
|
|
|
// String returns string representation of StorageGroup.
|
|
func (m *StorageGroup) String() string {
|
|
b := new(strings.Builder)
|
|
b.WriteString("<SG")
|
|
b.WriteString(" VDS=" + strconv.FormatUint(m.ValidationDataSize, 10))
|
|
data := base58.Encode(m.ValidationHash.Bytes())
|
|
b.WriteString(" Hash=" + data[:3] + "..." + data[len(data)-3:])
|
|
if m.Lifetime != nil {
|
|
b.WriteString(" Lifetime=(" + m.Lifetime.Unit.String() + " " + strconv.FormatInt(m.Lifetime.Value, 10) + ")")
|
|
}
|
|
b.WriteByte('>')
|
|
return b.String()
|
|
}
|
|
|
|
// Empty checks if storage group has some data for validation.
|
|
func (m StorageGroup) Empty() bool {
|
|
return m.ValidationDataSize == 0 && m.ValidationHash.Equal(hash.Hash{})
|
|
}
|
|
|
|
// Len returns amount of object ids in IDList.
|
|
func (s IDList) Len() int { return len(s) }
|
|
|
|
// Less returns byte comparision between IDList[i] and IDList[j].
|
|
func (s IDList) Less(i, j int) bool { return bytes.Compare(s[i].Bytes(), s[j].Bytes()) == -1 }
|
|
|
|
// Swap swaps element with i and j index in IDList.
|
|
func (s IDList) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
|
|
|
|
// CalculateSize combines length of all zones in storage group.
|
|
func CalculateSize(sg Provider) (size uint64) {
|
|
zoneList := sg.Zones()
|
|
for i := range zoneList {
|
|
size += zoneList[i].Size
|
|
}
|
|
return
|
|
}
|
|
|
|
// CalculateHash returns homomorphic sum of hashes
|
|
// fromm all zones in storage group.
|
|
func CalculateHash(sg Provider) (hash.Hash, error) {
|
|
var (
|
|
zones = sg.Zones()
|
|
hashes = make([]hash.Hash, len(zones))
|
|
)
|
|
for i := range zones {
|
|
hashes[i] = zones[i].Hash
|
|
}
|
|
return hash.Concat(hashes)
|
|
}
|