talos/internal/integration/base/cluster.go
Utku Ozdemir 8d2be5e315
feat: extend node definition used in health checks
Introduce `cluster.NodeInfo` to represent the basic info about a node which can be used in the health checks. This information, where possible, will be populated by the discovery service in following PRs. Part of siderolabs#5554.

Signed-off-by: Utku Ozdemir <utku.ozdemir@siderolabs.com>
2022-06-13 14:13:42 +02:00

30 lines
762 B
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//go:build integration
// +build integration
package base
import (
"github.com/talos-systems/talos/pkg/cluster"
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
)
type infoWrapper struct {
masterNodes []string
workerNodes []string
nodeInfos []cluster.NodeInfo
nodeInfosByType map[machine.Type][]cluster.NodeInfo
}
func (wrapper *infoWrapper) Nodes() []cluster.NodeInfo {
return wrapper.nodeInfos
}
func (wrapper *infoWrapper) NodesByType(t machine.Type) []cluster.NodeInfo {
return wrapper.nodeInfosByType[t]
}