mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-08 22:21:16 +02:00
8d2be5e315
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>
93 lines
2.4 KiB
Go
93 lines
2.4 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 cluster provides functions to access, check and inspect Talos clusters.
|
|
package cluster
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/netip"
|
|
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/rest"
|
|
|
|
k8s "github.com/talos-systems/talos/pkg/kubernetes"
|
|
"github.com/talos-systems/talos/pkg/machinery/client"
|
|
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
|
|
)
|
|
|
|
// ClientProvider builds Talos client by endpoint.
|
|
//
|
|
// Client instance should be cached and closed when Close() is called.
|
|
type ClientProvider interface {
|
|
// Client returns Talos client instance for default (if no endpoints are given) or
|
|
// specific endpoint.
|
|
Client(endpoints ...string) (*client.Client, error)
|
|
// Close client connections.
|
|
Close() error
|
|
}
|
|
|
|
// K8sProvider builds Kubernetes client to access Talos cluster.
|
|
type K8sProvider interface {
|
|
Kubeconfig(ctx context.Context) ([]byte, error)
|
|
K8sRestConfig(ctx context.Context) (*rest.Config, error)
|
|
K8sClient(ctx context.Context) (*kubernetes.Clientset, error)
|
|
K8sHelper(ctx context.Context) (*k8s.Client, error)
|
|
K8sClose() error
|
|
}
|
|
|
|
// CrashDumper captures Talos cluster state to the specified writer for debugging.
|
|
type CrashDumper interface {
|
|
CrashDump(ctx context.Context, out io.Writer)
|
|
}
|
|
|
|
// NodeInfo describes a Talos node.
|
|
type NodeInfo struct {
|
|
InternalIP netip.Addr
|
|
IPs []netip.Addr
|
|
}
|
|
|
|
// Info describes the Talos cluster.
|
|
type Info interface {
|
|
// Nodes returns list of all node infos.
|
|
Nodes() []NodeInfo
|
|
// NodesByType return list of node endpoints by type.
|
|
NodesByType(machine.Type) []NodeInfo
|
|
}
|
|
|
|
// Bootstrapper performs Talos cluster bootstrap.
|
|
type Bootstrapper interface {
|
|
Bootstrap(ctx context.Context, out io.Writer) error
|
|
}
|
|
|
|
// IPsToNodeInfos converts list of IPs to a list of NodeInfos.
|
|
func IPsToNodeInfos(ips []string) ([]NodeInfo, error) {
|
|
result := make([]NodeInfo, len(ips))
|
|
|
|
for i, ip := range ips {
|
|
info, err := IPToNodeInfo(ip)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result[i] = *info
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// IPToNodeInfo converts a node internal IP to a NodeInfo.
|
|
func IPToNodeInfo(ip string) (*NodeInfo, error) {
|
|
parsed, err := netip.ParseAddr(ip)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &NodeInfo{
|
|
InternalIP: parsed,
|
|
IPs: []netip.Addr{parsed},
|
|
}, nil
|
|
}
|