Füge Modelle für Flows, FlowRuns, TaskRuns, WorkPools, WorkQueues, Deployments, Variablen, FlowRunStates, Logs, und Blocks samt zugehöriger Unmarshal-Logik und Zeitfeld-Unterstützung hinzu; ergänze Tests für die FlowRunStates-Service-Methoden.

2026-03-27 14:02:32 +01:00
parent 3aff707116
commit 57531a7d95
36 changed files with 3165 additions and 0 deletions
--- a/pkg/models/flow_runs.go
+++ b/pkg/models/flow_runs.go
@@ -0,0 +1,110 @@
+package models
+
+import (
+	"encoding/json"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+// FlowRun represents a Prefect flow run.
+type FlowRun struct {
+	ID                     uuid.UUID              `json:"id"`
+	Created                *time.Time             `json:"created"`
+	Updated                *time.Time             `json:"updated"`
+	FlowID                 uuid.UUID              `json:"flow_id"`
+	Name                   string                 `json:"name"`
+	StateID                *uuid.UUID             `json:"state_id"`
+	DeploymentID           *uuid.UUID             `json:"deployment_id"`
+	WorkQueueID            *uuid.UUID             `json:"work_queue_id"`
+	WorkQueueName          *string                `json:"work_queue_name"`
+	FlowVersion            *string                `json:"flow_version"`
+	Parameters             map[string]interface{} `json:"parameters,omitempty"`
+	IdempotencyKey         *string                `json:"idempotency_key"`
+	Context                map[string]interface{} `json:"context,omitempty"`
+	Tags                   []string               `json:"tags,omitempty"`
+	Labels                 map[string]interface{} `json:"labels,omitempty"`
+	ParentTaskRunID        *uuid.UUID             `json:"parent_task_run_id"`
+	StateType              *StateType             `json:"state_type"`
+	StateName              *string                `json:"state_name"`
+	RunCount               int                    `json:"run_count"`
+	ExpectedStartTime      *time.Time             `json:"expected_start_time"`
+	NextScheduledStartTime *time.Time             `json:"next_scheduled_start_time"`
+	StartTime              *time.Time             `json:"start_time"`
+	EndTime                *time.Time             `json:"end_time"`
+	TotalRunTime           float64                `json:"total_run_time"`
+	State                  *State                 `json:"state"`
+}
+
+// FlowRunCreate represents the request to create a flow run.
+type FlowRunCreate struct {
+	FlowID         uuid.UUID              `json:"flow_id"`
+	Name           string                 `json:"name,omitempty"`
+	State          *StateCreate           `json:"state,omitempty"`
+	Parameters     map[string]interface{} `json:"parameters,omitempty"`
+	Context        map[string]interface{} `json:"context,omitempty"`
+	Tags           []string               `json:"tags,omitempty"`
+	Labels         map[string]interface{} `json:"labels,omitempty"`
+	IdempotencyKey *string                `json:"idempotency_key,omitempty"`
+	WorkPoolName   *string                `json:"work_pool_name,omitempty"`
+	WorkQueueName  *string                `json:"work_queue_name,omitempty"`
+	DeploymentID   *uuid.UUID             `json:"deployment_id,omitempty"`
+}
+
+// FlowRunUpdate represents the request to update a flow run.
+type FlowRunUpdate struct {
+	Name *string `json:"name,omitempty"`
+}
+
+// FlowRunFilter represents filter criteria for querying flow runs.
+type FlowRunFilter struct {
+	FlowID       *uuid.UUID `json:"flow_id,omitempty"`
+	DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
+	StateType    *StateType `json:"state_type,omitempty"`
+	Tags         []string   `json:"tags,omitempty"`
+	Offset       int        `json:"offset,omitempty"`
+	Limit        int        `json:"limit,omitempty"`
+}
+
+// FlowRunPolicy represents a flow run's retry policy.
+type FlowRunPolicy struct {
+	MaxRetries        int      `json:"max_retries,omitempty"`
+	RetryDelaySeconds float64  `json:"retry_delay_seconds,omitempty"`
+	Retries           *int     `json:"retries,omitempty"`
+	RetryDelay        *int     `json:"retry_delay,omitempty"`
+	PauseKeys         []string `json:"pause_keys,omitempty"`
+	Resuming          *bool    `json:"resuming,omitempty"`
+}
+
+// CreatedBy represents information about who created an object.
+type CreatedBy struct {
+	ID           *uuid.UUID `json:"id,omitempty"`
+	Type         *string    `json:"type,omitempty"`
+	DisplayValue *string    `json:"display_value,omitempty"`
+}
+
+// UnmarshalJSON implements custom JSON unmarshaling for time fields.
+func (fr *FlowRun) UnmarshalJSON(data []byte) error {
+	type Alias FlowRun
+	aux := &struct {
+		Created                optTime `json:"created"`
+		Updated                optTime `json:"updated"`
+		ExpectedStartTime      optTime `json:"expected_start_time"`
+		NextScheduledStartTime optTime `json:"next_scheduled_start_time"`
+		StartTime              optTime `json:"start_time"`
+		EndTime                optTime `json:"end_time"`
+		*Alias
+	}{
+		Alias: (*Alias)(fr),
+	}
+	if err := json.Unmarshal(data, aux); err != nil {
+		return err
+	}
+	fr.Created = aux.Created.V
+	fr.Updated = aux.Updated.V
+	fr.ExpectedStartTime = aux.ExpectedStartTime.V
+	fr.NextScheduledStartTime = aux.NextScheduledStartTime.V
+	fr.StartTime = aux.StartTime.V
+	fr.EndTime = aux.EndTime.V
+	return nil
+}