forked from TrueCloudLab/frostfs-node
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package meta
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/metaerr"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"github.com/cockroachdb/pebble"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
// GetChildren returns parent -> children map.
|
|
// If an object has no children, then map will contain addr -> empty slice value.
|
|
func (db *DB) GetChildren(ctx context.Context, addresses []oid.Address) (map[oid.Address][]oid.Address, error) {
|
|
var (
|
|
startedAt = time.Now()
|
|
success = false
|
|
)
|
|
defer func() {
|
|
db.metrics.AddMethodDuration("GetChildren", time.Since(startedAt), success)
|
|
}()
|
|
|
|
_, span := tracing.StartSpanFromContext(ctx, "metabase.GetChildren",
|
|
trace.WithAttributes(
|
|
attribute.Int("addr_count", len(addresses)),
|
|
))
|
|
defer span.End()
|
|
|
|
db.modeMtx.RLock()
|
|
defer db.modeMtx.RUnlock()
|
|
|
|
if db.mode.NoMetabase() {
|
|
return nil, ErrDegradedMode
|
|
}
|
|
|
|
result := make(map[oid.Address][]oid.Address, len(addresses))
|
|
err := db.snapshot(func(s *pebble.Snapshot) error {
|
|
for _, addr := range addresses {
|
|
if _, found := result[addr]; found {
|
|
continue
|
|
}
|
|
|
|
result[addr] = []oid.Address{}
|
|
|
|
for {
|
|
keys, err := selectByPrefixBatch(ctx, s, parentKeyLongPrefix(addr.Container(), addr.Object()), batchSize)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, key := range keys {
|
|
resultAddress, err := addressOfTargetFromParentKey(key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
result[addr] = append(result[addr], resultAddress)
|
|
}
|
|
if len(keys) < batchSize {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, metaerr.Wrap(err)
|
|
}
|
|
success = true
|
|
return result, nil
|
|
}
|