frostfs-node/pkg/services/object_manager/transformer/fmt.go

package transformer

import (
	"context"
	"crypto/ecdsa"
	"fmt"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/session"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/version"
)

type formatter struct {
	prm *FormatterParams

	obj *object.Object

	sz uint64
}

// FormatterParams groups NewFormatTarget parameters.
type FormatterParams struct {
	Key *ecdsa.PrivateKey

	NextTarget ObjectTarget

	SessionToken *session.Object

	NetworkState netmap.State
}

// NewFormatTarget returns ObjectTarget instance that finalizes object structure
// and writes it to the next target.
//
// Chunks must be written before the WriteHeader call.
//
// Object changes:
// - sets version to current SDK version;
// - sets payload size to the total length of all written chunks;
// - sets session token;
// - sets number of creation epoch;
// - calculates and sets verification fields (ID, Signature).
func NewFormatTarget(p *FormatterParams) ObjectTarget {
	return &formatter{
		prm: p,
	}
}

func (f *formatter) WriteHeader(_ context.Context, obj *object.Object) error {
	f.obj = obj

	return nil
}

func (f *formatter) Write(ctx context.Context, p []byte) (n int, err error) {
	n, err = f.prm.NextTarget.Write(ctx, p)

	f.sz += uint64(n)

	return
}

func (f *formatter) Close(ctx context.Context) (*AccessIdentifiers, error) {
	curEpoch := f.prm.NetworkState.CurrentEpoch()
	ver := version.Current()

	f.obj.SetVersion(&ver)
	f.obj.SetPayloadSize(f.sz)
	f.obj.SetSessionToken(f.prm.SessionToken)
	f.obj.SetCreationEpoch(curEpoch)

	var (
		parID  *oid.ID
		parHdr *object.Object
	)

	if par := f.obj.Parent(); par != nil && par.Signature() == nil {
		rawPar := object.NewFromV2(par.ToV2())

		rawPar.SetSessionToken(f.prm.SessionToken)
		rawPar.SetCreationEpoch(curEpoch)

		if err := object.SetIDWithSignature(*f.prm.Key, rawPar); err != nil {
			return nil, fmt.Errorf("could not finalize parent object: %w", err)
		}

		id, _ := rawPar.ID()
		parID = &id
		parHdr = rawPar

		f.obj.SetParent(parHdr)
	}

	if err := object.SetIDWithSignature(*f.prm.Key, f.obj); err != nil {
		return nil, fmt.Errorf("could not finalize object: %w", err)
	}

	if err := f.prm.NextTarget.WriteHeader(ctx, f.obj); err != nil {
		return nil, fmt.Errorf("could not write header to next target: %w", err)
	}

	if _, err := f.prm.NextTarget.Close(ctx); err != nil {
		return nil, fmt.Errorf("could not close next target: %w", err)
	}

	id, _ := f.obj.ID()

	return new(AccessIdentifiers).
		WithSelfID(id).
		WithParentID(parID).
		WithParent(parHdr), nil
}