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/concurrency_limits.go
+++ b/pkg/models/concurrency_limits.go
@@ -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
+}