mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-26 14:01:39 +01:00
8eacc4ba80
This allows to roll all nodes to use a new CA, to refresh it, or e.g. when the `talosconfig` was exposed accidentally. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
71 lines
1.6 KiB
Go
71 lines
1.6 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 backend
|
|
|
|
import (
|
|
"crypto/tls"
|
|
|
|
"github.com/siderolabs/gen/containers"
|
|
"github.com/siderolabs/grpc-proxy/proxy"
|
|
)
|
|
|
|
// APIDFactory caches connection to apid instances by target.
|
|
//
|
|
// TODO: need to clean up idle connections from time to time.
|
|
type APIDFactory struct {
|
|
cache containers.SyncMap[string, *APID]
|
|
provider TLSConfigProvider
|
|
}
|
|
|
|
// TLSConfigProvider provides tls.Config for client connections.
|
|
type TLSConfigProvider interface {
|
|
ClientConfig() (*tls.Config, error)
|
|
}
|
|
|
|
// NewAPIDFactory creates new APIDFactory with given tls.Config.
|
|
//
|
|
// Client TLS config is used to connect to other apid instances.
|
|
func NewAPIDFactory(provider TLSConfigProvider) *APIDFactory {
|
|
return &APIDFactory{
|
|
provider: provider,
|
|
}
|
|
}
|
|
|
|
// Get backend by target.
|
|
//
|
|
// Get performs caching of backends.
|
|
func (factory *APIDFactory) Get(target string) (proxy.Backend, error) {
|
|
b, ok := factory.cache.Load(target)
|
|
if ok {
|
|
return b, nil
|
|
}
|
|
|
|
backend, err := NewAPID(target, factory.provider.ClientConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
existing, loaded := factory.cache.LoadOrStore(target, backend)
|
|
if loaded {
|
|
// race: another Get() call built different backend
|
|
backend.Close()
|
|
|
|
return existing, nil
|
|
}
|
|
|
|
return backend, nil
|
|
}
|
|
|
|
// Flush all cached backends.
|
|
//
|
|
// This ensures that all connections are closed.
|
|
func (factory *APIDFactory) Flush() {
|
|
factory.cache.Range(func(key string, backend *APID) bool {
|
|
backend.Close()
|
|
|
|
return true
|
|
})
|
|
}
|