Andrey Smirnov bddd4f1bf6 refactor: move external API packages into machinery/
This moves `pkg/config`, `pkg/client` and `pkg/constants`
under `pkg/machinery` umbrella.

And `pkg/machinery` is published as Go module inside Talos repository.

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-08-17 09:56:14 -07:00

32 lines
776 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/.
// +build integration
package base
import "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(wrapper.masterNodes, wrapper.workerNodes...)
}
func (wrapper *infoWrapper) NodesByType(t machine.Type) []string {
switch t {
case machine.TypeInit:
return nil
case machine.TypeControlPlane:
return wrapper.masterNodes
case machine.TypeJoin:
return wrapper.workerNodes
default:
panic("unreachable")
}
}