frostfs-node/pkg/services/object/common.go

package object

import (
	"context"

	objectV2 "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/object"
	apistatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status"
)

// NodeState is storage node state processed by Object service.
type NodeState interface {
	// IsMaintenance checks if node is under maintenance. Node MUST NOT serve
	// local object operations. Node MUST respond with apistatus.NodeUnderMaintenance
	// error if IsMaintenance returns true.
	IsMaintenance() bool
}

// Common is an Object API ServiceServer which encapsulates logic spread to all
// object operations.
//
// If underlying NodeState.IsMaintenance returns true, all operations are
// immediately failed with apistatus.NodeUnderMaintenance.
type Common struct {
	state NodeState

	nextHandler ServiceServer
}

// Init initializes the Common instance.
func (x *Common) Init(state NodeState, nextHandler ServiceServer) {
	x.state = state
	x.nextHandler = nextHandler
}

func (x *Common) Get(req *objectV2.GetRequest, stream GetObjectStream) error {
	if x.state.IsMaintenance() {
		return new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Get(req, stream)
}

func (x *Common) Put() (PutObjectStream, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Put()
}

func (x *Common) Patch() (PatchObjectStream, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Patch()
}

func (x *Common) Head(ctx context.Context, req *objectV2.HeadRequest) (*objectV2.HeadResponse, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Head(ctx, req)
}

func (x *Common) Search(req *objectV2.SearchRequest, stream SearchStream) error {
	if x.state.IsMaintenance() {
		return new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Search(req, stream)
}

func (x *Common) Delete(ctx context.Context, req *objectV2.DeleteRequest) (*objectV2.DeleteResponse, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.Delete(ctx, req)
}

func (x *Common) GetRange(req *objectV2.GetRangeRequest, stream GetObjectRangeStream) error {
	if x.state.IsMaintenance() {
		return new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.GetRange(req, stream)
}

func (x *Common) GetRangeHash(ctx context.Context, req *objectV2.GetRangeHashRequest) (*objectV2.GetRangeHashResponse, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.GetRangeHash(ctx, req)
}

func (x *Common) PutSingle(ctx context.Context, req *objectV2.PutSingleRequest) (*objectV2.PutSingleResponse, error) {
	if x.state.IsMaintenance() {
		return nil, new(apistatus.NodeUnderMaintenance)
	}

	return x.nextHandler.PutSingle(ctx, req)
}