schultesdev/prefect-go
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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.

Browse Source
This commit is contained in:
Gregor Schulte
2026-03-27 14:02:32 +01:00
parent 3aff707116
commit 57531a7d95
36 changed files with 3165 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

127
pkg/models/artifacts.go Normal file
View File

@@ -0,0 +1,127 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// ArtifactSort represents sort options for artifacts.
type ArtifactSort string
const (
ArtifactSortCreatedDesc ArtifactSort = "CREATED_DESC"
ArtifactSortUpdatedDesc ArtifactSort = "UPDATED_DESC"
)
// ArtifactCollectionSort represents sort options for artifact collections.
type ArtifactCollectionSort string
const (
ArtifactCollectionSortCreatedDesc ArtifactCollectionSort = "CREATED_DESC"
ArtifactCollectionSortUpdatedDesc ArtifactCollectionSort = "UPDATED_DESC"
)
// Artifact represents a Prefect artifact.
type Artifact struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Key *string `json:"key"`
Type *string `json:"type"`
Description *string `json:"description"`
Data interface{} `json:"data,omitempty"`
Metadata map[string]string `json:"metadata_,omitempty"`
FlowRunID *uuid.UUID `json:"flow_run_id"`
TaskRunID *uuid.UUID `json:"task_run_id"`
}
// ArtifactCreate represents the request to create an artifact.
type ArtifactCreate struct {
Key *string `json:"key,omitempty"`
Type *string `json:"type,omitempty"`
Description *string `json:"description,omitempty"`
Data interface{} `json:"data,omitempty"`
Metadata map[string]string `json:"metadata_,omitempty"`
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
TaskRunID *uuid.UUID `json:"task_run_id,omitempty"`
}
// ArtifactUpdate represents the request to update an artifact.
type ArtifactUpdate struct {
Data interface{} `json:"data,omitempty"`
Description *string `json:"description,omitempty"`
Metadata map[string]string `json:"metadata_,omitempty"`
}
// ArtifactFilter represents filter criteria for querying artifacts.
type ArtifactFilter struct {
Key *string `json:"key,omitempty"`
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
TaskRunID *uuid.UUID `json:"task_run_id,omitempty"`
Type *string `json:"type,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// ArtifactCollection represents a collection of artifacts with the same key.
type ArtifactCollection struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Key string `json:"key"`
LatestID uuid.UUID `json:"latest_id"`
Type *string `json:"type"`
Description *string `json:"description"`
Data interface{} `json:"data,omitempty"`
Metadata map[string]string `json:"metadata_,omitempty"`
FlowRunID *uuid.UUID `json:"flow_run_id"`
TaskRunID *uuid.UUID `json:"task_run_id"`
}
// ArtifactCollectionFilter represents filter criteria for querying artifact collections.
type ArtifactCollectionFilter struct {
Key *string `json:"key,omitempty"`
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
TaskRunID *uuid.UUID `json:"task_run_id,omitempty"`
Type *string `json:"type,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (a *Artifact) UnmarshalJSON(data []byte) error {
type Alias Artifact
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
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (ac *ArtifactCollection) UnmarshalJSON(data []byte) error {
type Alias ArtifactCollection
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(ac),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
ac.Created = aux.Created.V
ac.Updated = aux.Updated.V
return nil
}

247
pkg/models/automations.go Normal file
View 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
}

176
pkg/models/blocks.go Normal file
View File

@@ -0,0 +1,176 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// BlockDocumentSort represents sort options for block documents.
type BlockDocumentSort string
const (
BlockDocumentSortNameAsc BlockDocumentSort = "NAME_ASC"
BlockDocumentSortNameDesc BlockDocumentSort = "NAME_DESC"
)
// BlockType represents a Prefect block type.
type BlockType struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Slug string `json:"slug"`
LogoURL *string `json:"logo_url"`
DocumentationURL *string `json:"documentation_url"`
Description *string `json:"description"`
CodeExample *string `json:"code_example"`
IsProtected bool `json:"is_protected"`
}
// BlockTypeCreate represents the request to create a block type.
type BlockTypeCreate struct {
Name string `json:"name"`
Slug string `json:"slug"`
LogoURL *string `json:"logo_url,omitempty"`
DocumentationURL *string `json:"documentation_url,omitempty"`
Description *string `json:"description,omitempty"`
CodeExample *string `json:"code_example,omitempty"`
}
// BlockTypeUpdate represents the request to update a block type.
type BlockTypeUpdate struct {
LogoURL *string `json:"logo_url,omitempty"`
DocumentationURL *string `json:"documentation_url,omitempty"`
Description *string `json:"description,omitempty"`
CodeExample *string `json:"code_example,omitempty"`
}
// BlockTypeFilter represents filter criteria for querying block types.
type BlockTypeFilter struct {
Name *string // Filter by name (partial match)
Slugs []string // Filter by slugs
Capabilities []string // Filter by block capabilities
}
// BlockSchema represents a Prefect block schema.
type BlockSchema struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Checksum string `json:"checksum"`
Fields map[string]interface{} `json:"fields"`
BlockTypeID *uuid.UUID `json:"block_type_id"`
BlockType *BlockType `json:"block_type"`
Capabilities []string `json:"capabilities"`
Version string `json:"version"`
}
// BlockSchemaCreate represents the request to create a block schema.
type BlockSchemaCreate struct {
Fields map[string]interface{} `json:"fields,omitempty"`
BlockTypeID uuid.UUID `json:"block_type_id"`
Capabilities []string `json:"capabilities,omitempty"`
Version *string `json:"version,omitempty"`
}
// BlockSchemaFilter represents filter criteria for querying block schemas.
type BlockSchemaFilter struct {
BlockTypeID *uuid.UUID // Filter by block type ID
Capabilities []string // Filter by required capabilities
Version *string // Filter by schema version
}
// BlockDocument represents a Prefect block document.
type BlockDocument struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name *string `json:"name"`
Data map[string]interface{} `json:"data"`
BlockSchemaID uuid.UUID `json:"block_schema_id"`
BlockSchema *BlockSchema `json:"block_schema"`
BlockTypeID uuid.UUID `json:"block_type_id"`
BlockTypeName *string `json:"block_type_name"`
BlockType *BlockType `json:"block_type"`
BlockDocumentReferences map[string]interface{} `json:"block_document_references"`
IsAnonymous bool `json:"is_anonymous"`
}
// BlockDocumentCreate represents the request to create a block document.
type BlockDocumentCreate struct {
Name *string `json:"name,omitempty"`
Data map[string]interface{} `json:"data,omitempty"`
BlockSchemaID uuid.UUID `json:"block_schema_id"`
BlockTypeID uuid.UUID `json:"block_type_id"`
IsAnonymous bool `json:"is_anonymous,omitempty"`
}
// BlockDocumentUpdate represents the request to update a block document.
type BlockDocumentUpdate struct {
BlockSchemaID *uuid.UUID `json:"block_schema_id,omitempty"`
Data map[string]interface{} `json:"data"`
MergeExisting *bool `json:"merge_existing_data,omitempty"`
}
// BlockDocumentFilter represents filter criteria for querying block documents.
type BlockDocumentFilter struct {
BlockTypeID *uuid.UUID // Filter by block type ID
Name *string // Filter by document name
IsAnonymous *bool // Filter by anonymity
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (bt *BlockType) UnmarshalJSON(data []byte) error {
type Alias BlockType
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(bt),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
bt.Created = aux.Created.V
bt.Updated = aux.Updated.V
return nil
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (bs *BlockSchema) UnmarshalJSON(data []byte) error {
type Alias BlockSchema
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(bs),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
bs.Created = aux.Created.V
bs.Updated = aux.Updated.V
return nil
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (bd *BlockDocument) UnmarshalJSON(data []byte) error {
type Alias BlockDocument
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(bd),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
bd.Created = aux.Created.V
bd.Updated = aux.Updated.V
return nil
}

137
pkg/models/concurrency_limits.go Normal file
View File

@@ -0,0 +1,137 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// ConcurrencyLimit represents a Prefect v1 concurrency limit.
type ConcurrencyLimit struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Tag string `json:"tag"`
ConcurrencyLimit int `json:"concurrency_limit"`
ActiveSlots []string `json:"active_slots,omitempty"`
}
// ConcurrencyLimitCreate represents the request to create a v1 concurrency limit.
type ConcurrencyLimitCreate struct {
Tag string `json:"tag"`
ConcurrencyLimit int `json:"concurrency_limit"`
}
// ConcurrencyLimitV2 represents a Prefect v2 concurrency limit.
type ConcurrencyLimitV2 struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Active bool `json:"active"`
Name string `json:"name"`
Limit int `json:"limit"`
ActiveSlots int `json:"active_slots"`
DeniedSlots int `json:"denied_slots"`
SlotDecayPerSecond float64 `json:"slot_decay_per_second"`
AvgSlotOccupancySeconds float64 `json:"avg_slot_occupancy_seconds"`
}
// ConcurrencyLimitV2Create represents the request to create a v2 concurrency limit.
type ConcurrencyLimitV2Create struct {
Active *bool `json:"active,omitempty"`
Name string `json:"name"`
Limit int `json:"limit"`
ActiveSlots *int `json:"active_slots,omitempty"`
DeniedSlots *int `json:"denied_slots,omitempty"`
SlotDecayPerSecond *float64 `json:"slot_decay_per_second,omitempty"`
}
// ConcurrencyLimitV2Update represents the request to update a v2 concurrency limit.
type ConcurrencyLimitV2Update struct {
Active *bool `json:"active,omitempty"`
Name *string `json:"name,omitempty"`
Limit *int `json:"limit,omitempty"`
ActiveSlots *int `json:"active_slots,omitempty"`
DeniedSlots *int `json:"denied_slots,omitempty"`
SlotDecayPerSecond *float64 `json:"slot_decay_per_second,omitempty"`
}
// GlobalConcurrencyLimitResponse represents the response for global concurrency limits.
type GlobalConcurrencyLimitResponse struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Active bool `json:"active"`
Name string `json:"name"`
Limit int `json:"limit"`
ActiveSlots int `json:"active_slots"`
SlotDecayPerSecond float64 `json:"slot_decay_per_second"`
}
// MinimalConcurrencyLimitResponse represents a minimal concurrency limit response.
type MinimalConcurrencyLimitResponse struct {
ID uuid.UUID `json:"id"`
Name string `json:"name"`
Limit int `json:"limit"`
}
// ConcurrencyLimitWithLeaseResponse represents a concurrency limit response with lease info.
type ConcurrencyLimitWithLeaseResponse struct {
LeaseID uuid.UUID `json:"lease_id"`
Limits []MinimalConcurrencyLimitResponse `json:"limits"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (cl *ConcurrencyLimit) UnmarshalJSON(data []byte) error {
type Alias ConcurrencyLimit
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(cl),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
cl.Created = aux.Created.V
cl.Updated = aux.Updated.V
return nil
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (cl *ConcurrencyLimitV2) UnmarshalJSON(data []byte) error {
type Alias ConcurrencyLimitV2
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(cl),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
cl.Created = aux.Created.V
cl.Updated = aux.Updated.V
return nil
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (g *GlobalConcurrencyLimitResponse) UnmarshalJSON(data []byte) error {
type Alias GlobalConcurrencyLimitResponse
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(g),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
g.Created = aux.Created.V
g.Updated = aux.Updated.V
return nil
}

102
pkg/models/deployments.go Normal file
View File

@@ -0,0 +1,102 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// DeploymentStatus represents the status of a deployment.
type DeploymentStatus string
const (
DeploymentStatusReady DeploymentStatus = "READY"
DeploymentStatusNotReady DeploymentStatus = "NOT_READY"
)
// CollisionStrategy represents the strategy for handling concurrent flow runs.
type CollisionStrategy string
const (
CollisionStrategyEnqueue CollisionStrategy = "ENQUEUE"
CollisionStrategyCancelNew CollisionStrategy = "CANCEL_NEW"
)
// ConcurrencyOptions represents concurrency configuration for a deployment.
type ConcurrencyOptions struct {
CollisionStrategy CollisionStrategy `json:"collision_strategy"`
GracePeriodSeconds *int `json:"grace_period_seconds,omitempty"`
}
// Deployment represents a Prefect deployment.
type Deployment struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
FlowID uuid.UUID `json:"flow_id"`
Version *string `json:"version"`
Description *string `json:"description"`
Paused bool `json:"paused"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Tags []string `json:"tags,omitempty"`
Labels map[string]interface{} `json:"labels,omitempty"`
WorkQueueName *string `json:"work_queue_name"`
WorkPoolName *string `json:"work_pool_name"`
Path *string `json:"path"`
Entrypoint *string `json:"entrypoint"`
Status *DeploymentStatus `json:"status"`
EnforceParameterSchema bool `json:"enforce_parameter_schema"`
}
// DeploymentCreate represents the request to create a deployment.
type DeploymentCreate struct {
Name string `json:"name"`
FlowID uuid.UUID `json:"flow_id"`
Paused bool `json:"paused,omitempty"`
Description *string `json:"description,omitempty"`
Version *string `json:"version,omitempty"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Tags []string `json:"tags,omitempty"`
Labels map[string]interface{} `json:"labels,omitempty"`
WorkPoolName *string `json:"work_pool_name,omitempty"`
WorkQueueName *string `json:"work_queue_name,omitempty"`
Path *string `json:"path,omitempty"`
Entrypoint *string `json:"entrypoint,omitempty"`
EnforceParameterSchema bool `json:"enforce_parameter_schema,omitempty"`
}
// DeploymentUpdate represents the request to update a deployment.
type DeploymentUpdate struct {
Paused *bool `json:"paused,omitempty"`
Description *string `json:"description,omitempty"`
}
// DeploymentFilter represents filter criteria for querying deployments.
type DeploymentFilter struct {
FlowID *uuid.UUID `json:"flow_id,omitempty"`
Name *string `json:"name,omitempty"`
Tags []string `json:"tags,omitempty"`
Paused *bool `json:"paused,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (d *Deployment) UnmarshalJSON(data []byte) error {
type Alias Deployment
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(d),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
d.Created = aux.Created.V
d.Updated = aux.Updated.V
return nil
}

58
pkg/models/events.go Normal file
View File

@@ -0,0 +1,58 @@
package models
import (
"time"
"github.com/google/uuid"
)
// EventOrder represents sort order for events.
type EventOrder string
const (
EventOrderAsc EventOrder = "ASC"
EventOrderDesc EventOrder = "DESC"
)
// Resource represents an observable business object.
type Resource map[string]string
// RelatedResource represents a resource with a specific role in an event.
type RelatedResource map[string]string
// Event represents a Prefect event.
type Event struct {
Occurred time.Time `json:"occurred"`
Event string `json:"event"`
Resource Resource `json:"resource"`
Related []RelatedResource `json:"related,omitempty"`
ID uuid.UUID `json:"id"`
Payload map[string]interface{} `json:"payload,omitempty"`
Received *time.Time `json:"received,omitempty"`
}
// EventFilter represents filter criteria for querying events.
type EventFilter struct {
Occurred interface{} `json:"occurred,omitempty"`
Event interface{} `json:"event,omitempty"`
Resource interface{} `json:"resource,omitempty"`
Related interface{} `json:"related,omitempty"`
ID interface{} `json:"id,omitempty"`
Order EventOrder `json:"order,omitempty"`
}
// EventPage represents a page of events.
type EventPage struct {
Events []Event `json:"events"`
Total int `json:"total"`
NextPage *string `json:"next_page"`
}
// EventCount represents a count of events for a given filter value.
type EventCount struct {
Value string `json:"value"`
Label string `json:"label"`
Count int `json:"count"`
StartTime time.Time `json:"start_time"`
EndTime time.Time `json:"end_time"`
}

110
pkg/models/flow_runs.go Normal file
View File

@@ -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
}

57
pkg/models/flows.go Normal file
View File

@@ -0,0 +1,57 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// Flow represents a Prefect flow.
type Flow struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Tags []string `json:"tags,omitempty"`
Labels map[string]interface{} `json:"labels,omitempty"`
}
// FlowCreate represents the request to create a flow.
type FlowCreate struct {
Name string `json:"name"`
Tags []string `json:"tags,omitempty"`
Labels map[string]interface{} `json:"labels,omitempty"`
}
// FlowUpdate represents the request to update a flow.
type FlowUpdate struct {
Tags *[]string `json:"tags,omitempty"`
Labels *map[string]interface{} `json:"labels,omitempty"`
}
// FlowFilter represents filter criteria for querying flows.
type FlowFilter struct {
Name *string `json:"name,omitempty"`
Tags []string `json:"tags,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (f *Flow) UnmarshalJSON(data []byte) error {
type Alias Flow
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(f),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
f.Created = aux.Created.V
f.Updated = aux.Updated.V
return nil
}

49
pkg/models/logs.go Normal file
View File

@@ -0,0 +1,49 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// Log represents a Prefect log entry.
type Log struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Level int `json:"level"`
Message string `json:"message"`
Timestamp time.Time `json:"timestamp"`
FlowRunID *uuid.UUID `json:"flow_run_id"`
TaskRunID *uuid.UUID `json:"task_run_id"`
}
// LogCreate represents the request to create a log.
type LogCreate struct {
Name string `json:"name"`
Level int `json:"level"`
Message string `json:"message"`
Timestamp time.Time `json:"timestamp"`
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
TaskRunID *uuid.UUID `json:"task_run_id,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (l *Log) UnmarshalJSON(data []byte) error {
type Alias Log
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(l),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
l.Created = aux.Created.V
l.Updated = aux.Updated.V
return nil
}

50
pkg/models/saved_searches.go Normal file
View File

@@ -0,0 +1,50 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// SavedSearch represents a Prefect saved search.
type SavedSearch struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Filters []SavedSearchFilter `json:"filters"`
}
// SavedSearchCreate represents the request to create a saved search.
type SavedSearchCreate struct {
Name string `json:"name"`
Filters []SavedSearchFilter `json:"filters,omitempty"`
}
// SavedSearchFilter represents a filter definition within a saved search.
type SavedSearchFilter struct {
Object string `json:"object"`
Property string `json:"property"`
Type string `json:"type"`
Operation string `json:"operation"`
Value interface{} `json:"value"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (ss *SavedSearch) UnmarshalJSON(data []byte) error {
type Alias SavedSearch
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(ss),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
ss.Created = aux.Created.V
ss.Updated = aux.Updated.V
return nil
}

77
pkg/models/schedules.go Normal file
View File

@@ -0,0 +1,77 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// DeploymentSchedule represents a schedule for a deployment.
type DeploymentSchedule struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
DeploymentID *uuid.UUID `json:"deployment_id"`
Schedule json.RawMessage `json:"schedule"`
Active bool `json:"active"`
MaxScheduledRuns *int `json:"max_scheduled_runs,omitempty"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Slug *string `json:"slug,omitempty"`
}
// DeploymentScheduleCreate represents the request to create a deployment schedule.
type DeploymentScheduleCreate struct {
Active *bool `json:"active,omitempty"`
Schedule json.RawMessage `json:"schedule"`
MaxScheduledRuns *int `json:"max_scheduled_runs,omitempty"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Slug *string `json:"slug,omitempty"`
}
// DeploymentScheduleUpdate represents the request to update a deployment schedule.
type DeploymentScheduleUpdate struct {
Active *bool `json:"active,omitempty"`
Schedule json.RawMessage `json:"schedule,omitempty"`
MaxScheduledRuns *int `json:"max_scheduled_runs,omitempty"`
Parameters map[string]interface{} `json:"parameters,omitempty"`
Slug *string `json:"slug,omitempty"`
}
// IntervalSchedule represents an interval-based schedule.
type IntervalSchedule struct {
Interval float64 `json:"interval"`
AnchorDate *string `json:"anchor_date,omitempty"`
Timezone *string `json:"timezone,omitempty"`
}
// CronSchedule represents a cron-based schedule.
type CronSchedule struct {
Cron string `json:"cron"`
Timezone *string `json:"timezone,omitempty"`
DayOr *bool `json:"day_or,omitempty"`
}
// RRuleSchedule represents an RRule-based schedule.
type RRuleSchedule struct {
RRule string `json:"rrule"`
Timezone *string `json:"timezone,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (ds *DeploymentSchedule) UnmarshalJSON(data []byte) error {
type Alias DeploymentSchedule
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(ds),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
ds.Created = aux.Created.V
ds.Updated = aux.Updated.V
return nil
}

55
pkg/models/states.go Normal file
View File

@@ -0,0 +1,55 @@
package models
import (
"time"
"github.com/google/uuid"
)
// StateType represents the type of a flow or task run state.
type StateType string
const (
StateTypePending StateType = "PENDING"
StateTypeRunning StateType = "RUNNING"
StateTypeCompleted StateType = "COMPLETED"
StateTypeFailed StateType = "FAILED"
StateTypeCancelled StateType = "CANCELLED"
StateTypeCrashed StateType = "CRASHED"
StateTypePaused StateType = "PAUSED"
StateTypeScheduled StateType = "SCHEDULED"
StateTypeCancelling StateType = "CANCELLING"
)
// State represents the state of a flow or task run.
type State struct {
ID uuid.UUID `json:"id"`
Type StateType `json:"type"`
Name *string `json:"name"`
Timestamp time.Time `json:"timestamp"`
Message *string `json:"message"`
Data interface{} `json:"data,omitempty"`
StateDetails map[string]interface{} `json:"state_details,omitempty"`
}
// StateCreate represents the request to create a state.
type StateCreate struct {
Type StateType `json:"type"`
Name *string `json:"name,omitempty"`
Message *string `json:"message,omitempty"`
Data interface{} `json:"data,omitempty"`
StateDetails map[string]interface{} `json:"state_details,omitempty"`
}
// StateDetails represents detailed information about a state.
type StateDetails struct {
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
TaskRunID *uuid.UUID `json:"task_run_id,omitempty"`
ChildFlowRunID *uuid.UUID `json:"child_flow_run_id,omitempty"`
ScheduledTime *time.Time `json:"scheduled_time,omitempty"`
CacheKey *string `json:"cache_key,omitempty"`
CacheExpiration *time.Time `json:"cache_expiration,omitempty"`
Deferred *bool `json:"deferred,omitempty"`
UntrackableResult bool `json:"untrackable_result,omitempty"`
PauseTimeout *time.Time `json:"pause_timeout,omitempty"`
}

83
pkg/models/task_runs.go Normal file
View File

@@ -0,0 +1,83 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// TaskRun represents a Prefect task run.
type TaskRun struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
FlowRunID uuid.UUID `json:"flow_run_id"`
TaskKey string `json:"task_key"`
DynamicKey string `json:"dynamic_key"`
CacheKey *string `json:"cache_key"`
StartTime *time.Time `json:"start_time"`
EndTime *time.Time `json:"end_time"`
TotalRunTime float64 `json:"total_run_time"`
Status *StateType `json:"status"`
StateID *uuid.UUID `json:"state_id"`
Tags []string `json:"tags,omitempty"`
State *State `json:"state"`
}
// TaskRunCreate represents the request to create a task run.
type TaskRunCreate struct {
Name string `json:"name"`
FlowRunID uuid.UUID `json:"flow_run_id"`
TaskKey string `json:"task_key"`
DynamicKey string `json:"dynamic_key"`
CacheKey *string `json:"cache_key,omitempty"`
State *StateCreate `json:"state,omitempty"`
Tags []string `json:"tags,omitempty"`
}
// TaskRunUpdate represents the request to update a task run.
type TaskRunUpdate struct {
Name *string `json:"name,omitempty"`
}
// TaskRunFilter represents filter criteria for querying task runs.
type TaskRunFilter struct {
FlowRunID *uuid.UUID `json:"flow_run_id,omitempty"`
StateType *StateType `json:"state_type,omitempty"`
Tags []string `json:"tags,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// TaskRunPolicy represents a task run's retry policy.
type TaskRunPolicy struct {
MaxRetries int `json:"max_retries,omitempty"`
RetryDelaySeconds float64 `json:"retry_delay_seconds,omitempty"`
Retries *int `json:"retries,omitempty"`
RetryDelay interface{} `json:"retry_delay,omitempty"`
RetryJitterFactor *float64 `json:"retry_jitter_factor,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (tr *TaskRun) UnmarshalJSON(data []byte) error {
type Alias TaskRun
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
StartTime optTime `json:"start_time"`
EndTime optTime `json:"end_time"`
*Alias
}{
Alias: (*Alias)(tr),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
tr.Created = aux.Created.V
tr.Updated = aux.Updated.V
tr.StartTime = aux.StartTime.V
tr.EndTime = aux.EndTime.V
return nil
}

15
pkg/models/task_workers.go Normal file
View File

@@ -0,0 +1,15 @@
package models
import "time"
// TaskWorkerFilter represents filter criteria for querying task workers.
type TaskWorkerFilter struct {
TaskKeys []string `json:"task_keys"`
}
// TaskWorkerResponse represents a task worker response.
type TaskWorkerResponse struct {
Identifier string `json:"identifier"`
TaskKeys []string `json:"task_keys"`
Timestamp time.Time `json:"timestamp"`
}

45
pkg/models/time.go Normal file
View File

@@ -0,0 +1,45 @@
package models
import (
"encoding/json"
"time"
)
// parseOptionalTime parses a time string, returning nil for empty strings.
// Supports RFC3339Nano and RFC3339 formats used by the Prefect API.
func parseOptionalTime(s string) (*time.Time, error) {
if s == "" {
return nil, nil
}
t, err := time.Parse(time.RFC3339Nano, s)
if err != nil {
t, err = time.Parse(time.RFC3339, s)
if err != nil {
return nil, err
}
}
return &t, nil
}
// optTime handles JSON unmarshaling of nullable time fields that may be
// represented as empty strings or JSON null by the Prefect API.
type optTime struct {
V *time.Time
}
func (o *optTime) UnmarshalJSON(data []byte) error {
if string(data) == "null" {
o.V = nil
return nil
}
var s string
if err := json.Unmarshal(data, &s); err != nil {
return err
}
v, err := parseOptionalTime(s)
if err != nil {
return err
}
o.V = v
return nil
}

57
pkg/models/variables.go Normal file
View File

@@ -0,0 +1,57 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// Variable represents a Prefect variable.
type Variable struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Value string `json:"value"`
Tags []string `json:"tags,omitempty"`
}
// VariableCreate represents the request to create a variable.
type VariableCreate struct {
Name string `json:"name"`
Value string `json:"value"`
Tags []string `json:"tags,omitempty"`
}
// VariableUpdate represents the request to update a variable.
type VariableUpdate struct {
Value *string `json:"value,omitempty"`
Tags []string `json:"tags,omitempty"`
}
// VariableFilter represents filter criteria for querying variables.
type VariableFilter struct {
Name *string `json:"name,omitempty"`
Tags []string `json:"tags,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (v *Variable) UnmarshalJSON(data []byte) error {
type Alias Variable
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(v),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
v.Created = aux.Created.V
v.Updated = aux.Updated.V
return nil
}

72
pkg/models/work_pools.go Normal file
View File

@@ -0,0 +1,72 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// WorkPoolStatus represents the status of a work pool.
type WorkPoolStatus string
const (
WorkPoolStatusReady WorkPoolStatus = "READY"
WorkPoolStatusNotReady WorkPoolStatus = "NOT_READY"
WorkPoolStatusPaused WorkPoolStatus = "PAUSED"
)
// WorkPool represents a Prefect work pool.
type WorkPool struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Description *string `json:"description"`
Type string `json:"type"`
IsPaused bool `json:"is_paused"`
}
// WorkPoolCreate represents the request to create a work pool.
type WorkPoolCreate struct {
Name string `json:"name"`
Description *string `json:"description,omitempty"`
Type string `json:"type"`
BaseJobTemplate map[string]interface{} `json:"base_job_template,omitempty"`
IsPaused bool `json:"is_paused,omitempty"`
ConcurrencyLimit *int `json:"concurrency_limit,omitempty"`
}
// WorkPoolUpdate represents the request to update a work pool.
type WorkPoolUpdate struct {
Description *string `json:"description,omitempty"`
IsPaused *bool `json:"is_paused,omitempty"`
BaseJobTemplate map[string]interface{} `json:"base_job_template,omitempty"`
ConcurrencyLimit *int `json:"concurrency_limit,omitempty"`
}
// WorkPoolFilter represents filter criteria for querying work pools.
type WorkPoolFilter struct {
Name *string `json:"name,omitempty"`
Type *string `json:"type,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (wp *WorkPool) UnmarshalJSON(data []byte) error {
type Alias WorkPool
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(wp),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
wp.Created = aux.Created.V
wp.Updated = aux.Updated.V
return nil
}

62
pkg/models/work_queues.go Normal file
View File

@@ -0,0 +1,62 @@
package models
import (
"encoding/json"
"time"
"github.com/google/uuid"
)
// WorkQueue represents a Prefect work queue.
type WorkQueue struct {
ID uuid.UUID `json:"id"`
Created *time.Time `json:"created"`
Updated *time.Time `json:"updated"`
Name string `json:"name"`
Description *string `json:"description"`
IsPaused bool `json:"is_paused"`
Priority int `json:"priority"`
}
// WorkQueueCreate represents the request to create a work queue.
type WorkQueueCreate struct {
Name string `json:"name"`
Description string `json:"description,omitempty"`
IsPaused bool `json:"is_paused,omitempty"`
ConcurrencyLimit *int `json:"concurrency_limit,omitempty"`
Priority *int `json:"priority,omitempty"`
}
// WorkQueueUpdate represents the request to update a work queue.
type WorkQueueUpdate struct {
Name *string `json:"name,omitempty"`
Description *string `json:"description,omitempty"`
IsPaused *bool `json:"is_paused,omitempty"`
ConcurrencyLimit *int `json:"concurrency_limit,omitempty"`
Priority *int `json:"priority,omitempty"`
}
// WorkQueueFilter represents filter criteria for querying work queues.
type WorkQueueFilter struct {
Name *string `json:"name,omitempty"`
Offset int `json:"offset,omitempty"`
Limit int `json:"limit,omitempty"`
}
// UnmarshalJSON implements custom JSON unmarshaling for time fields.
func (wq *WorkQueue) UnmarshalJSON(data []byte) error {
type Alias WorkQueue
aux := &struct {
Created optTime `json:"created"`
Updated optTime `json:"updated"`
*Alias
}{
Alias: (*Alias)(wq),
}
if err := json.Unmarshal(data, aux); err != nil {
return err
}
wq.Created = aux.Created.V
wq.Updated = aux.Updated.V
return nil
}