1 files changed, 68 insertions, 0 deletions
diff --git a/pkg/types/notification_type.go b/pkg/types/notification_type.go
new file mode 100644
index 0000000..f2980f4
--- /dev/null
+++ b/pkg/types/notification_type.go
@@ -0,0 +1,68 @@
+package types
+
+import (
+	"database/sql/driver"
+	"encoding"
+	"github.com/icinga/icingadb/internal"
+	"github.com/pkg/errors"
+	"strconv"
+)
+
+// NotificationType specifies the reason of a sent notification.
+type NotificationType uint16
+
+// UnmarshalText implements the encoding.TextUnmarshaler interface.
+func (nt *NotificationType) UnmarshalText(text []byte) error {
+	s := string(text)
+
+	i, err := strconv.ParseUint(s, 10, 64)
+	if err != nil {
+		return internal.CantParseUint64(err, s)
+	}
+
+	n := NotificationType(i)
+	if uint64(n) != i {
+		// Truncated due to above cast, obviously too high
+		return badNotificationType(s)
+	}
+
+	if _, ok := notificationTypes[n]; !ok {
+		return badNotificationType(s)
+	}
+
+	*nt = n
+	return nil
+}
+
+// Value implements the driver.Valuer interface.
+func (nt NotificationType) Value() (driver.Value, error) {
+	if v, ok := notificationTypes[nt]; ok {
+		return v, nil
+	} else {
+		return nil, badNotificationType(nt)
+	}
+}
+
+// badNotificationType returns an error about a syntactically, but not semantically valid NotificationType.
+func badNotificationType(t interface{}) error {
+	return errors.Errorf("bad notification type: %#v", t)
+}
+
+// notificationTypes maps all valid NotificationType values to their SQL representation.
+var notificationTypes = map[NotificationType]string{
+	1:   "downtime_start",
+	2:   "downtime_end",
+	4:   "downtime_removed",
+	8:   "custom",
+	16:  "acknowledgement",
+	32:  "problem",
+	64:  "recovery",
+	128: "flapping_start",
+	256: "flapping_end",
+}
+
+// Assert interface compliance.
+var (
+	_ encoding.TextUnmarshaler = (*NotificationType)(nil)
+	_ driver.Valuer            = NotificationType(0)
+)