mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-20 22:21:13 +02:00
ebd40bd0eb
There are few workarounds for Drone way of running integration test: DinD runs as a separate pod, and we can only access its exposed on the "host" ports, while from Talos cluster this endpoint is not reachable. So internally Talos nodes still use addresses like "10.5.0.2", while test is using "docker" to access it (that's name of the `docker` service in the pipeline). When running locally, 127.0.0.1 is used as endpoint, which should work fine both on OS X and Linux. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
120 lines
2.5 KiB
Go
120 lines
2.5 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 (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
clientcmdapi "k8s.io/client-go/tools/clientcmd/api"
|
|
|
|
"github.com/talos-systems/talos/cmd/osctl/pkg/client"
|
|
"github.com/talos-systems/talos/internal/pkg/provision"
|
|
"github.com/talos-systems/talos/pkg/constants"
|
|
)
|
|
|
|
// NewAdapter returns ClusterAccess object from Cluster.
|
|
func NewAdapter(cluster provision.Cluster, opts ...provision.Option) provision.ClusterAccess {
|
|
options := provision.DefaultOptions()
|
|
|
|
for _, opt := range opts {
|
|
if err := opt(&options); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
return &adapter{
|
|
Cluster: cluster,
|
|
clients: make(map[string]*client.Client),
|
|
options: &options,
|
|
}
|
|
}
|
|
|
|
type adapter struct {
|
|
provision.Cluster
|
|
|
|
clients map[string]*client.Client
|
|
clientset *kubernetes.Clientset
|
|
options *provision.Options
|
|
}
|
|
|
|
func (a *adapter) Client(endpoints ...string) (*client.Client, error) {
|
|
key := strings.Join(endpoints, ",")
|
|
|
|
if cli := a.clients[key]; cli != nil {
|
|
return cli, nil
|
|
}
|
|
|
|
configContext, creds, err := client.NewClientContextAndCredentialsFromParsedConfig(a.TalosConfig(), "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(endpoints) == 0 {
|
|
endpoints = configContext.Endpoints
|
|
}
|
|
|
|
client, err := client.NewClient(creds, endpoints, constants.ApidPort)
|
|
if err == nil {
|
|
a.clients[key] = client
|
|
}
|
|
|
|
return client, err
|
|
}
|
|
|
|
func (a *adapter) K8sClient(ctx context.Context) (*kubernetes.Clientset, error) {
|
|
if a.clientset != nil {
|
|
return a.clientset, nil
|
|
}
|
|
|
|
client, err := a.Client()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
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
|
|
config.Timeout = time.Minute
|
|
|
|
if a.options.ForceEndpoint != "" {
|
|
config.Host = fmt.Sprintf("%s:%d", a.options.ForceEndpoint, 6443)
|
|
}
|
|
|
|
clientset, err := kubernetes.NewForConfig(config)
|
|
if err == nil {
|
|
a.clientset = clientset
|
|
}
|
|
|
|
return clientset, err
|
|
}
|
|
|
|
func (a *adapter) Close() error {
|
|
for _, cli := range a.clients {
|
|
if err := cli.Close(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
a.clients = nil
|
|
a.clientset = nil
|
|
|
|
return nil
|
|
}
|