node_execution_repo.go

package interfaces

import (
	"context"

	"github.com/lyft/flyteadmin/pkg/repositories/models"
	"github.com/lyft/flyteidl/gen/pb-go/flyteidl/core"
)

// Defines the interface for interacting with node execution models.
type NodeExecutionRepoInterface interface {
	// Inserts a new node execution model and the first event that triggers it into the database store.
	Create(ctx context.Context, event *models.NodeExecutionEvent, execution *models.NodeExecution) error
	// Updates an existing node execution in the database store with all non-empty fields in the input.
	// This execution and event correspond to entire graph (workflow) executions.
	Update(ctx context.Context, event *models.NodeExecutionEvent, execution *models.NodeExecution) error
	// Returns a matching execution if it exists.
	Get(ctx context.Context, input GetNodeExecutionInput) (models.NodeExecution, error)
	// Returns node executions matching query parameters. A limit must be provided for the results page size.
	List(ctx context.Context, input ListResourceInput) (NodeExecutionCollectionOutput, error)
	// Return node execution events matching query parameters. A limit must be provided for the results page size.
	ListEvents(ctx context.Context, input ListResourceInput) (NodeExecutionEventCollectionOutput, error)
}

type GetNodeExecutionInput struct {
	NodeExecutionIdentifier core.NodeExecutionIdentifier
}

// Response format for a query on node executions.
type NodeExecutionCollectionOutput struct {
	NodeExecutions []models.NodeExecution
}

// Response format for a query on node execution events.
type NodeExecutionEventCollectionOutput struct {
	NodeExecutionEvents []models.NodeExecutionEvent
}