zdravko/database/models/models.go

127 lines
2.5 KiB
Go

package models
import (
"database/sql/driver"
"fmt"
"time"
)
type Time struct {
Time time.Time
}
// rfc3339Milli is like time.RFC3339Nano, but with millisecond precision, and fractional seconds do not have trailing
// zeros removed.
const rfc3339Milli = "2006-01-02T15:04:05.000Z07:00"
// Value satisfies driver.Valuer interface.
func (t *Time) Value() (driver.Value, error) {
return t.Time.UTC().Format(rfc3339Milli), nil
}
// Scan satisfies sql.Scanner interface.
func (t *Time) Scan(src any) error {
if src == nil {
return nil
}
s, ok := src.(string)
if !ok {
return fmt.Errorf("error scanning time, got %+v", src)
}
parsedT, err := time.Parse(rfc3339Milli, s)
if err != nil {
return err
}
t.Time = parsedT.UTC()
return nil
}
type OAuth2State struct {
State string `db:"state"`
ExpiresAt *Time `db:"expires_at"`
}
type MonitorStatus string
const (
MonitorSuccess MonitorStatus = "SUCCESS"
MonitorFailure MonitorStatus = "FAILURE"
MonitorError MonitorStatus = "ERROR"
MonitorUnknown MonitorStatus = "UNKNOWN"
)
type Monitor struct {
CreatedAt *Time `db:"created_at"`
UpdatedAt *Time `db:"updated_at"`
Id string `db:"id"`
Name string `db:"name"`
Group string `db:"group"`
Schedule string `db:"schedule"`
Script string `db:"script"`
}
type MonitorWithWorkerGroups struct {
Monitor
// List of worker group names
WorkerGroups []string
}
type MonitorHistory struct {
CreatedAt *Time `db:"created_at"`
MonitorId string `db:"monitor_id"`
Status MonitorStatus `db:"status"`
Note string `db:"note"`
WorkerGroupId string `db:"worker_group_id"`
WorkerGroupName string `db:"worker_group_name"`
}
type WorkerGroup struct {
CreatedAt *Time `db:"created_at"`
UpdatedAt *Time `db:"updated_at"`
Id string `db:"id"`
Name string `db:"name"`
}
type WorkerGroupWithMonitors struct {
WorkerGroup
// List of worker group names
Monitors []string
}
type TriggerStatus string
const (
TriggerSuccess TriggerStatus = "SUCCESS"
TriggerFailure TriggerStatus = "FAILURE"
TriggerError TriggerStatus = "ERROR"
TriggerUnknown TriggerStatus = "UNKNOWN"
)
type Trigger struct {
CreatedAt *Time `db:"created_at"`
UpdatedAt *Time `db:"updated_at"`
Id string `db:"id"`
Name string `db:"name"`
Script string `db:"script"`
Status TriggerStatus `db:"status"`
}
type TriggerHistory struct {
CreatedAt *Time `db:"created_at"`
TriggerId string `db:"trigger_id"`
Status TriggerStatus `db:"status"`
Note string `db:"note"`
}