1 files changed, 161 insertions, 0 deletions
diff --git a/pkg/icingadb/v1/history/downtime.go b/pkg/icingadb/v1/history/downtime.go
new file mode 100644
index 0000000..99f93f6
--- /dev/null
+++ b/pkg/icingadb/v1/history/downtime.go
@@ -0,0 +1,161 @@
+package history
+
+import (
+	"database/sql/driver"
+	"github.com/icinga/icingadb/pkg/contracts"
+	"github.com/icinga/icingadb/pkg/types"
+)
+
+type DowntimeHistoryEntity struct {
+	DowntimeId types.Binary `json:"downtime_id"`
+}
+
+// Fingerprint implements part of the contracts.Entity interface.
+func (dhe DowntimeHistoryEntity) Fingerprint() contracts.Fingerprinter {
+	return dhe
+}
+
+// ID implements part of the contracts.Entity interface.
+func (dhe DowntimeHistoryEntity) ID() contracts.ID {
+	return dhe.DowntimeId
+}
+
+// SetID implements part of the contracts.Entity interface.
+func (dhe *DowntimeHistoryEntity) SetID(id contracts.ID) {
+	dhe.DowntimeId = id.(types.Binary)
+}
+
+type DowntimeHistoryUpserter struct {
+	CancelledBy      types.String    `json:"cancelled_by"`
+	HasBeenCancelled types.Bool      `json:"has_been_cancelled"`
+	CancelTime       types.UnixMilli `json:"cancel_time"`
+}
+
+// Upsert implements the contracts.Upserter interface.
+func (dhu *DowntimeHistoryUpserter) Upsert() interface{} {
+	return dhu
+}
+
+type DowntimeHistory struct {
+	DowntimeHistoryEntity   `json:",inline"`
+	HistoryTableMeta        `json:",inline"`
+	DowntimeHistoryUpserter `json:",inline"`
+	TriggeredById           types.Binary    `json:"triggered_by_id"`
+	ParentId                types.Binary    `json:"parent_id"`
+	EntryTime               types.UnixMilli `json:"entry_time"`
+	Author                  string          `json:"author"`
+	Comment                 string          `json:"comment"`
+	IsFlexible              types.Bool      `json:"is_flexible"`
+	FlexibleDuration        uint64          `json:"flexible_duration"`
+	ScheduledStartTime      types.UnixMilli `json:"scheduled_start_time"`
+	ScheduledEndTime        types.UnixMilli `json:"scheduled_end_time"`
+	StartTime               types.UnixMilli `json:"start_time"`
+	EndTime                 types.UnixMilli `json:"end_time"`
+	ScheduledBy             types.String    `json:"scheduled_by"`
+	TriggerTime             types.UnixMilli `json:"trigger_time"`
+}
+
+type HistoryDowntime struct {
+	HistoryMeta       `json:",inline"`
+	DowntimeHistoryId types.Binary `json:"downtime_id"`
+
+	// Idea: read StartTime, CancelTime, EndTime and HasBeenCancelled from Redis
+	// and let EventTime decide based on HasBeenCancelled which of the others to write to MySQL.
+	// So EventTime doesn't have to be read from Redis (json:"-")
+	// and the others don't have to be written to MySQL (db:"-").
+	StartTime        types.UnixMilli   `json:"start_time" db:"-"`
+	CancelTime       types.UnixMilli   `json:"cancel_time" db:"-"`
+	EndTime          types.UnixMilli   `json:"end_time" db:"-"`
+	HasBeenCancelled types.Bool        `json:"has_been_cancelled" db:"-"`
+	EventTime        DowntimeEventTime `json:"-"`
+}
+
+// Init implements the contracts.Initer interface.
+func (h *HistoryDowntime) Init() {
+	h.EventTime.History = h
+}
+
+// TableName implements the contracts.TableNamer interface.
+func (*HistoryDowntime) TableName() string {
+	return "history"
+}
+
+type SlaHistoryDowntime struct {
+	DowntimeHistoryEntity      `json:",inline"`
+	HistoryTableMeta           `json:",inline"`
+	SlaHistoryDowntimeUpserter `json:",inline"`
+	DowntimeStart              types.UnixMilli `json:"start_time"`
+	HasBeenCancelled           types.Bool      `json:"has_been_cancelled" db:"-"`
+	CancelTime                 types.UnixMilli `json:"cancel_time" db:"-"`
+	EndTime                    types.UnixMilli `json:"end_time" db:"-"`
+}
+
+// Init implements the contracts.Initer interface.
+func (s *SlaHistoryDowntime) Init() {
+	s.DowntimeEnd.History = s
+}
+
+type SlaHistoryDowntimeUpserter struct {
+	DowntimeEnd SlaDowntimeEndTime `json:"-"`
+}
+
+// Upsert implements the contracts.Upserter interface.
+func (h *SlaHistoryDowntimeUpserter) Upsert() interface{} {
+	return h
+}
+
+type DowntimeEventTime struct {
+	History *HistoryDowntime `db:"-"`
+}
+
+// Value implements the driver.Valuer interface.
+// Supports SQL NULL.
+func (et DowntimeEventTime) Value() (driver.Value, error) {
+	if et.History == nil {
+		return nil, nil
+	}
+
+	switch et.History.EventType {
+	case "downtime_start":
+		return et.History.StartTime.Value()
+	case "downtime_end":
+		if !et.History.HasBeenCancelled.Valid {
+			return nil, nil
+		}
+
+		if et.History.HasBeenCancelled.Bool {
+			return et.History.CancelTime.Value()
+		} else {
+			return et.History.EndTime.Value()
+		}
+	default:
+		return nil, nil
+	}
+}
+
+type SlaDowntimeEndTime struct {
+	History *SlaHistoryDowntime `db:"-"`
+}
+
+// Value implements the driver.Valuer interface.
+func (et SlaDowntimeEndTime) Value() (driver.Value, error) {
+	if et.History.HasBeenCancelled.Valid && et.History.HasBeenCancelled.Bool {
+		return et.History.CancelTime.Value()
+	} else {
+		return et.History.EndTime.Value()
+	}
+}
+
+// Assert interface compliance.
+var (
+	_ contracts.Entity     = (*DowntimeHistoryEntity)(nil)
+	_ contracts.Upserter   = (*DowntimeHistoryUpserter)(nil)
+	_ UpserterEntity       = (*DowntimeHistory)(nil)
+	_ contracts.Initer     = (*HistoryDowntime)(nil)
+	_ contracts.TableNamer = (*HistoryDowntime)(nil)
+	_ UpserterEntity       = (*HistoryDowntime)(nil)
+	_ contracts.Initer     = (*SlaHistoryDowntime)(nil)
+	_ UpserterEntity       = (*SlaHistoryDowntime)(nil)
+	_ driver.Valuer        = DowntimeEventTime{}
+	_ driver.Valuer        = SlaDowntimeEndTime{}
+)