Add Icinga states to daemon set

pkg/schema/v1/daemon_set.go

@@ -1,6 +1,7 @@
 package v1
 
 import (
+	"fmt"
 	"github.com/icinga/icinga-go-library/types"
 	"github.com/icinga/icinga-kubernetes/pkg/database"
 	"github.com/icinga/icinga-kubernetes/pkg/strcase"
@@ -24,6 +25,8 @@ type DaemonSet struct {
 	NumberAvailable        int32
 	NumberUnavailable      int32
 	Yaml                   string
+	IcingaState            IcingaState
+	IcingaStateReason      string
 	Conditions             []DaemonSetCondition `db:"-"`
 	Labels                 []Label              `db:"-"`
 	DaemonSetLabels        []DaemonSetLabel     `db:"-"`
@@ -61,6 +64,7 @@ func (d *DaemonSet) Obtain(k8s kmetav1.Object) {
 	d.UpdateNumberScheduled = daemonSet.Status.UpdatedNumberScheduled
 	d.NumberAvailable = daemonSet.Status.NumberAvailable
 	d.NumberUnavailable = daemonSet.Status.NumberUnavailable
+	d.IcingaState, d.IcingaStateReason = d.getIcingaState()
 
 	for _, condition := range daemonSet.Status.Conditions {
 		d.Conditions = append(d.Conditions, DaemonSetCondition{
@@ -92,6 +96,33 @@ func (d *DaemonSet) Obtain(k8s kmetav1.Object) {
 	d.Yaml = string(output)
 }
 
+func (d *DaemonSet) getIcingaState() (IcingaState, string) {
+	if d.DesiredNumberScheduled < 1 {
+		reason := fmt.Sprintf("DaemonSet %s/%s has an invalid desired node count: %d", d.Namespace, d.Name, d.DesiredNumberScheduled)
+
+		return Unknown, reason
+	}
+
+	if gracePeriodReason := IsWithinGracePeriod(d); gracePeriodReason != nil {
+		return Ok, *gracePeriodReason
+	}
+
+	switch {
+	case d.NumberAvailable == 0:
+		reason := fmt.Sprintf("DaemonSet %s/%s does not have a single pod available which should run on %d desired nodes", d.Namespace, d.Name, d.DesiredNumberScheduled)
+
+		return Critical, reason
+	case d.NumberAvailable < d.DesiredNumberScheduled:
+		reason := fmt.Sprintf("DaemonSet %s/%s pods are only available on %d out of %d desired nodes", d.Namespace, d.Name, d.NumberAvailable, d.DesiredNumberScheduled)
+
+		return Warning, reason
+	default:
+		reason := fmt.Sprintf("DaemonSet %s/%s has pods available on all %d desired nodes", d.Namespace, d.Name, d.DesiredNumberScheduled)
+
+		return Ok, reason
+	}
+}
+
 func (d *DaemonSet) Relations() []database.Relation {
 	fk := database.WithForeignKey("daemon_set_uuid")
 

schema/mysql/schema.sql

@@ -402,6 +402,8 @@ CREATE TABLE daemon_set (
   number_available int unsigned NOT NULL,
   number_unavailable int unsigned NOT NULL,
   yaml mediumblob DEFAULT NULL,
+  icinga_state enum('ok', 'warning', 'critical', 'unknown') COLLATE utf8mb4_unicode_ci NOT NULL,
+  icinga_state_reason text NOT NULL,
   created bigint unsigned NOT NULL,
   PRIMARY KEY (uuid)
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;