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.
This commit is contained in:
Gregor Schulte
247
pkg/models/automations.go
Normal file
247
pkg/models/automations.go
Normal file
@@ -0,0 +1,247 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
// AutomationSort represents sort options for automations.
|
||||
type AutomationSort string
|
||||
|
||||
const (
|
||||
AutomationSortCreatedDesc AutomationSort = "CREATED_DESC"
|
||||
AutomationSortUpdatedDesc AutomationSort = "UPDATED_DESC"
|
||||
AutomationSortNameAsc AutomationSort = "NAME_ASC"
|
||||
AutomationSortNameDesc AutomationSort = "NAME_DESC"
|
||||
)
|
||||
|
||||
// TriggerPosture represents the posture of an event trigger.
|
||||
type TriggerPosture string
|
||||
|
||||
const (
|
||||
TriggerPostureReactive TriggerPosture = "Reactive"
|
||||
TriggerPostureProactive TriggerPosture = "Proactive"
|
||||
)
|
||||
|
||||
// ActionSource represents the source type for automation actions.
|
||||
type ActionSource string
|
||||
|
||||
const (
|
||||
ActionSourceSelected ActionSource = "selected"
|
||||
ActionSourceInferred ActionSource = "inferred"
|
||||
)
|
||||
|
||||
// ResourceSpecification represents resource labels to match.
|
||||
type ResourceSpecification map[string]string
|
||||
|
||||
// Automation represents a Prefect automation.
|
||||
type Automation struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
Created *time.Time `json:"created"`
|
||||
Updated *time.Time `json:"updated"`
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
Trigger json.RawMessage `json:"trigger"`
|
||||
Actions []json.RawMessage `json:"actions"`
|
||||
ActionsOnTrigger []json.RawMessage `json:"actions_on_trigger,omitempty"`
|
||||
ActionsOnResolve []json.RawMessage `json:"actions_on_resolve,omitempty"`
|
||||
}
|
||||
|
||||
// AutomationCreate represents the request to create an automation.
|
||||
type AutomationCreate struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Enabled *bool `json:"enabled,omitempty"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
Trigger json.RawMessage `json:"trigger"`
|
||||
Actions []json.RawMessage `json:"actions"`
|
||||
ActionsOnTrigger []json.RawMessage `json:"actions_on_trigger,omitempty"`
|
||||
ActionsOnResolve []json.RawMessage `json:"actions_on_resolve,omitempty"`
|
||||
}
|
||||
|
||||
// AutomationUpdate represents the request to fully update an automation.
|
||||
type AutomationUpdate struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Enabled *bool `json:"enabled,omitempty"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
Trigger json.RawMessage `json:"trigger"`
|
||||
Actions []json.RawMessage `json:"actions"`
|
||||
ActionsOnTrigger []json.RawMessage `json:"actions_on_trigger,omitempty"`
|
||||
ActionsOnResolve []json.RawMessage `json:"actions_on_resolve,omitempty"`
|
||||
}
|
||||
|
||||
// AutomationPartialUpdate represents the request to partially update an automation.
|
||||
type AutomationPartialUpdate struct {
|
||||
Enabled *bool `json:"enabled,omitempty"`
|
||||
}
|
||||
|
||||
// AutomationFilter represents filter criteria for querying automations.
|
||||
type AutomationFilter struct {
|
||||
Name *string `json:"name,omitempty"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
Offset int `json:"offset,omitempty"`
|
||||
Limit int `json:"limit,omitempty"`
|
||||
}
|
||||
|
||||
// EventTrigger represents an event-based automation trigger.
|
||||
type EventTrigger struct {
|
||||
Type string `json:"type"`
|
||||
ID uuid.UUID `json:"id"`
|
||||
Match ResourceSpecification `json:"match,omitempty"`
|
||||
MatchRelated json.RawMessage `json:"match_related,omitempty"`
|
||||
After []string `json:"after,omitempty"`
|
||||
Expect []string `json:"expect,omitempty"`
|
||||
ForEach []string `json:"for_each,omitempty"`
|
||||
Posture TriggerPosture `json:"posture"`
|
||||
Threshold int `json:"threshold,omitempty"`
|
||||
Within float64 `json:"within,omitempty"`
|
||||
}
|
||||
|
||||
// CompoundTrigger represents a compound automation trigger.
|
||||
type CompoundTrigger struct {
|
||||
Type string `json:"type"`
|
||||
ID uuid.UUID `json:"id"`
|
||||
Triggers json.RawMessage `json:"triggers"`
|
||||
Within *float64 `json:"within"`
|
||||
Require json.RawMessage `json:"require"`
|
||||
}
|
||||
|
||||
// SequenceTrigger represents a sequence automation trigger.
|
||||
type SequenceTrigger struct {
|
||||
Type string `json:"type"`
|
||||
ID uuid.UUID `json:"id"`
|
||||
Triggers json.RawMessage `json:"triggers"`
|
||||
Within *float64 `json:"within"`
|
||||
}
|
||||
|
||||
// DoNothing represents a no-op automation action.
|
||||
type DoNothing struct {
|
||||
Type string `json:"type"`
|
||||
}
|
||||
|
||||
// RunDeployment represents an action to run a deployment.
|
||||
type RunDeployment struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
|
||||
Parameters map[string]interface{} `json:"parameters,omitempty"`
|
||||
JobVariables map[string]interface{} `json:"job_variables,omitempty"`
|
||||
}
|
||||
|
||||
// PauseDeploymentAction represents an action to pause a deployment.
|
||||
type PauseDeploymentAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
|
||||
}
|
||||
|
||||
// ResumeDeploymentAction represents an action to resume a deployment.
|
||||
type ResumeDeploymentAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
|
||||
}
|
||||
|
||||
// CancelFlowRun represents an action to cancel a flow run.
|
||||
type CancelFlowRun struct {
|
||||
Type string `json:"type"`
|
||||
}
|
||||
|
||||
// SuspendFlowRun represents an action to suspend a flow run.
|
||||
type SuspendFlowRun struct {
|
||||
Type string `json:"type"`
|
||||
}
|
||||
|
||||
// ResumeFlowRunAction represents an action to resume a flow run.
|
||||
type ResumeFlowRunAction struct {
|
||||
Type string `json:"type"`
|
||||
}
|
||||
|
||||
// ChangeFlowRunState represents an action to change a flow run's state.
|
||||
type ChangeFlowRunState struct {
|
||||
Type string `json:"type"`
|
||||
Name *string `json:"name,omitempty"`
|
||||
State StateType `json:"state"`
|
||||
Message *string `json:"message,omitempty"`
|
||||
Force bool `json:"force,omitempty"`
|
||||
}
|
||||
|
||||
// PauseWorkQueueAction represents an action to pause a work queue.
|
||||
type PauseWorkQueueAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
WorkQueueID *uuid.UUID `json:"work_queue_id,omitempty"`
|
||||
}
|
||||
|
||||
// ResumeWorkQueueAction represents an action to resume a work queue.
|
||||
type ResumeWorkQueueAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
WorkQueueID *uuid.UUID `json:"work_queue_id,omitempty"`
|
||||
}
|
||||
|
||||
// PauseWorkPoolAction represents an action to pause a work pool.
|
||||
type PauseWorkPoolAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
WorkPoolID *uuid.UUID `json:"work_pool_id,omitempty"`
|
||||
}
|
||||
|
||||
// ResumeWorkPoolAction represents an action to resume a work pool.
|
||||
type ResumeWorkPoolAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
WorkPoolID *uuid.UUID `json:"work_pool_id,omitempty"`
|
||||
}
|
||||
|
||||
// PauseAutomationAction represents an action to pause an automation.
|
||||
type PauseAutomationAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
AutomationID *uuid.UUID `json:"automation_id,omitempty"`
|
||||
}
|
||||
|
||||
// ResumeAutomationAction represents an action to resume an automation.
|
||||
type ResumeAutomationAction struct {
|
||||
Type string `json:"type"`
|
||||
Source ActionSource `json:"source,omitempty"`
|
||||
AutomationID *uuid.UUID `json:"automation_id,omitempty"`
|
||||
}
|
||||
|
||||
// SendNotification represents an action to send a notification.
|
||||
type SendNotification struct {
|
||||
Type string `json:"type"`
|
||||
BlockDocumentID uuid.UUID `json:"block_document_id"`
|
||||
Subject string `json:"subject,omitempty"`
|
||||
Body string `json:"body"`
|
||||
}
|
||||
|
||||
// CallWebhook represents an action to call a webhook.
|
||||
type CallWebhook struct {
|
||||
Type string `json:"type"`
|
||||
BlockDocumentID uuid.UUID `json:"block_document_id"`
|
||||
Payload string `json:"payload,omitempty"`
|
||||
}
|
||||
|
||||
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
|
||||
func (a *Automation) UnmarshalJSON(data []byte) error {
|
||||
type Alias Automation
|
||||
aux := &struct {
|
||||
Created optTime `json:"created"`
|
||||
Updated optTime `json:"updated"`
|
||||
*Alias
|
||||
}{
|
||||
Alias: (*Alias)(a),
|
||||
}
|
||||
if err := json.Unmarshal(data, aux); err != nil {
|
||||
return err
|
||||
}
|
||||
a.Created = aux.Created.V
|
||||
a.Updated = aux.Updated.V
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user