frostfs-node/pkg/services/control/server/convert.go

package control

import (
	"fmt"
	"time"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/engine"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
	"github.com/mr-tron/base58"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

func stateToResponse(state *engine.EvacuationState) (*control.GetShardEvacuationStatusResponse, error) {
	shardIDs := make([][]byte, 0, len(state.ShardIDs()))
	for _, shID := range state.ShardIDs() {
		id, err := base58.Decode(shID)
		if err != nil {
			return nil, status.Error(codes.Internal, fmt.Sprintf("invalid shard id format: %s", shID))
		}
		shardIDs = append(shardIDs, id)
	}
	var evacStatus control.GetShardEvacuationStatusResponse_Body_Status
	switch state.ProcessingStatus() {
	case engine.EvacuateProcessStateRunning:
		evacStatus = control.GetShardEvacuationStatusResponse_Body_RUNNING
	case engine.EvacuateProcessStateCompleted:
		evacStatus = control.GetShardEvacuationStatusResponse_Body_COMPLETED
	default:
		evacStatus = control.GetShardEvacuationStatusResponse_Body_EVACUATE_SHARD_STATUS_UNDEFINED
	}
	var startedAt *control.GetShardEvacuationStatusResponse_Body_UnixTimestamp
	if state.StartedAt() != nil {
		startedAt = &control.GetShardEvacuationStatusResponse_Body_UnixTimestamp{
			Value: state.StartedAt().Unix(),
		}
	}
	var duration *control.GetShardEvacuationStatusResponse_Body_Duration
	if state.StartedAt() != nil {
		end := time.Now().UTC()
		if state.FinishedAt() != nil {
			end = *state.FinishedAt()
		}
		duration = &control.GetShardEvacuationStatusResponse_Body_Duration{
			Seconds: int64(end.Sub(*state.StartedAt()).Seconds()),
		}
	}
	return &control.GetShardEvacuationStatusResponse{
		Body: &control.GetShardEvacuationStatusResponse_Body{
			Shard_ID:         shardIDs,
			EvacuatedObjects: state.ObjectsEvacuated(),
			TotalObjects:     state.ObjectsTotal(),
			FailedObjects:    state.ObjectsFailed(),
			Status:           evacStatus,
			StartedAt:        startedAt,
			Duration:         duration,
			ErrorMessage:     state.ErrorMessage(),
			SkippedObjects:   state.ObjectsSkipped(),
			TotalTrees:       state.TreesTotal(),
			EvacuatedTrees:   state.TreesEvacuated(),
			FailedTrees:      state.TreesFailed(),
		},
	}, nil
}