scheduler

type Alert struct {
	On       EventCategory
	Channels []string
	Config   map[string]string
	Severity string
	Team     string
}

type AlertAttrs struct {
	Owner         string
	JobURN        string
	Title         string
	SchedulerHost string
	Status        EventStatus
	JobEvent      *Event

	JobWithDetails *JobWithDetails

	AlertManager AlertManagerConfig
}

type AlertManagerConfig struct {
	Endpoint string
}

type BackfilConfig struct {
	Dstart      time.Time
	Dend        time.Time
	JobConfig   map[string]string
	Assets      map[string]string
	Description string
	Category    string
	ApprovalID  string
	UserID      string
}

type Backfill struct {
	SchedulerRunID string
	CanceledBy     string
	// contains filtered or unexported fields
}

type BackfillState string

type Change struct {
	Property string
	Diff     ChangeDiff
}

type ChangeDiff string

type Changelog struct {
	Change    []Change
	Type      string
	CreatedAt time.Time
}

type ChangelogFilter struct {
	ProjectName tenant.ProjectName
	Name        string
	StartTime   time.Time
	EndTime     time.Time
}

type ConfigMap map[string]string

type DagRun struct {
	RunID         string     `json:"run_id"`
	JobName       string     `json:"dag_id"`
	ScheduledAt   time.Time  `json:"scheduled_at"`
	ExecutionDate time.Time  `json:"execution_date"`
	StartTime     time.Time  `json:"start_date"`
	EndTime       *time.Time `json:"end_date,omitempty"`
}

type DamperFactor struct {
	BaseValue float64
	Alpha     float64
	MinValue  float64
	Method    DamperFactorMethod
}

type DamperFactorMethod int

type DataCompletenessByDate struct {
	Date       time.Time
	IsComplete bool
}

type DataCompletenessStatus struct {
	IsComplete             bool
	DataCompletenessByDate []*DataCompletenessByDate
}

type Event struct {
	JobName        JobName
	URN            string
	Tenant         tenant.Tenant
	Type           JobEventType
	EventTime      time.Time
	OperatorName   string
	Status         State
	JobScheduledAt time.Time
	Values         map[string]any
	SLAObjectList  []*SLAObject

	EventContext *EventContext

	SkipAlerting bool
}

type EventCategory string

type EventContext struct {
	Tenant              tenant.Tenant
	Type                JobEventType        `json:"event_type"`
	OperatorType        OperatorType        `json:"operator_type"`
	OperatorRunInstance OperatorRunInstance `json:"task_instance"`
	DagRun              DagRun              `json:"dag_run"`
	Task                OperatorObj         `json:"task"`
	EventReason         *string             `json:"event_reason,omitempty"`
}

type EventName string

type EventStatus string

type Executor struct {
	Name string
	Type ExecutorType
}

type ExecutorInput struct {
	Configs ConfigMap
	Secrets ConfigMap
	Files   ConfigMap
}

type ExecutorType string

type HTTPUpstreams struct {
	Name    string
	URL     string
	Headers map[string]string
	Params  map[string]string
}

type Hook struct {
	Name        string               `json:"name,omitempty"`
	Version     string               `json:"version,omitempty"`
	Config      map[string]string    `json:"config,omitempty"`
	AlertConfig *OperatorAlertConfig `json:"alert_config,omitempty"`
}

type Job struct {
	ID     uuid.UUID
	Name   JobName
	Tenant tenant.Tenant

	Destination resource.URN
	Task        *Task
	Hooks       []*Hook

	WindowConfig window.Config
	Assets       map[string]string
}

type JobEventType string

type JobExecutionSummary struct {
	JobName JobName
	SLA     SLAConfig
	// Level marks the distance from the original job in question
	Level              int
	JobRunSummary      *JobRunSummary
	DownstreamPathName string
	DelaySummary       *JobRunDelaySummary
	HistoricalSummary  JobHistoricalDuration
}

type JobHistoricalDuration struct {
	TaskDuration time.Duration
	HookDuration time.Duration
}

type JobLineageSummary struct {
	JobName   JobName
	IsEnabled bool
	Upstreams []*JobLineageSummary

	Tenant           tenant.Tenant
	ScheduleInterval string
	SLA              SLAConfig
	Window           *window.Config

	// JobRuns contain the mapping of downstream's job name to their respective job run summaries
	JobRuns map[JobName]*JobRunSummary
}

type JobMetadata struct {
	Version     int
	Owner       string
	Description string
	Labels      map[string]string
}

type JobName string

type JobRun struct {
	ID uuid.UUID

	JobName       JobName
	Tenant        tenant.Tenant
	State         State
	ScheduledAt   time.Time
	SLAAlert      bool
	StartTime     time.Time
	EndTime       *time.Time
	WindowStart   *time.Time
	WindowEnd     *time.Time
	SLADefinition int64

	Monitoring map[string]any
}

type JobRunDelaySummary struct {
	ScheduledWayTooLateSeconds   int64
	SystemSchedulingDelaySeconds int64
}

type JobRunDetailsList []*JobRunWithDetails

type JobRunID uuid.UUID

type JobRunIdentifier struct {
	JobName     JobName
	ScheduledAt time.Time
}

type JobRunLineage struct {
	JobName          JobName
	ScheduledAt      time.Time
	JobRuns          []*JobExecutionSummary
	ExecutionSummary *LineageExecutionSummary
}

type JobRunMeta struct {
	Labels         map[string]string
	DestinationURN resource.URN
}

type JobRunStatus struct {
	ScheduledAt time.Time
	State       State
}

type JobRunStatusList []*JobRunStatus

type JobRunSummary struct {
	JobName     JobName
	ScheduledAt time.Time
	SLATime     *time.Time

	JobStartTime  *time.Time
	JobEndTime    *time.Time
	JobStatus     string
	WaitStartTime *time.Time
	WaitEndTime   *time.Time
	TaskStartTime *time.Time
	TaskEndTime   *time.Time
	HookStartTime *time.Time
	HookEndTime   *time.Time

	SensorName *string
	TaskName   *string
	HookName   *string
}

type JobRunWithDetails struct {
	ScheduledAt     time.Time
	State           State
	RunType         string
	ExternalTrigger bool
	DagRunID        string
	DagID           string
}

type JobRunsCriteria struct {
	Name        string
	StartDate   time.Time
	EndDate     time.Time
	Filter      []string
	OnlyLastRun bool
}

type JobSLAState struct {
	EstimatedDuration *time.Duration
	InferredSLA       *time.Time
}

type JobSchedule struct {
	JobName     JobName
	ScheduledAt time.Time
}

type JobState struct {
	JobSLAState
	JobName       JobName
	JobRun        JobRunSummary
	Tenant        tenant.Tenant
	RelativeLevel int
	Status        SLABreachCause
}

type JobSummary struct {
	JobName          JobName
	IsEnabled        bool
	Tenant           tenant.Tenant
	ScheduleInterval string
	SLA              SLAConfig
	Window           window.Config
}

type JobUpstream struct {
	JobName        string
	Host           string
	TaskName       string        // TODO: remove after airflow migration
	DestinationURN resource.URN  //- bigquery://pilot.playground.table
	Tenant         tenant.Tenant // Current or external tenant
	Type           string
	External       bool
	State          string
}

type JobWithDetails struct {
	Name JobName

	Job           *Job
	JobMetadata   *JobMetadata
	Schedule      *Schedule
	Retry         Retry
	Alerts        []Alert
	Webhook       []Webhook
	RuntimeConfig RuntimeConfig
	Priority      int
	Upstreams     Upstreams
}

type JobWithTaskDuration struct {
	JobName      JobName
	ScheduledAt  time.Time
	TaskDuration time.Duration
	Level        int
}

type Kubernetes struct {
	ServiceAccount string
}

type LineageDelaySummary struct {
	JobName             JobName
	ScheduledAt         time.Time
	UpstreamJobName     JobName
	UpstreamScheduledAt time.Time
	DelayDuration       int64
}

type LineageExecutionSummary struct {
	TotalScheduledWayTooLateSeconds     int64
	TotalSystemSchedulingDelaySeconds   int64
	AverageSystemSchedulingDelaySeconds int64

	TotalLineageDelaySeconds    int64
	TotalLineageDurationSeconds int64

	LargestScheduledWayTooLateJob   LineageDelaySummary
	LargestSystemSchedulingDelayJob LineageDelaySummary

	TopLongestTaskDurationJobs []JobWithTaskDuration
	TopLongestHookDurationJobs []JobWithTaskDuration
}

type NotifyAttrs struct {
	Owner    string
	JobEvent *Event
	Route    string
	Secret   string
}

type OperatorAlertConfig struct {
	SLAAlertConfigs []*SLAAlertConfig `json:"sla_alert_configs,omitempty"`
	Team            string            `json:"team,omitempty"`
}

type OperatorObj struct {
	DownstreamTaskIDs []string `json:"downstream_task_ids"`
}

type OperatorRun struct {
	ID           uuid.UUID
	Name         string
	JobRunID     uuid.UUID
	OperatorType OperatorType
	Status       State
	StartTime    time.Time
	EndTime      *time.Time
}

type OperatorRunInstance struct {
	MaxTries     int        `json:"max_tries"`
	OperatorName string     `json:"task_id"`
	StartTime    time.Time  `json:"start_date"`
	OperatorKey  string     `json:"task_instance_key_str"`
	TryNumber    int        `json:"attempt"`
	State        string     `json:"status"`
	EndTime      *time.Time `json:"end_date,omitempty"`
	LogURL       string     `json:"log_url"`
}

type OperatorSLAAlertAttrs struct {
	Team               string
	Project            string
	Namespace          string
	JobName            string
	OperatorName       string
	OperatorType       string
	Message            string
	Severity           string
	ScheduledAt        time.Time
	StartTime          time.Time
	ExpectedSLAEndTime time.Time
	CurrentState       State

	AlertManager AlertManagerConfig
}

type OperatorType string

type OperatorsSLA struct {
	ID           uuid.UUID
	JobName      JobName
	ProjectName  tenant.ProjectName
	OperatorName string
	RunID        string
	OperatorType OperatorType
	SLATime      time.Time

	AlertTag string

	ScheduledAt       time.Time
	OperatorStartTime time.Time

	WorkerSignature string
	WorkerLockUntil time.Time
}

type PotentialSLABreachAttrs struct {
	TeamName string
	Projects []SLABreachProject
}

type Replay struct {
	// contains filtered or unexported fields
}

type ReplayConfig struct {
	StartTime   time.Time
	EndTime     time.Time
	Parallel    bool
	JobConfig   map[string]string
	Description string
	Category    string
	ApprovalID  string
	UserID      string
}

type ReplayNotificationAttrs struct {
	JobName  string
	ReplayID string
	Tenant   tenant.Tenant
	JobURN   string
	State    ReplayState

	JobWithDetails *JobWithDetails

	AlertManager AlertManagerConfig
}

type ReplayState string // contract status for business layer

type ReplayUserState string // contract status for presentation layer

type ReplayWithRun struct {
	Replay *Replay
	Runs   []*JobRunStatus // TODO: JobRunStatus does not have `message/log`
}

type Resource struct {
	Request *ResourceConfig
	Limit   *ResourceConfig
}

type ResourceConfig struct {
	CPU    string
	Memory string
}

type Retry struct {
	ExponentialBackoff bool
	Count              int
	Delay              int32
}

type RunConfig struct {
	Executor Executor

	ScheduledAt time.Time
	JobRunID    JobRunID
	DagRunID    string
	RunMode     RunMode
}

type RunMode string

type RuntimeConfig struct {
	Resource   *Resource
	Scheduler  map[string]string
	Kubernetes *Kubernetes
}

type SLAAlertConfig struct {
	DurationThreshold time.Duration `json:"duration_threshold,omitempty"`
	Severity          Severity      `json:"severity,omitempty"`
	Team              string        `json:"team,omitempty"`
	AutoThreshold     bool          `json:"auto_threshold,omitempty"`
}

type SLABreachCause string

type SLABreachCombo struct {
	ProjectName tenant.ProjectName
	JobNames    []JobName
	Labels      map[string]string
	GroupName   string // display name for the SLA group; derived from labels if empty
}

type SLABreachGroup struct {
	Name     string
	Severity string
	Targets  []SLABreachTarget
}

type SLABreachProject struct {
	Name   string
	Groups []SLABreachGroup
}

type SLABreachTarget struct {
	JobName string
	Causes  []UpstreamAttrs
}

type SLAConfig struct {
	Duration time.Duration
}

type SLAObject struct {
	JobName        JobName
	JobScheduledAt time.Time
}

type Schedule struct {
	DependsOnPast bool
	CatchUp       bool
	StartDate     time.Time
	EndDate       *time.Time
	Interval      string
}

type Severity string

type State string

type TargetBreach struct {
	TargetProject string
	TargetJobName JobName
	Upstreams     map[JobName]*JobState
}

type Task struct {
	Name        string               `json:"name,omitempty"`
	Version     string               `json:"version,omitempty"`
	Config      map[string]string    `json:"config,omitempty"`
	AlertConfig *OperatorAlertConfig `json:"alert_config,omitempty"`
}

type TeamBreachAggregator struct {
	// contains filtered or unexported fields
}

type ThirdPartyUpstream struct {
	Type       string
	Identifier string
	Config     map[string]string
}

type UpstreamAttrs struct {
	JobName       string
	RelativeLevel int
	Status        string
}

type Upstreams struct {
	HTTP         []*HTTPUpstreams
	UpstreamJobs []*JobUpstream
	ThirdParty   []*ThirdPartyUpstream
}

type Webhook struct {
	On        EventCategory
	Endpoints []WebhookEndPoint
}

type WebhookAttrs struct {
	Owner    string
	JobEvent *Event
	Meta     *JobRunMeta
	Route    string
	Headers  map[string]string
}

type WebhookEndPoint struct {
	URL     string
	Headers map[string]string
}