talos/internal/integration/base/cluster.go
Andrey Smirnov badbc51e63
refactor: rewrite code to include preliminary support for multi-doc
`config.Container` implements a multi-doc container which implements
both `Container` interface (encoding, validation, etc.), and `Conifg`
interface (accessing parts of the config).

Refactor `generate` and `bundle` packages to support multi-doc, and
provide backwards compatibility.

Implement a first (mostly example) machine config document for
SideroLink API URL.

Many places don't properly support multi-doc yet (e.g. config patches).

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2023-05-31 18:38:05 +04:00

48 lines
1.2 KiB
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
package base
import (
"github.com/siderolabs/gen/slices"
"github.com/siderolabs/talos/pkg/cluster"
"github.com/siderolabs/talos/pkg/machinery/config/machine"
)
type infoWrapper struct {
nodeInfos []cluster.NodeInfo
nodeInfosByType map[machine.Type][]cluster.NodeInfo
}
func newNodeInfo(masterNodes, workerNodes []string) (*infoWrapper, error) {
controlPlaneNodeInfos, err := cluster.IPsToNodeInfos(masterNodes)
if err != nil {
return nil, err
}
workerNodeInfos, err := cluster.IPsToNodeInfos(workerNodes)
if err != nil {
return nil, err
}
return &infoWrapper{
nodeInfos: append(slices.Clone(controlPlaneNodeInfos), workerNodeInfos...),
nodeInfosByType: map[machine.Type][]cluster.NodeInfo{
machine.TypeControlPlane: controlPlaneNodeInfos,
machine.TypeWorker: workerNodeInfos,
},
}, nil
}
func (wrapper *infoWrapper) Nodes() []cluster.NodeInfo {
return wrapper.nodeInfos
}
func (wrapper *infoWrapper) NodesByType(t machine.Type) []cluster.NodeInfo {
return wrapper.nodeInfosByType[t]
}