tensorflow_serving_client/
model_service_grpc.rs

1// This file is generated. Do not edit
2// @generated
3
4// https://github.com/Manishearth/rust-clippy/issues/702
5#![allow(unknown_lints)]
6#![allow(clippy::all)]
7
8#![cfg_attr(rustfmt, rustfmt_skip)]
9
10#![allow(box_pointers)]
11#![allow(dead_code)]
12#![allow(missing_docs)]
13#![allow(non_camel_case_types)]
14#![allow(non_snake_case)]
15#![allow(non_upper_case_globals)]
16#![allow(trivial_casts)]
17#![allow(unsafe_code)]
18#![allow(unused_imports)]
19#![allow(unused_results)]
20
21const METHOD_MODEL_SERVICE_GET_MODEL_STATUS: ::grpcio::Method<super::get_model_status::GetModelStatusRequest, super::get_model_status::GetModelStatusResponse> = ::grpcio::Method {
22    ty: ::grpcio::MethodType::Unary,
23    name: "/tensorflow.serving.ModelService/GetModelStatus",
24    req_mar: ::grpcio::Marshaller { ser: ::grpcio::pb_ser, de: ::grpcio::pb_de },
25    resp_mar: ::grpcio::Marshaller { ser: ::grpcio::pb_ser, de: ::grpcio::pb_de },
26};
27
28const METHOD_MODEL_SERVICE_HANDLE_RELOAD_CONFIG_REQUEST: ::grpcio::Method<super::model_management::ReloadConfigRequest, super::model_management::ReloadConfigResponse> = ::grpcio::Method {
29    ty: ::grpcio::MethodType::Unary,
30    name: "/tensorflow.serving.ModelService/HandleReloadConfigRequest",
31    req_mar: ::grpcio::Marshaller { ser: ::grpcio::pb_ser, de: ::grpcio::pb_de },
32    resp_mar: ::grpcio::Marshaller { ser: ::grpcio::pb_ser, de: ::grpcio::pb_de },
33};
34
35#[derive(Clone)]
36pub struct ModelServiceClient {
37    client: ::grpcio::Client,
38}
39
40impl ModelServiceClient {
41    pub fn new(channel: ::grpcio::Channel) -> Self {
42        ModelServiceClient {
43            client: ::grpcio::Client::new(channel),
44        }
45    }
46
47    pub fn get_model_status_opt(&self, req: &super::get_model_status::GetModelStatusRequest, opt: ::grpcio::CallOption) -> ::grpcio::Result<super::get_model_status::GetModelStatusResponse> {
48        self.client.unary_call(&METHOD_MODEL_SERVICE_GET_MODEL_STATUS, req, opt)
49    }
50
51    pub fn get_model_status(&self, req: &super::get_model_status::GetModelStatusRequest) -> ::grpcio::Result<super::get_model_status::GetModelStatusResponse> {
52        self.get_model_status_opt(req, ::grpcio::CallOption::default())
53    }
54
55    pub fn get_model_status_async_opt(&self, req: &super::get_model_status::GetModelStatusRequest, opt: ::grpcio::CallOption) -> ::grpcio::Result<::grpcio::ClientUnaryReceiver<super::get_model_status::GetModelStatusResponse>> {
56        self.client.unary_call_async(&METHOD_MODEL_SERVICE_GET_MODEL_STATUS, req, opt)
57    }
58
59    pub fn get_model_status_async(&self, req: &super::get_model_status::GetModelStatusRequest) -> ::grpcio::Result<::grpcio::ClientUnaryReceiver<super::get_model_status::GetModelStatusResponse>> {
60        self.get_model_status_async_opt(req, ::grpcio::CallOption::default())
61    }
62
63    pub fn handle_reload_config_request_opt(&self, req: &super::model_management::ReloadConfigRequest, opt: ::grpcio::CallOption) -> ::grpcio::Result<super::model_management::ReloadConfigResponse> {
64        self.client.unary_call(&METHOD_MODEL_SERVICE_HANDLE_RELOAD_CONFIG_REQUEST, req, opt)
65    }
66
67    pub fn handle_reload_config_request(&self, req: &super::model_management::ReloadConfigRequest) -> ::grpcio::Result<super::model_management::ReloadConfigResponse> {
68        self.handle_reload_config_request_opt(req, ::grpcio::CallOption::default())
69    }
70
71    pub fn handle_reload_config_request_async_opt(&self, req: &super::model_management::ReloadConfigRequest, opt: ::grpcio::CallOption) -> ::grpcio::Result<::grpcio::ClientUnaryReceiver<super::model_management::ReloadConfigResponse>> {
72        self.client.unary_call_async(&METHOD_MODEL_SERVICE_HANDLE_RELOAD_CONFIG_REQUEST, req, opt)
73    }
74
75    pub fn handle_reload_config_request_async(&self, req: &super::model_management::ReloadConfigRequest) -> ::grpcio::Result<::grpcio::ClientUnaryReceiver<super::model_management::ReloadConfigResponse>> {
76        self.handle_reload_config_request_async_opt(req, ::grpcio::CallOption::default())
77    }
78    pub fn spawn<F>(&self, f: F) where F: ::futures::Future<Output = ()> + Send + 'static {
79        self.client.spawn(f)
80    }
81}
82
83pub trait ModelService {
84    fn get_model_status(&mut self, ctx: ::grpcio::RpcContext, req: super::get_model_status::GetModelStatusRequest, sink: ::grpcio::UnarySink<super::get_model_status::GetModelStatusResponse>);
85    fn handle_reload_config_request(&mut self, ctx: ::grpcio::RpcContext, req: super::model_management::ReloadConfigRequest, sink: ::grpcio::UnarySink<super::model_management::ReloadConfigResponse>);
86}
87
88pub fn create_model_service<S: ModelService + Send + Clone + 'static>(s: S) -> ::grpcio::Service {
89    let mut builder = ::grpcio::ServiceBuilder::new();
90    let mut instance = s.clone();
91    builder = builder.add_unary_handler(&METHOD_MODEL_SERVICE_GET_MODEL_STATUS, move |ctx, req, resp| {
92        instance.get_model_status(ctx, req, resp)
93    });
94    let mut instance = s;
95    builder = builder.add_unary_handler(&METHOD_MODEL_SERVICE_HANDLE_RELOAD_CONFIG_REQUEST, move |ctx, req, resp| {
96        instance.handle_reload_config_request(ctx, req, resp)
97    });
98    builder.build()
99}
tensorflow_serving_client/model_service_grpc.rs

tensorflow_serving_client/
model_service_grpc.rs
