WIP

Author: yhabteab

pkg/icingadb/sla.go

@@ -0,0 +1,79 @@
+package icingadb
+
+import (
+	"context"
+	"crypto/rand"
+	"github.com/icinga/icingadb/pkg/com"
+	"github.com/icinga/icingadb/pkg/contracts"
+	v1 "github.com/icinga/icingadb/pkg/icingadb/v1"
+	"github.com/icinga/icingadb/pkg/types"
+	"github.com/icinga/icingadb/pkg/utils"
+	"github.com/pkg/errors"
+	"golang.org/x/sync/errgroup"
+)
+
+type SlaHistoryTrail struct {
+	v1.EntityWithoutChecksum `json:",inline"`
+	EnvironmentId            types.Binary    `json:"environment_id"`
+	ObjectType               string          `json:"object_type"`
+	HostId                   types.Binary    `json:"host_id"`
+	ServiceId                types.Binary    `json:"service_id"`
+	EventTime                types.UnixMilli `json:"event_time"`
+	EventType                string          `json:"event_type"`
+}
+
+func CheckableToSlaTrailEntities(ctx context.Context, checkables <-chan contracts.Entity, eventType string) (<-chan contracts.Entity, <-chan error) {
+	entities := make(chan contracts.Entity)
+	g, ctx := errgroup.WithContext(ctx)
+
+	g.Go(func() error {
+		defer close(entities)
+
+		for {
+			select {
+			case checkable, ok := <-checkables:
+				if !ok {
+					return nil
+				}
+
+				id, err := generateBinaryId()
+				if err != nil {
+					return errors.Wrap(err, "can't generate sla history trail ID")
+				}
+
+				entity := &SlaHistoryTrail{
+					EntityWithoutChecksum: v1.EntityWithoutChecksum{
+						IdMeta: v1.IdMeta{Id: id},
+					},
+					ObjectType: utils.Name(checkable),
+					EventTime:  types.UnixMilli{},
+					EventType:  eventType,
+				}
+
+				switch ptr := checkable.(type) {
+				case *v1.Host:
+					entity.HostId = ptr.Id
+					entity.EnvironmentId = ptr.EnvironmentId
+				case *v1.Service:
+					entity.HostId = ptr.HostId
+					entity.ServiceId = ptr.Id
+					entity.EnvironmentId = ptr.EnvironmentId
+				}
+
+				entities <- entity
+			case <-ctx.Done():
+				return ctx.Err()
+			}
+		}
+	})
+
+	return entities, com.WaitAsync(g)
+}
+
+// GenerateBinaryId generates a 20 byte length random id
+func generateBinaryId() (types.Binary, error) {
+	id := make([]byte, 20)
+	_, err := rand.Read(id)
+
+	return id, err
+}

pkg/icingadb/sync.go

@@ -131,6 +131,20 @@ func (s Sync) ApplyDelta(ctx context.Context, delta *Delta) error {
 		g.Go(func() error {
 			return s.db.CreateStreamed(ctx, entities, OnSuccessIncrement[contracts.Entity](stat))
 		})
+
+		subjectType := delta.Subject.Entity()
+		switch subjectType.(type) {
+		case *v1.Host, *v1.Service:
+			s.logger.Infof("Inserting %d items of type %s sla history trails", len(delta.Create), utils.Key(utils.Name(delta.Subject.Entity()), ' '))
+
+			var slaTrails <-chan contracts.Entity
+			slaTrails, errs := CheckableToSlaTrailEntities(ctx, entities, "create")
+			com.ErrgroupReceive(g, errs)
+
+			g.Go(func() error {
+				return s.db.CreateStreamed(ctx, slaTrails, OnSuccessIncrement[contracts.Entity](stat))
+			})
+		}
 	}
 
 	// Update

schema/mysql/schema.sql

@@ -1321,6 +1321,19 @@ CREATE TABLE sla_history_downtime (
   INDEX idx_sla_history_downtime_env_downtime_end (environment_id, downtime_end) COMMENT 'Filter for sla history retention'
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin ROW_FORMAT=DYNAMIC;
 
+CREATE TABLE sla_history_trail (
+  id binary(20) NOT NULL,
+  environment_id binary(20) NOT NULL COMMENT 'environment.id',
+  object_type enum('host', 'service') NOT NULL,
+  host_id binary(20) NOT NULL COMMENT 'host.id (may reference already deleted hosts)',
+  service_id binary(20) NOT NULL COMMENT 'service.id (may reference already deleted services)',
+
+  PRIMARY KEY (id),
+
+  event_type enum('delete', 'create') NOT NULL,
+  event_time bigint unsigned NOT NULL COMMENT 'unix timestamp the event occurred'
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin ROW_FORMAT=DYNAMIC;
+
 CREATE TABLE icingadb_schema (
   id int unsigned NOT NULL AUTO_INCREMENT,
   version smallint unsigned NOT NULL,

tests/object_sync_test.go

@@ -310,6 +310,24 @@ func TestObjectSync(t *testing.T) {
 		t.Skip()
 	})
 
+	t.Run("Sla History Trail", func(t *testing.T) {
+		t.Parallel()
+
+		assert.Eventuallyf(t, func() bool {
+			var count int
+			err := db.Get(&count, db.Rebind("SELECT COUNT(*) FROM sla_history_trail WHERE service_id IS NULL"))
+			require.NoError(t, err, "querying hosts sla history trail should not fail")
+			return count == len(data.Hosts)
+		}, 20*time.Second, 200*time.Millisecond, "Newly created hosts should exists in database")
+
+		assert.Eventuallyf(t, func() bool {
+			var count int
+			err := db.Get(&count, db.Rebind("SELECT COUNT(*) FROM sla_history_trail WHERE service_id IS NOT NULL"))
+			require.NoError(t, err, "querying services sla history trail should not fail")
+			return count == len(data.Services)
+		}, 20*time.Second, 200*time.Millisecond, "Newly created services should exists in database")
+	})
+
 	t.Run("RuntimeUpdates", func(t *testing.T) {
 		t.Parallel()