forked from TrueCloudLab/xk6-frostfs
33 lines
951 B
Go
33 lines
951 B
Go
|
package s3local
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
||
|
|
)
|
||
|
|
|
||
|
|
// fixedBucketResolver is a static bucket resolver from the provided map.
|
||
|
|
// This is needed to replace the normal resolver for local storage engine clients, since
|
||
|
|
// those should not use DNS or NNS for resolution.
|
||
|
|
type fixedBucketResolver map[string]cid.ID
|
||
|
|
|
||
|
|
func newFixedBucketResolver(bucketMapping map[string]string) (fixedBucketResolver, error) {
|
||
|
|
r := fixedBucketResolver{}
|
||
|
|
for bucket, cidStr := range bucketMapping {
|
||
|
|
var id cid.ID
|
||
|
|
if err := id.DecodeString(cidStr); err != nil {
|
||
|
|
return nil, fmt.Errorf("decoding container id %q: %v", cidStr, err)
|
||
|
|
}
|
||
|
|
r[bucket] = id
|
||
|
|
}
|
||
|
|
return r, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r fixedBucketResolver) Resolve(_ context.Context, bucket string) (cid.ID, error) {
|
||
|
|
if cid, resolved := r[bucket]; resolved {
|
||
|
|
return cid, nil
|
||
|
|
}
|
||
|
|
return cid.ID{}, fmt.Errorf("bucket %s is not mapped to any container", bucket)
|
||
|
|
}
|