mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-23 15:41:10 +02:00
9379cf9ee1
This change is only moving packages and updating import paths. Goal: expose `internal/pkg/provision` as `pkg/provision` to enable other projects to import Talos provisioning library. As cluster checks are almost always required as part of provisioning process, package `internal/pkg/cluster` was also made public as `pkg/cluster`. Other changes were direct dependencies discovered by `importvet` which were updated. Public packages (useful, general purpose packages with stable API): * `internal/pkg/conditions` -> `pkg/conditions` * `internal/pkg/tail` -> `pkg/tail` Private packages (used only on provisioning library internally): * `internal/pkg/inmemhttp` -> `pkg/provision/internal/inmemhttp` * `internal/pkg/kernel/vmlinuz` -> `pkg/provision/internal/vmlinuz` * `internal/pkg/cniutils` -> `pkg/provision/internal/cniutils` Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
75 lines
1.8 KiB
Go
75 lines
1.8 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/.
|
|
|
|
package access
|
|
|
|
import (
|
|
"github.com/talos-systems/talos/pkg/cluster"
|
|
"github.com/talos-systems/talos/pkg/config/types/v1alpha1/machine"
|
|
"github.com/talos-systems/talos/pkg/provision"
|
|
)
|
|
|
|
// Adapter provides cluster access via provision.Cluster.
|
|
type Adapter struct {
|
|
cluster.ConfigClientProvider
|
|
cluster.KubernetesClient
|
|
cluster.APICrashDumper
|
|
cluster.Info
|
|
}
|
|
|
|
type infoWrapper struct {
|
|
clusterInfo provision.ClusterInfo
|
|
}
|
|
|
|
func (wrapper *infoWrapper) Nodes() []string {
|
|
nodes := make([]string, len(wrapper.clusterInfo.Nodes))
|
|
|
|
for i := range nodes {
|
|
nodes[i] = wrapper.clusterInfo.Nodes[i].PrivateIP.String()
|
|
}
|
|
|
|
return nodes
|
|
}
|
|
|
|
func (wrapper *infoWrapper) NodesByType(t machine.Type) []string {
|
|
var nodes []string
|
|
|
|
for _, node := range wrapper.clusterInfo.Nodes {
|
|
if node.Type == t {
|
|
nodes = append(nodes, node.PrivateIP.String())
|
|
}
|
|
}
|
|
|
|
return nodes
|
|
}
|
|
|
|
// NewAdapter returns ClusterAccess object from Cluster.
|
|
func NewAdapter(clusterInfo provision.Cluster, opts ...provision.Option) *Adapter {
|
|
options := provision.DefaultOptions()
|
|
|
|
for _, opt := range opts {
|
|
if err := opt(&options); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
configProvider := cluster.ConfigClientProvider{
|
|
DefaultClient: options.TalosClient,
|
|
TalosConfig: options.TalosConfig,
|
|
}
|
|
|
|
return &Adapter{
|
|
ConfigClientProvider: configProvider,
|
|
KubernetesClient: cluster.KubernetesClient{
|
|
ClientProvider: &configProvider,
|
|
ForceEndpoint: options.ForceEndpoint,
|
|
},
|
|
APICrashDumper: cluster.APICrashDumper{
|
|
ClientProvider: &configProvider,
|
|
Info: &infoWrapper{clusterInfo: clusterInfo.Info()},
|
|
},
|
|
Info: &infoWrapper{clusterInfo: clusterInfo.Info()},
|
|
}
|
|
}
|