forked from TrueCloudLab/frostfs-node
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package blobovnicza
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal/schema/common"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"github.com/mr-tron/base58"
|
|
)
|
|
|
|
var BlobovniczaParser = common.WithFallback(
|
|
common.Any(
|
|
MetaBucketParser,
|
|
BucketParser,
|
|
),
|
|
common.RawParser.ToFallbackParser(),
|
|
)
|
|
|
|
func MetaBucketParser(key, value []byte) (common.SchemaEntry, common.Parser, error) {
|
|
if value != nil {
|
|
return nil, nil, errors.New("not a bucket")
|
|
}
|
|
|
|
if string(key) != "META" {
|
|
return nil, nil, errors.New("invalid bucket name")
|
|
}
|
|
|
|
return &MetaBucket{}, MetaRecordParser, nil
|
|
}
|
|
|
|
func MetaRecordParser(key, value []byte) (common.SchemaEntry, common.Parser, error) {
|
|
var r MetaRecord
|
|
|
|
if len(key) == 0 {
|
|
return nil, nil, errors.New("invalid key")
|
|
}
|
|
|
|
r.label = string(key)
|
|
r.count = binary.LittleEndian.Uint64(value)
|
|
|
|
return &r, nil, nil
|
|
}
|
|
|
|
func BucketParser(key, value []byte) (common.SchemaEntry, common.Parser, error) {
|
|
if value != nil {
|
|
return nil, nil, errors.New("not a bucket")
|
|
}
|
|
|
|
size, n := binary.Varint(key)
|
|
if n <= 0 {
|
|
return nil, nil, errors.New("invalid size")
|
|
}
|
|
|
|
return &Bucket{size: size}, RecordParser, nil
|
|
}
|
|
|
|
func RecordParser(key, value []byte) (common.SchemaEntry, common.Parser, error) {
|
|
parts := strings.Split(string(key), "/")
|
|
|
|
if len(parts) != 2 {
|
|
return nil, nil, errors.New("invalid key, expected address string <CID>/<OID>")
|
|
}
|
|
|
|
cnrRaw, err := base58.Decode(parts[0])
|
|
if err != nil {
|
|
return nil, nil, errors.New("can't decode CID string")
|
|
}
|
|
objRaw, err := base58.Decode(parts[1])
|
|
if err != nil {
|
|
return nil, nil, errors.New("can't decode OID string")
|
|
}
|
|
|
|
cnr := cid.ID{}
|
|
if err := cnr.Decode(cnrRaw); err != nil {
|
|
return nil, nil, fmt.Errorf("can't decode CID: %w", err)
|
|
}
|
|
obj := oid.ID{}
|
|
if err := obj.Decode(objRaw); err != nil {
|
|
return nil, nil, fmt.Errorf("can't decode OID: %w", err)
|
|
}
|
|
|
|
var r Record
|
|
|
|
r.addr.SetContainer(cnr)
|
|
r.addr.SetObject(obj)
|
|
|
|
if err := r.object.Unmarshal(value); err != nil {
|
|
return nil, nil, errors.New("can't unmarshal object")
|
|
}
|
|
|
|
return &r, nil, nil
|
|
}
|