41 lines
971 B
Go
41 lines
971 B
Go
package controlplane
|
|
|
|
import (
|
|
"context"
|
|
|
|
controlplanev1 "proxy-pool/gen/controlplane/v1"
|
|
)
|
|
|
|
type GeneratedClient struct {
|
|
client controlplanev1.CheckerControlPlaneClient
|
|
}
|
|
|
|
func NewGeneratedClient(client controlplanev1.CheckerControlPlaneClient) *GeneratedClient {
|
|
if client == nil {
|
|
return nil
|
|
}
|
|
return &GeneratedClient{client: client}
|
|
}
|
|
|
|
func (client *GeneratedClient) StreamCheckTasks(
|
|
ctx context.Context,
|
|
request *controlplanev1.StreamCheckTasksRequest,
|
|
) (TaskStream, error) {
|
|
if client == nil || client.client == nil {
|
|
return nil, ErrInvalidRunner
|
|
}
|
|
return client.client.StreamCheckTasks(ctx, request)
|
|
}
|
|
|
|
func (client *GeneratedClient) ReportObservations(
|
|
ctx context.Context,
|
|
batch *controlplanev1.ObservationBatch,
|
|
) (*controlplanev1.ReportObservationsResponse, error) {
|
|
if client == nil || client.client == nil {
|
|
return nil, ErrInvalidRunner
|
|
}
|
|
return client.client.ReportObservations(ctx, batch)
|
|
}
|
|
|
|
var _ Client = (*GeneratedClient)(nil)
|