diff options
Diffstat (limited to 'src/go/plugin/go.d/modules/k8s_state/resource.go')
-rw-r--r-- | src/go/plugin/go.d/modules/k8s_state/resource.go | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/src/go/plugin/go.d/modules/k8s_state/resource.go b/src/go/plugin/go.d/modules/k8s_state/resource.go new file mode 100644 index 000000000..cabd41a67 --- /dev/null +++ b/src/go/plugin/go.d/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)) + } +} |