talos/internal/integration/base/cluster.go
Andrey Smirnov a059454045
chore: build using Go 1.17
`initramfs` size for amd64 shrinks by 1.3 MiB.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2021-09-13 22:33:47 +03:00

39 lines
964 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 (
"fmt"
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
)
type infoWrapper struct {
masterNodes []string
workerNodes []string
}
func (wrapper *infoWrapper) Nodes() []string {
return append([]string(nil), append(wrapper.masterNodes, wrapper.workerNodes...)...)
}
func (wrapper *infoWrapper) NodesByType(t machine.Type) []string {
switch t {
case machine.TypeInit:
return nil
case machine.TypeControlPlane:
return append([]string(nil), wrapper.masterNodes...)
case machine.TypeWorker:
return append([]string(nil), wrapper.workerNodes...)
case machine.TypeUnknown:
fallthrough
default:
panic(fmt.Sprintf("unexpected machine type %v", t))
}
}