frostfs-node/pkg/services/object/get/writer.go

package getsvc

import (
	"context"
	"io"

	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
)

// ChunkWriter is an interface of target component
// to write payload chunk.
type ChunkWriter interface {
	WriteChunk(context.Context, []byte) error
}

// HeaderWriter is an interface of target component
// to write object header.
type HeaderWriter interface {
	WriteHeader(context.Context, *objectSDK.Object) error
}

// ObjectWriter is an interface of target component to write object.
type ObjectWriter interface {
	HeaderWriter
	ChunkWriter
}

type SimpleObjectWriter struct {
	obj *objectSDK.Object

	pld []byte
}

type partWriter struct {
	ObjectWriter

	headWriter HeaderWriter

	chunkWriter ChunkWriter
}

type hasherWrapper struct {
	hash io.Writer
}

func NewSimpleObjectWriter() *SimpleObjectWriter {
	return &SimpleObjectWriter{
		obj: objectSDK.New(),
	}
}

func (s *SimpleObjectWriter) SetPayloadBuffer(buffer []byte) {
	if buffer != nil {
		s.pld = buffer[:0]
	}
}

func (s *SimpleObjectWriter) WriteHeader(_ context.Context, obj *objectSDK.Object) error {
	s.obj = obj

	if s.pld == nil {
		s.pld = make([]byte, 0, obj.PayloadSize())
	}

	return nil
}

func (s *SimpleObjectWriter) WriteChunk(_ context.Context, p []byte) error {
	s.pld = append(s.pld, p...)
	return nil
}

func (s *SimpleObjectWriter) Object() *objectSDK.Object {
	if len(s.pld) > 0 {
		s.obj.SetPayload(s.pld)
	}

	return s.obj
}

func (w *partWriter) WriteChunk(ctx context.Context, p []byte) error {
	return w.chunkWriter.WriteChunk(ctx, p)
}

func (w *partWriter) WriteHeader(ctx context.Context, o *objectSDK.Object) error {
	return w.headWriter.WriteHeader(ctx, o)
}

func (h *hasherWrapper) WriteChunk(_ context.Context, p []byte) error {
	_, err := h.hash.Write(p)
	return err
}