mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 08:21:25 +01:00
e8ced2c2dd
(This is not user-facing, but rather internal use of the kubeconfig in the tests/inside the machine). This was added 4 years ago as a workaround, but instead of a global timeout we should rather use contexts with timeouts/deadlines (and we do!). Setting a global timeout breaks streaming Kubernetes pod logs. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
78 lines
1.8 KiB
Go
78 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/.
|
|
|
|
//go:build integration_k8s
|
|
|
|
package base
|
|
|
|
import (
|
|
"context"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
clientcmdapi "k8s.io/client-go/tools/clientcmd/api"
|
|
|
|
"github.com/siderolabs/talos/pkg/cluster"
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
"github.com/siderolabs/talos/pkg/machinery/constants"
|
|
)
|
|
|
|
//nolint:gocyclo
|
|
func discoverNodesK8s(ctx context.Context, client *client.Client, suite *TalosSuite) (cluster.Info, error) {
|
|
kubeconfig, err := client.Kubeconfig(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
config, err := clientcmd.BuildConfigFromKubeconfigGetter("", func() (*clientcmdapi.Config, error) {
|
|
return clientcmd.Load(kubeconfig)
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// patch timeout
|
|
if suite.K8sEndpoint != "" {
|
|
config.Host = suite.K8sEndpoint
|
|
}
|
|
|
|
clientset, err := kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
nodes, err := clientset.CoreV1().Nodes().List(ctx, metav1.ListOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var masterNodes, workerNodes []string
|
|
|
|
for _, node := range nodes.Items {
|
|
var address string
|
|
|
|
for _, nodeAddress := range node.Status.Addresses {
|
|
if nodeAddress.Type == v1.NodeInternalIP {
|
|
address = nodeAddress.Address
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
if address == "" {
|
|
continue
|
|
}
|
|
|
|
if _, ok := node.Labels[constants.LabelNodeRoleControlPlane]; ok {
|
|
masterNodes = append(masterNodes, address)
|
|
} else {
|
|
workerNodes = append(workerNodes, address)
|
|
}
|
|
}
|
|
|
|
return newNodeInfo(masterNodes, workerNodes)
|
|
}
|