forked from TrueCloudLab/frostfs-node
[#199] putsvc: Refactor put object
Resolve containedctx linter for streamer and remote target Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
This commit is contained in:
parent
cecea8053a
commit
27bdddc48f
24 changed files with 171 additions and 125 deletions
52
pkg/services/object_manager/transformer/writer.go
Normal file
52
pkg/services/object_manager/transformer/writer.go
Normal file
|
@ -0,0 +1,52 @@
|
|||
package transformer
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
)
|
||||
|
||||
type writer interface {
|
||||
Write(ctx context.Context, p []byte) (n int, err error)
|
||||
}
|
||||
|
||||
type multiWriter struct {
|
||||
writers []writer
|
||||
}
|
||||
|
||||
func (t *multiWriter) Write(ctx context.Context, p []byte) (n int, err error) {
|
||||
for _, w := range t.writers {
|
||||
n, err = w.Write(ctx, p)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if n != len(p) {
|
||||
err = io.ErrShortWrite
|
||||
return
|
||||
}
|
||||
}
|
||||
return len(p), nil
|
||||
}
|
||||
|
||||
func newMultiWriter(writers ...writer) writer {
|
||||
allWriters := make([]writer, 0, len(writers))
|
||||
for _, w := range writers {
|
||||
if mw, ok := w.(*multiWriter); ok {
|
||||
allWriters = append(allWriters, mw.writers...)
|
||||
} else {
|
||||
allWriters = append(allWriters, w)
|
||||
}
|
||||
}
|
||||
return &multiWriter{allWriters}
|
||||
}
|
||||
|
||||
type writerWrapper struct {
|
||||
Writer io.Writer
|
||||
}
|
||||
|
||||
func (w *writerWrapper) Write(_ context.Context, p []byte) (n int, err error) {
|
||||
return w.Writer.Write(p)
|
||||
}
|
||||
|
||||
func newWriter(w io.Writer) writer {
|
||||
return &writerWrapper{Writer: w}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue