forked from TrueCloudLab/frostfs-node
4415f8dc5b
Implement functions to wrap Writer or Iterator. The resulting wrapper provides WriterProvider or IteratorProvider interface respectively. Such a wrapper can be used as a single storage instance provider regardless of context. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
38 lines
675 B
Go
38 lines
675 B
Go
package loadcontroller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/container"
|
|
)
|
|
|
|
func usedSpaceFilterEpochEQ(epoch uint64) UsedSpaceFilter {
|
|
return func(a container.UsedSpaceAnnouncement) bool {
|
|
return a.Epoch() == epoch
|
|
}
|
|
}
|
|
|
|
type storageWrapper struct {
|
|
w Writer
|
|
i Iterator
|
|
}
|
|
|
|
func (s storageWrapper) InitIterator(context.Context) (Iterator, error) {
|
|
return s.i, nil
|
|
}
|
|
|
|
func (s storageWrapper) InitWriter(context.Context) (Writer, error) {
|
|
return s.w, nil
|
|
}
|
|
|
|
func SimpleIteratorProvider(i Iterator) IteratorProvider {
|
|
return &storageWrapper{
|
|
i: i,
|
|
}
|
|
}
|
|
|
|
func SimpleWriterProvider(w Writer) WriterProvider {
|
|
return &storageWrapper{
|
|
w: w,
|
|
}
|
|
}
|