Add custom healthcheck for NATS resources

Signed-off-by: DanielS <[email protected]>

Author: Koalk

resource_customizations/jetstream.nats.io/Account/health.lua

@@ -0,0 +1,34 @@
+hs = {}
+local healthy_cons = { Created=true, Updated=true, Noop=true, SkipCreate=true, SkipUpdate=true }
+if obj.status ~= nil then
+    if obj.status.conditions ~= nil then
+    for i, condition in ipairs(obj.status.conditions) do
+        if condition.status == "False" then
+        hs.status = "Degraded"
+        hs.message = condition.message
+        return hs
+        end
+        if condition.type == "Ready" and healthy_cons[condition.reason]  then
+        hs.status = "Healthy"
+        hs.message = condition.message
+        return hs
+        end
+        if condition.type == "Ready" and not healthy_cons[condition.reason] then
+        hs.status = "Progressing"
+        hs.message = condition.message
+        local pattern = "(%d+)%-(%d+)%-(%d+)%a(%d+)%:(%d+)%:([%d]+)%.%d+Z"
+        local year, month, day, hour, minute, seconds = condition.lastTransitionTime:match(pattern)
+        local event_time = os.time{year = year, month = month, day = day, hour = hour, 
+            min = minute, sec = seconds}
+        if os.difftime(os.time(), event_time) > 30 then
+            hs.status = "Degraded"
+            hs.message = "Trying to create resource for more than 30 seconds"
+        end
+        return hs
+        end
+    end
+    end
+end
+hs.status = "Progressing"
+hs.message = "Waiting for Nacks"
+return hs

resource_customizations/jetstream.nats.io/Account/health_test.yaml

@@ -0,0 +1,21 @@
+tests:
+- healthStatus:
+    status: Progressing
+    message: Waiting for Nacks
+  inputPath: test_data/progressing.yaml
+- healthStatus:
+    status: Degraded
+    message: 'failed to create account'
+  inputPath: test_data/degraded_failure.yaml
+- healthStatus:
+    status: Degraded
+    message: 'Trying to create resource for more than 30 seconds'
+  inputPath: test_data/degraded_stuck.yaml
+- healthStatus:
+    status: Healthy
+    message: 'Account successfully updated'
+  inputPath: test_data/healthy_updated.yaml
+- healthStatus:
+    status: Healthy
+    message: 'Account successfully created'
+  inputPath: test_data/healthy_created.yaml

resource_customizations/jetstream.nats.io/Account/test_data/degraded_failure.yaml

@@ -0,0 +1,23 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Account
+metadata:
+  name: account-name
+spec:
+  name: a
+  servers:
+    - nats://nats:4222
+  tls:
+    secret:
+      name: nack-a-tls
+    ca: "ca.crt"
+    cert: "tls.crt"
+    key: "tls.key"
+status:
+  conditions:
+    - lastTransitionTime: '2025-05-14T14:25:20.89689315Z'
+      message: >-
+        failed to create account
+      reason: Errored
+      status: 'False'
+      type: Ready
+  observedGeneration: 0

resource_customizations/jetstream.nats.io/Account/test_data/degraded_stuck.yaml

@@ -0,0 +1,22 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Account
+metadata:
+  name: account-name
+spec:
+  name: a
+  servers:
+    - nats://nats:4222
+  tls:
+    secret:
+      name: nack-a-tls
+    ca: "ca.crt"
+    cert: "tls.crt"
+    key: "tls.key"
+status:
+  conditions:
+    - lastTransitionTime: '2025-05-14T14:22:01.962295009Z'
+      message: Creating account
+      reason: Creating
+      status: 'True'
+      type: Ready
+  observedGeneration: 0

resource_customizations/jetstream.nats.io/Account/test_data/healthy_created.yaml

@@ -0,0 +1,22 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Account
+metadata:
+  name: account-name
+spec:
+  name: a
+  servers:
+    - nats://nats:4222
+  tls:
+    secret:
+      name: nack-a-tls
+    ca: "ca.crt"
+    cert: "tls.crt"
+    key: "tls.key"
+status:
+  conditions:
+    - lastTransitionTime: '2025-05-14T14:22:01.962295009Z'
+      message: Account successfully created
+      reason: Created
+      status: 'True'
+      type: Ready
+  observedGeneration: 1

resource_customizations/jetstream.nats.io/Account/test_data/healthy_updated.yaml

@@ -0,0 +1,22 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Account
+metadata:
+  name: account-name
+spec:
+  name: a
+  servers:
+    - nats://nats:4222
+  tls:
+    secret:
+      name: nack-a-tls
+    ca: "ca.crt"
+    cert: "tls.crt"
+    key: "tls.key"
+status:
+  conditions:
+    - lastTransitionTime: '2025-05-14T14:22:01.962295009Z'
+      message: Account successfully updated
+      reason: Updated
+      status: 'True'
+      type: Ready
+  observedGeneration: 1

resource_customizations/jetstream.nats.io/Account/test_data/progressing.yaml

@@ -0,0 +1,14 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Account
+metadata:
+  name: account-name
+spec:
+  name: a
+  servers:
+    - nats://nats:4222
+  tls:
+    secret:
+      name: nack-a-tls
+    ca: "ca.crt"
+    cert: "tls.crt"
+    key: "tls.key"

resource_customizations/jetstream.nats.io/Consumer/health.lua

@@ -0,0 +1,34 @@
+hs = {}
+local healthy_cons = { Created=true, Updated=true, Noop=true, SkipCreate=true, SkipUpdate=true }
+if obj.status ~= nil then
+    if obj.status.conditions ~= nil then
+    for i, condition in ipairs(obj.status.conditions) do
+        if condition.status == "False" then
+        hs.status = "Degraded"
+        hs.message = condition.message
+        return hs
+        end
+        if condition.type == "Ready" and healthy_cons[condition.reason]  then
+        hs.status = "Healthy"
+        hs.message = condition.message
+        return hs
+        end
+        if condition.type == "Ready" and not healthy_cons[condition.reason] then
+        hs.status = "Progressing"
+        hs.message = condition.message
+        local pattern = "(%d+)%-(%d+)%-(%d+)%a(%d+)%:(%d+)%:([%d]+)%.%d+Z"
+        local year, month, day, hour, minute, seconds = condition.lastTransitionTime:match(pattern)
+        local event_time = os.time{year = year, month = month, day = day, hour = hour, 
+            min = minute, sec = seconds}
+        if os.difftime(os.time(), event_time) > 30 then
+            hs.status = "Degraded"
+            hs.message = "Trying to create resource for more than 30 seconds"
+        end
+        return hs
+        end
+    end
+    end
+end
+hs.status = "Progressing"
+hs.message = "Waiting for Nacks"
+return hs

resource_customizations/jetstream.nats.io/Consumer/health_test.yaml

@@ -0,0 +1,21 @@
+tests:
+- healthStatus:
+    status: Progressing
+    message: Waiting for Nacks
+  inputPath: test_data/progressing.yaml
+- healthStatus:
+    status: Degraded
+    message: 'failed to create consumer'
+  inputPath: test_data/degraded_failure.yaml
+- healthStatus:
+    status: Degraded
+    message: 'Trying to create resource for more than 30 seconds'
+  inputPath: test_data/degraded_stuck.yaml
+- healthStatus:
+    status: Healthy
+    message: 'Consumer successfully updated'
+  inputPath: test_data/healthy_updated.yaml
+- healthStatus:
+    status: Healthy
+    message: 'Consumer successfully created'
+  inputPath: test_data/healthy_created.yaml

resource_customizations/jetstream.nats.io/Consumer/test_data/degraded_failure.yaml

@@ -0,0 +1,20 @@
+apiVersion: jetstream.nats.io/v1beta2
+kind: Consumer
+metadata:
+  name: consumer-name
+spec:
+  streamName: mystream
+  durableName: my-pull-consumer
+  deliverPolicy: all
+  filterSubject: orders.received
+  maxDeliver: 20
+  ackPolicy: explicit
+status:
+  conditions:
+    - lastTransitionTime: '2025-05-14T14:25:20.89689315Z'
+      message: >-
+        failed to create consumer
+      reason: Errored
+      status: 'False'
+      type: Ready
+  observedGeneration: 0