mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-18 21:21:10 +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>
37 lines
859 B
Go
37 lines
859 B
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 director_test
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/talos-systems/grpc-proxy/proxy"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type mockBackend struct {
|
|
target string
|
|
}
|
|
|
|
func (m *mockBackend) String() string {
|
|
return m.target
|
|
}
|
|
|
|
func (m *mockBackend) GetConnection(ctx context.Context) (context.Context, *grpc.ClientConn, error) {
|
|
return ctx, nil, nil
|
|
}
|
|
|
|
func (m *mockBackend) AppendInfo(streaming bool, resp []byte) ([]byte, error) {
|
|
return resp, nil
|
|
}
|
|
|
|
func (m *mockBackend) BuildError(streaming bool, err error) ([]byte, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func mockBackendFactory(target string) (proxy.Backend, error) {
|
|
return &mockBackend{target: target}, nil
|
|
}
|