1
2
3
4
5
6
7 package agent
8
9 import (
10 context "context"
11 grpc "google.golang.org/grpc"
12 codes "google.golang.org/grpc/codes"
13 status "google.golang.org/grpc/status"
14 )
15
16
17
18
19 const _ = grpc.SupportPackageIsVersion7
20
21
22
23
24 type DirectorClient interface {
25
26
27
28 Report(ctx context.Context, in *Snapshot, opts ...grpc.CallOption) (*SnapshotResponse, error)
29
30 ReportStream(ctx context.Context, opts ...grpc.CallOption) (Director_ReportStreamClient, error)
31
32 StreamDiagnostics(ctx context.Context, opts ...grpc.CallOption) (Director_StreamDiagnosticsClient, error)
33
34 StreamMetrics(ctx context.Context, opts ...grpc.CallOption) (Director_StreamMetricsClient, error)
35
36 Retrieve(ctx context.Context, in *Identity, opts ...grpc.CallOption) (Director_RetrieveClient, error)
37
38 ReportCommandResult(ctx context.Context, in *CommandResult, opts ...grpc.CallOption) (*CommandResultResponse, error)
39 }
40
41 type directorClient struct {
42 cc grpc.ClientConnInterface
43 }
44
45 func NewDirectorClient(cc grpc.ClientConnInterface) DirectorClient {
46 return &directorClient{cc}
47 }
48
49
50 func (c *directorClient) Report(ctx context.Context, in *Snapshot, opts ...grpc.CallOption) (*SnapshotResponse, error) {
51 out := new(SnapshotResponse)
52 err := c.cc.Invoke(ctx, "/agent.Director/Report", in, out, opts...)
53 if err != nil {
54 return nil, err
55 }
56 return out, nil
57 }
58
59 func (c *directorClient) ReportStream(ctx context.Context, opts ...grpc.CallOption) (Director_ReportStreamClient, error) {
60 stream, err := c.cc.NewStream(ctx, &Director_ServiceDesc.Streams[0], "/agent.Director/ReportStream", opts...)
61 if err != nil {
62 return nil, err
63 }
64 x := &directorReportStreamClient{stream}
65 return x, nil
66 }
67
68 type Director_ReportStreamClient interface {
69 Send(*RawSnapshotChunk) error
70 CloseAndRecv() (*SnapshotResponse, error)
71 grpc.ClientStream
72 }
73
74 type directorReportStreamClient struct {
75 grpc.ClientStream
76 }
77
78 func (x *directorReportStreamClient) Send(m *RawSnapshotChunk) error {
79 return x.ClientStream.SendMsg(m)
80 }
81
82 func (x *directorReportStreamClient) CloseAndRecv() (*SnapshotResponse, error) {
83 if err := x.ClientStream.CloseSend(); err != nil {
84 return nil, err
85 }
86 m := new(SnapshotResponse)
87 if err := x.ClientStream.RecvMsg(m); err != nil {
88 return nil, err
89 }
90 return m, nil
91 }
92
93 func (c *directorClient) StreamDiagnostics(ctx context.Context, opts ...grpc.CallOption) (Director_StreamDiagnosticsClient, error) {
94 stream, err := c.cc.NewStream(ctx, &Director_ServiceDesc.Streams[1], "/agent.Director/StreamDiagnostics", opts...)
95 if err != nil {
96 return nil, err
97 }
98 x := &directorStreamDiagnosticsClient{stream}
99 return x, nil
100 }
101
102 type Director_StreamDiagnosticsClient interface {
103 Send(*RawDiagnosticsChunk) error
104 CloseAndRecv() (*DiagnosticsResponse, error)
105 grpc.ClientStream
106 }
107
108 type directorStreamDiagnosticsClient struct {
109 grpc.ClientStream
110 }
111
112 func (x *directorStreamDiagnosticsClient) Send(m *RawDiagnosticsChunk) error {
113 return x.ClientStream.SendMsg(m)
114 }
115
116 func (x *directorStreamDiagnosticsClient) CloseAndRecv() (*DiagnosticsResponse, error) {
117 if err := x.ClientStream.CloseSend(); err != nil {
118 return nil, err
119 }
120 m := new(DiagnosticsResponse)
121 if err := x.ClientStream.RecvMsg(m); err != nil {
122 return nil, err
123 }
124 return m, nil
125 }
126
127 func (c *directorClient) StreamMetrics(ctx context.Context, opts ...grpc.CallOption) (Director_StreamMetricsClient, error) {
128 stream, err := c.cc.NewStream(ctx, &Director_ServiceDesc.Streams[2], "/agent.Director/StreamMetrics", opts...)
129 if err != nil {
130 return nil, err
131 }
132 x := &directorStreamMetricsClient{stream}
133 return x, nil
134 }
135
136 type Director_StreamMetricsClient interface {
137 Send(*StreamMetricsMessage) error
138 CloseAndRecv() (*StreamMetricsResponse, error)
139 grpc.ClientStream
140 }
141
142 type directorStreamMetricsClient struct {
143 grpc.ClientStream
144 }
145
146 func (x *directorStreamMetricsClient) Send(m *StreamMetricsMessage) error {
147 return x.ClientStream.SendMsg(m)
148 }
149
150 func (x *directorStreamMetricsClient) CloseAndRecv() (*StreamMetricsResponse, error) {
151 if err := x.ClientStream.CloseSend(); err != nil {
152 return nil, err
153 }
154 m := new(StreamMetricsResponse)
155 if err := x.ClientStream.RecvMsg(m); err != nil {
156 return nil, err
157 }
158 return m, nil
159 }
160
161 func (c *directorClient) Retrieve(ctx context.Context, in *Identity, opts ...grpc.CallOption) (Director_RetrieveClient, error) {
162 stream, err := c.cc.NewStream(ctx, &Director_ServiceDesc.Streams[3], "/agent.Director/Retrieve", opts...)
163 if err != nil {
164 return nil, err
165 }
166 x := &directorRetrieveClient{stream}
167 if err := x.ClientStream.SendMsg(in); err != nil {
168 return nil, err
169 }
170 if err := x.ClientStream.CloseSend(); err != nil {
171 return nil, err
172 }
173 return x, nil
174 }
175
176 type Director_RetrieveClient interface {
177 Recv() (*Directive, error)
178 grpc.ClientStream
179 }
180
181 type directorRetrieveClient struct {
182 grpc.ClientStream
183 }
184
185 func (x *directorRetrieveClient) Recv() (*Directive, error) {
186 m := new(Directive)
187 if err := x.ClientStream.RecvMsg(m); err != nil {
188 return nil, err
189 }
190 return m, nil
191 }
192
193 func (c *directorClient) ReportCommandResult(ctx context.Context, in *CommandResult, opts ...grpc.CallOption) (*CommandResultResponse, error) {
194 out := new(CommandResultResponse)
195 err := c.cc.Invoke(ctx, "/agent.Director/ReportCommandResult", in, out, opts...)
196 if err != nil {
197 return nil, err
198 }
199 return out, nil
200 }
201
202
203
204
205 type DirectorServer interface {
206
207
208
209 Report(context.Context, *Snapshot) (*SnapshotResponse, error)
210
211 ReportStream(Director_ReportStreamServer) error
212
213 StreamDiagnostics(Director_StreamDiagnosticsServer) error
214
215 StreamMetrics(Director_StreamMetricsServer) error
216
217 Retrieve(*Identity, Director_RetrieveServer) error
218
219 ReportCommandResult(context.Context, *CommandResult) (*CommandResultResponse, error)
220 mustEmbedUnimplementedDirectorServer()
221 }
222
223
224 type UnimplementedDirectorServer struct {
225 }
226
227 func (UnimplementedDirectorServer) Report(context.Context, *Snapshot) (*SnapshotResponse, error) {
228 return nil, status.Errorf(codes.Unimplemented, "method Report not implemented")
229 }
230 func (UnimplementedDirectorServer) ReportStream(Director_ReportStreamServer) error {
231 return status.Errorf(codes.Unimplemented, "method ReportStream not implemented")
232 }
233 func (UnimplementedDirectorServer) StreamDiagnostics(Director_StreamDiagnosticsServer) error {
234 return status.Errorf(codes.Unimplemented, "method StreamDiagnostics not implemented")
235 }
236 func (UnimplementedDirectorServer) StreamMetrics(Director_StreamMetricsServer) error {
237 return status.Errorf(codes.Unimplemented, "method StreamMetrics not implemented")
238 }
239 func (UnimplementedDirectorServer) Retrieve(*Identity, Director_RetrieveServer) error {
240 return status.Errorf(codes.Unimplemented, "method Retrieve not implemented")
241 }
242 func (UnimplementedDirectorServer) ReportCommandResult(context.Context, *CommandResult) (*CommandResultResponse, error) {
243 return nil, status.Errorf(codes.Unimplemented, "method ReportCommandResult not implemented")
244 }
245 func (UnimplementedDirectorServer) mustEmbedUnimplementedDirectorServer() {}
246
247
248
249
250 type UnsafeDirectorServer interface {
251 mustEmbedUnimplementedDirectorServer()
252 }
253
254 func RegisterDirectorServer(s grpc.ServiceRegistrar, srv DirectorServer) {
255 s.RegisterService(&Director_ServiceDesc, srv)
256 }
257
258 func _Director_Report_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
259 in := new(Snapshot)
260 if err := dec(in); err != nil {
261 return nil, err
262 }
263 if interceptor == nil {
264 return srv.(DirectorServer).Report(ctx, in)
265 }
266 info := &grpc.UnaryServerInfo{
267 Server: srv,
268 FullMethod: "/agent.Director/Report",
269 }
270 handler := func(ctx context.Context, req interface{}) (interface{}, error) {
271 return srv.(DirectorServer).Report(ctx, req.(*Snapshot))
272 }
273 return interceptor(ctx, in, info, handler)
274 }
275
276 func _Director_ReportStream_Handler(srv interface{}, stream grpc.ServerStream) error {
277 return srv.(DirectorServer).ReportStream(&directorReportStreamServer{stream})
278 }
279
280 type Director_ReportStreamServer interface {
281 SendAndClose(*SnapshotResponse) error
282 Recv() (*RawSnapshotChunk, error)
283 grpc.ServerStream
284 }
285
286 type directorReportStreamServer struct {
287 grpc.ServerStream
288 }
289
290 func (x *directorReportStreamServer) SendAndClose(m *SnapshotResponse) error {
291 return x.ServerStream.SendMsg(m)
292 }
293
294 func (x *directorReportStreamServer) Recv() (*RawSnapshotChunk, error) {
295 m := new(RawSnapshotChunk)
296 if err := x.ServerStream.RecvMsg(m); err != nil {
297 return nil, err
298 }
299 return m, nil
300 }
301
302 func _Director_StreamDiagnostics_Handler(srv interface{}, stream grpc.ServerStream) error {
303 return srv.(DirectorServer).StreamDiagnostics(&directorStreamDiagnosticsServer{stream})
304 }
305
306 type Director_StreamDiagnosticsServer interface {
307 SendAndClose(*DiagnosticsResponse) error
308 Recv() (*RawDiagnosticsChunk, error)
309 grpc.ServerStream
310 }
311
312 type directorStreamDiagnosticsServer struct {
313 grpc.ServerStream
314 }
315
316 func (x *directorStreamDiagnosticsServer) SendAndClose(m *DiagnosticsResponse) error {
317 return x.ServerStream.SendMsg(m)
318 }
319
320 func (x *directorStreamDiagnosticsServer) Recv() (*RawDiagnosticsChunk, error) {
321 m := new(RawDiagnosticsChunk)
322 if err := x.ServerStream.RecvMsg(m); err != nil {
323 return nil, err
324 }
325 return m, nil
326 }
327
328 func _Director_StreamMetrics_Handler(srv interface{}, stream grpc.ServerStream) error {
329 return srv.(DirectorServer).StreamMetrics(&directorStreamMetricsServer{stream})
330 }
331
332 type Director_StreamMetricsServer interface {
333 SendAndClose(*StreamMetricsResponse) error
334 Recv() (*StreamMetricsMessage, error)
335 grpc.ServerStream
336 }
337
338 type directorStreamMetricsServer struct {
339 grpc.ServerStream
340 }
341
342 func (x *directorStreamMetricsServer) SendAndClose(m *StreamMetricsResponse) error {
343 return x.ServerStream.SendMsg(m)
344 }
345
346 func (x *directorStreamMetricsServer) Recv() (*StreamMetricsMessage, error) {
347 m := new(StreamMetricsMessage)
348 if err := x.ServerStream.RecvMsg(m); err != nil {
349 return nil, err
350 }
351 return m, nil
352 }
353
354 func _Director_Retrieve_Handler(srv interface{}, stream grpc.ServerStream) error {
355 m := new(Identity)
356 if err := stream.RecvMsg(m); err != nil {
357 return err
358 }
359 return srv.(DirectorServer).Retrieve(m, &directorRetrieveServer{stream})
360 }
361
362 type Director_RetrieveServer interface {
363 Send(*Directive) error
364 grpc.ServerStream
365 }
366
367 type directorRetrieveServer struct {
368 grpc.ServerStream
369 }
370
371 func (x *directorRetrieveServer) Send(m *Directive) error {
372 return x.ServerStream.SendMsg(m)
373 }
374
375 func _Director_ReportCommandResult_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
376 in := new(CommandResult)
377 if err := dec(in); err != nil {
378 return nil, err
379 }
380 if interceptor == nil {
381 return srv.(DirectorServer).ReportCommandResult(ctx, in)
382 }
383 info := &grpc.UnaryServerInfo{
384 Server: srv,
385 FullMethod: "/agent.Director/ReportCommandResult",
386 }
387 handler := func(ctx context.Context, req interface{}) (interface{}, error) {
388 return srv.(DirectorServer).ReportCommandResult(ctx, req.(*CommandResult))
389 }
390 return interceptor(ctx, in, info, handler)
391 }
392
393
394
395
396 var Director_ServiceDesc = grpc.ServiceDesc{
397 ServiceName: "agent.Director",
398 HandlerType: (*DirectorServer)(nil),
399 Methods: []grpc.MethodDesc{
400 {
401 MethodName: "Report",
402 Handler: _Director_Report_Handler,
403 },
404 {
405 MethodName: "ReportCommandResult",
406 Handler: _Director_ReportCommandResult_Handler,
407 },
408 },
409 Streams: []grpc.StreamDesc{
410 {
411 StreamName: "ReportStream",
412 Handler: _Director_ReportStream_Handler,
413 ClientStreams: true,
414 },
415 {
416 StreamName: "StreamDiagnostics",
417 Handler: _Director_StreamDiagnostics_Handler,
418 ClientStreams: true,
419 },
420 {
421 StreamName: "StreamMetrics",
422 Handler: _Director_StreamMetrics_Handler,
423 ClientStreams: true,
424 },
425 {
426 StreamName: "Retrieve",
427 Handler: _Director_Retrieve_Handler,
428 ServerStreams: true,
429 },
430 },
431 Metadata: "agent/director.proto",
432 }
433
View as plain text