1 files changed, 44 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/k8s_state/resource.go b/src/go/collectors/go.d.plugin/modules/k8s_state/resource.go
new file mode 100644
index 000000000..cabd41a67
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/k8s_state/resource.go
@@ -0,0 +1,44 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package k8s_state
+
+import (
+	"fmt"
+
+	corev1 "k8s.io/api/core/v1"
+)
+
+type resource interface {
+	source() string
+	kind() kubeResourceKind
+	value() interface{}
+}
+
+type kubeResourceKind uint8
+
+const (
+	kubeResourceNode kubeResourceKind = iota + 1
+	kubeResourcePod
+)
+
+func toNode(i interface{}) (*corev1.Node, error) {
+	switch v := i.(type) {
+	case *corev1.Node:
+		return v, nil
+	case resource:
+		return toNode(v.value())
+	default:
+		return nil, fmt.Errorf("unexpected type: %T (expected %T or %T)", v, &corev1.Node{}, resource(nil))
+	}
+}
+
+func toPod(i interface{}) (*corev1.Pod, error) {
+	switch v := i.(type) {
+	case *corev1.Pod:
+		return v, nil
+	case resource:
+		return toPod(v.value())
+	default:
+		return nil, fmt.Errorf("unexpected type: %T (expected %T or %T)", v, &corev1.Pod{}, resource(nil))
+	}
+}