mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-10 23:21:34 +02:00
6fc38bae69
This uses all available (potential) etcd endpoints, which includes the member being promoted as well. We avoid failures by iterating over the list of endpoints on each attempt to make sure each and every endpoint is tried. Part of #5889 Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
59 lines
1.8 KiB
Go
59 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 etcd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/cosi-project/runtime/pkg/resource"
|
|
"github.com/cosi-project/runtime/pkg/safe"
|
|
"github.com/cosi-project/runtime/pkg/state"
|
|
|
|
"github.com/talos-systems/talos/pkg/machinery/constants"
|
|
"github.com/talos-systems/talos/pkg/machinery/nethelpers"
|
|
"github.com/talos-systems/talos/pkg/machinery/resources/k8s"
|
|
)
|
|
|
|
// GetEndpoints returns expected endpoints of etcd cluster members.
|
|
//
|
|
// It is not guaranteed that etcd is running on each listed endpoint.
|
|
func GetEndpoints(ctx context.Context, resources state.State) ([]string, error) {
|
|
return getEndpoints(ctx, resources, "")
|
|
}
|
|
|
|
func getEndpoints(ctx context.Context, resources state.State, ignoreEndpointID string) ([]string, error) {
|
|
endpointResources, err := safe.StateList[*k8s.Endpoint](ctx, resources, resource.NewMetadata(k8s.ControlPlaneNamespaceName, k8s.EndpointType, "", resource.VersionUndefined))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting endpoints resources: %w", err)
|
|
}
|
|
|
|
iter := safe.IteratorFromList(endpointResources)
|
|
|
|
var endpointAddrs k8s.EndpointList
|
|
|
|
// merge all endpoints into a single list
|
|
for iter.Next() {
|
|
if iter.Value().Metadata().ID() == ignoreEndpointID {
|
|
continue
|
|
}
|
|
|
|
endpointAddrs = endpointAddrs.Merge(iter.Value())
|
|
}
|
|
|
|
if len(endpointAddrs) == 0 {
|
|
return nil, fmt.Errorf("no controlplane endpoints discovered yet")
|
|
}
|
|
|
|
endpoints := endpointAddrs.Strings()
|
|
|
|
// Etcd expects host:port format.
|
|
for i := 0; i < len(endpoints); i++ {
|
|
endpoints[i] = nethelpers.JoinHostPort(endpoints[i], constants.EtcdClientPort)
|
|
}
|
|
|
|
return endpoints, nil
|
|
}
|