mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-21 14:41:12 +02:00
5b7bea2471
This replaces codegen version of apid proxying with talos-systems/grpc-proxy based version. Proxying is transparent, it doesn't require exact information about methods and response types. It requires some common layout response to enhance it properly with node metadata or errors. There should be no signifcant changes to the API with the previous version, but it's worth mentioning a few changes: 1. grpc.ClientConn is established just once per upstream (either local service or remote apid instance). 2. When called without `-t` (`targets`), apid proxies immediately down to local service skipping proxying to itself (as before), which results in empty node metadata in response (before it had local node IP). Might revert this later to proxy to itself (?). 3. Streaming APIs are now fully supported with multiple targets, but message definition doesn't contain `ResponseMetadata`, so streaming APIs are broken now with targets (needs a fix). 4. Errors are now returned as responses with `Error` field set in `ResponseMetadata`, this requires client library update and `osctl` to handle it properly. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
56 lines
1.3 KiB
Go
56 lines
1.3 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"
|
|
"sync"
|
|
|
|
"github.com/talos-systems/grpc-proxy/proxy"
|
|
"google.golang.org/grpc/credentials"
|
|
)
|
|
|
|
// APIDFactory caches connection to apid instances by target.
|
|
//
|
|
// TODO: need to clean up idle connections from time to time.
|
|
type APIDFactory struct {
|
|
cache sync.Map
|
|
creds credentials.TransportCredentials
|
|
}
|
|
|
|
// NewAPIDFactory creates new APIDFactory with given tls.Config.
|
|
//
|
|
// Client TLS config is used to connect to other apid instances.
|
|
func NewAPIDFactory(config *tls.Config) *APIDFactory {
|
|
return &APIDFactory{
|
|
creds: credentials.NewTLS(config),
|
|
}
|
|
}
|
|
|
|
// 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.(proxy.Backend), nil
|
|
}
|
|
|
|
backend, err := NewAPID(target, factory.creds)
|
|
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.(proxy.Backend), nil
|
|
}
|
|
|
|
return backend, nil
|
|
}
|