/
trainer_grpc.pb.go
140 lines (118 loc) · 3.99 KB
/
trainer_grpc.pb.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
// versions:
// - protoc-gen-go-grpc v1.2.0
// - protoc v3.21.6
// source: pkg/apis/trainer/v1/trainer.proto
package trainer
import (
context "context"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
emptypb "google.golang.org/protobuf/types/known/emptypb"
)
// This is a compile-time assertion to ensure that this generated file
// is compatible with the grpc package it is being compiled against.
// Requires gRPC-Go v1.32.0 or later.
const _ = grpc.SupportPackageIsVersion7
// TrainerClient is the client API for Trainer service.
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type TrainerClient interface {
// Train trains models of scheduler using dataset.
Train(ctx context.Context, opts ...grpc.CallOption) (Trainer_TrainClient, error)
}
type trainerClient struct {
cc grpc.ClientConnInterface
}
func NewTrainerClient(cc grpc.ClientConnInterface) TrainerClient {
return &trainerClient{cc}
}
func (c *trainerClient) Train(ctx context.Context, opts ...grpc.CallOption) (Trainer_TrainClient, error) {
stream, err := c.cc.NewStream(ctx, &Trainer_ServiceDesc.Streams[0], "/trainer.v1.Trainer/Train", opts...)
if err != nil {
return nil, err
}
x := &trainerTrainClient{stream}
return x, nil
}
type Trainer_TrainClient interface {
Send(*TrainRequest) error
CloseAndRecv() (*emptypb.Empty, error)
grpc.ClientStream
}
type trainerTrainClient struct {
grpc.ClientStream
}
func (x *trainerTrainClient) Send(m *TrainRequest) error {
return x.ClientStream.SendMsg(m)
}
func (x *trainerTrainClient) CloseAndRecv() (*emptypb.Empty, error) {
if err := x.ClientStream.CloseSend(); err != nil {
return nil, err
}
m := new(emptypb.Empty)
if err := x.ClientStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
// TrainerServer is the server API for Trainer service.
// All implementations should embed UnimplementedTrainerServer
// for forward compatibility
type TrainerServer interface {
// Train trains models of scheduler using dataset.
Train(Trainer_TrainServer) error
}
// UnimplementedTrainerServer should be embedded to have forward compatible implementations.
type UnimplementedTrainerServer struct {
}
func (UnimplementedTrainerServer) Train(Trainer_TrainServer) error {
return status.Errorf(codes.Unimplemented, "method Train not implemented")
}
// UnsafeTrainerServer may be embedded to opt out of forward compatibility for this service.
// Use of this interface is not recommended, as added methods to TrainerServer will
// result in compilation errors.
type UnsafeTrainerServer interface {
mustEmbedUnimplementedTrainerServer()
}
func RegisterTrainerServer(s grpc.ServiceRegistrar, srv TrainerServer) {
s.RegisterService(&Trainer_ServiceDesc, srv)
}
func _Trainer_Train_Handler(srv interface{}, stream grpc.ServerStream) error {
return srv.(TrainerServer).Train(&trainerTrainServer{stream})
}
type Trainer_TrainServer interface {
SendAndClose(*emptypb.Empty) error
Recv() (*TrainRequest, error)
grpc.ServerStream
}
type trainerTrainServer struct {
grpc.ServerStream
}
func (x *trainerTrainServer) SendAndClose(m *emptypb.Empty) error {
return x.ServerStream.SendMsg(m)
}
func (x *trainerTrainServer) Recv() (*TrainRequest, error) {
m := new(TrainRequest)
if err := x.ServerStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
// Trainer_ServiceDesc is the grpc.ServiceDesc for Trainer service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
var Trainer_ServiceDesc = grpc.ServiceDesc{
ServiceName: "trainer.v1.Trainer",
HandlerType: (*TrainerServer)(nil),
Methods: []grpc.MethodDesc{},
Streams: []grpc.StreamDesc{
{
StreamName: "Train",
Handler: _Trainer_Train_Handler,
ClientStreams: true,
},
},
Metadata: "pkg/apis/trainer/v1/trainer.proto",
}