111 lines
4.7 KiB
Go
111 lines
4.7 KiB
Go
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
|
|
}
|