mirror of https://github.com/grpc/grpc.git
Craig Tiller
8 years ago
commit
f6221516f4
16 changed files with 959 additions and 559 deletions
@ -0,0 +1,210 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H |
||||
#define GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H |
||||
|
||||
#include "src/compiler/config.h" |
||||
#include "src/compiler/cpp_generator_helpers.h" |
||||
#include "src/compiler/python_generator_helpers.h" |
||||
#include "src/compiler/python_private_generator.h" |
||||
#include "src/compiler/schema_interface.h" |
||||
|
||||
#include <vector> |
||||
|
||||
// Get leading or trailing comments in a string.
|
||||
template <typename DescriptorType> |
||||
inline grpc::string GetCommentsHelper(const DescriptorType *desc, bool leading, |
||||
const grpc::string &prefix) { |
||||
return grpc_generator::GetPrefixedComments(desc, leading, prefix); |
||||
} |
||||
|
||||
class ProtoBufMethod : public grpc_generator::Method { |
||||
public: |
||||
ProtoBufMethod(const grpc::protobuf::MethodDescriptor *method) |
||||
: method_(method) {} |
||||
|
||||
grpc::string name() const { return method_->name(); } |
||||
|
||||
grpc::string input_type_name() const { |
||||
return grpc_cpp_generator::ClassName(method_->input_type(), true); |
||||
} |
||||
grpc::string output_type_name() const { |
||||
return grpc_cpp_generator::ClassName(method_->output_type(), true); |
||||
} |
||||
|
||||
grpc::string get_input_type_name() const { |
||||
return method_->input_type()->file()->name(); |
||||
} |
||||
grpc::string get_output_type_name() const { |
||||
return method_->output_type()->file()->name(); |
||||
} |
||||
|
||||
bool get_module_and_message_path_input(grpc::string *str, |
||||
grpc::string generator_file_name, |
||||
bool generate_in_pb2_grpc, |
||||
grpc::string import_prefix) const { |
||||
return grpc_python_generator::GetModuleAndMessagePath( |
||||
method_->input_type(), str, generator_file_name, generate_in_pb2_grpc, |
||||
import_prefix); |
||||
} |
||||
|
||||
bool get_module_and_message_path_output(grpc::string *str, |
||||
grpc::string generator_file_name, |
||||
bool generate_in_pb2_grpc, |
||||
grpc::string import_prefix) const { |
||||
return grpc_python_generator::GetModuleAndMessagePath( |
||||
method_->output_type(), str, generator_file_name, generate_in_pb2_grpc, |
||||
import_prefix); |
||||
} |
||||
|
||||
bool NoStreaming() const { |
||||
return !method_->client_streaming() && !method_->server_streaming(); |
||||
} |
||||
|
||||
bool ClientStreaming() const { return method_->client_streaming(); } |
||||
|
||||
bool ServerStreaming() const { return method_->server_streaming(); } |
||||
|
||||
bool BidiStreaming() const { |
||||
return method_->client_streaming() && method_->server_streaming(); |
||||
} |
||||
|
||||
grpc::string GetLeadingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(method_, true, prefix); |
||||
} |
||||
|
||||
grpc::string GetTrailingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(method_, false, prefix); |
||||
} |
||||
|
||||
vector<grpc::string> GetAllComments() const { |
||||
return grpc_python_generator::get_all_comments(method_); |
||||
} |
||||
|
||||
private: |
||||
const grpc::protobuf::MethodDescriptor *method_; |
||||
}; |
||||
|
||||
class ProtoBufService : public grpc_generator::Service { |
||||
public: |
||||
ProtoBufService(const grpc::protobuf::ServiceDescriptor *service) |
||||
: service_(service) {} |
||||
|
||||
grpc::string name() const { return service_->name(); } |
||||
|
||||
int method_count() const { return service_->method_count(); }; |
||||
std::unique_ptr<const grpc_generator::Method> method(int i) const { |
||||
return std::unique_ptr<const grpc_generator::Method>( |
||||
new ProtoBufMethod(service_->method(i))); |
||||
}; |
||||
|
||||
grpc::string GetLeadingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(service_, true, prefix); |
||||
} |
||||
|
||||
grpc::string GetTrailingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(service_, false, prefix); |
||||
} |
||||
|
||||
vector<grpc::string> GetAllComments() const { |
||||
return grpc_python_generator::get_all_comments(service_); |
||||
} |
||||
|
||||
private: |
||||
const grpc::protobuf::ServiceDescriptor *service_; |
||||
}; |
||||
|
||||
class ProtoBufPrinter : public grpc_generator::Printer { |
||||
public: |
||||
ProtoBufPrinter(grpc::string *str) |
||||
: output_stream_(str), printer_(&output_stream_, '$') {} |
||||
|
||||
void Print(const std::map<grpc::string, grpc::string> &vars, |
||||
const char *string_template) { |
||||
printer_.Print(vars, string_template); |
||||
} |
||||
|
||||
void Print(const char *string) { printer_.Print(string); } |
||||
void Indent() { printer_.Indent(); } |
||||
void Outdent() { printer_.Outdent(); } |
||||
|
||||
private: |
||||
grpc::protobuf::io::StringOutputStream output_stream_; |
||||
grpc::protobuf::io::Printer printer_; |
||||
}; |
||||
|
||||
class ProtoBufFile : public grpc_generator::File { |
||||
public: |
||||
ProtoBufFile(const grpc::protobuf::FileDescriptor *file) : file_(file) {} |
||||
|
||||
grpc::string filename() const { return file_->name(); } |
||||
grpc::string filename_without_ext() const { |
||||
return grpc_generator::StripProto(filename()); |
||||
} |
||||
|
||||
grpc::string package() const { return file_->package(); } |
||||
std::vector<grpc::string> package_parts() const { |
||||
return grpc_generator::tokenize(package(), "."); |
||||
} |
||||
|
||||
grpc::string additional_headers() const { return ""; } |
||||
|
||||
int service_count() const { return file_->service_count(); }; |
||||
std::unique_ptr<const grpc_generator::Service> service(int i) const { |
||||
return std::unique_ptr<const grpc_generator::Service>( |
||||
new ProtoBufService(file_->service(i))); |
||||
} |
||||
|
||||
std::unique_ptr<grpc_generator::Printer> CreatePrinter( |
||||
grpc::string *str) const { |
||||
return std::unique_ptr<grpc_generator::Printer>(new ProtoBufPrinter(str)); |
||||
} |
||||
|
||||
grpc::string GetLeadingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(file_, true, prefix); |
||||
} |
||||
|
||||
grpc::string GetTrailingComments(const grpc::string prefix) const { |
||||
return GetCommentsHelper(file_, false, prefix); |
||||
} |
||||
|
||||
vector<grpc::string> GetAllComments() const { |
||||
return grpc_python_generator::get_all_comments(file_); |
||||
} |
||||
|
||||
private: |
||||
const grpc::protobuf::FileDescriptor *file_; |
||||
}; |
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H
|
||||
@ -0,0 +1,142 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_HELPERS_H |
||||
#define GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_HELPERS_H |
||||
|
||||
#include <cstring> |
||||
#include <fstream> |
||||
#include <iostream> |
||||
#include <vector> |
||||
|
||||
#include "src/compiler/config.h" |
||||
#include "src/compiler/generator_helpers.h" |
||||
#include "src/compiler/python_generator.h" |
||||
#include "src/compiler/python_private_generator.h" |
||||
|
||||
using std::vector; |
||||
using grpc_generator::StringReplace; |
||||
using grpc_generator::StripProto; |
||||
using grpc::protobuf::Descriptor; |
||||
using grpc::protobuf::FileDescriptor; |
||||
using grpc::protobuf::MethodDescriptor; |
||||
using grpc::protobuf::ServiceDescriptor; |
||||
using grpc::protobuf::compiler::GeneratorContext; |
||||
using grpc::protobuf::io::CodedOutputStream; |
||||
using grpc::protobuf::io::Printer; |
||||
using grpc::protobuf::io::StringOutputStream; |
||||
using grpc::protobuf::io::ZeroCopyOutputStream; |
||||
|
||||
namespace grpc_python_generator { |
||||
|
||||
namespace { |
||||
|
||||
typedef vector<const Descriptor*> DescriptorVector; |
||||
typedef vector<grpc::string> StringVector; |
||||
|
||||
// TODO(https://github.com/google/protobuf/issues/888):
|
||||
// Export `ModuleName` from protobuf's
|
||||
// `src/google/protobuf/compiler/python/python_generator.cc` file.
|
||||
grpc::string ModuleName(const grpc::string& filename, |
||||
const grpc::string& import_prefix) { |
||||
grpc::string basename = StripProto(filename); |
||||
basename = StringReplace(basename, "-", "_"); |
||||
basename = StringReplace(basename, "/", "."); |
||||
return import_prefix + basename + "_pb2"; |
||||
} |
||||
|
||||
// TODO(https://github.com/google/protobuf/issues/888):
|
||||
// Export `ModuleAlias` from protobuf's
|
||||
// `src/google/protobuf/compiler/python/python_generator.cc` file.
|
||||
grpc::string ModuleAlias(const grpc::string& filename, |
||||
const grpc::string& import_prefix) { |
||||
grpc::string module_name = ModuleName(filename, import_prefix); |
||||
// We can't have dots in the module name, so we replace each with _dot_.
|
||||
// But that could lead to a collision between a.b and a_dot_b, so we also
|
||||
// duplicate each underscore.
|
||||
module_name = StringReplace(module_name, "_", "__"); |
||||
module_name = StringReplace(module_name, ".", "_dot_"); |
||||
return module_name; |
||||
} |
||||
|
||||
bool GetModuleAndMessagePath(const Descriptor* type, grpc::string* out, |
||||
grpc::string generator_file_name, |
||||
bool generate_in_pb2_grpc, |
||||
grpc::string& import_prefix) { |
||||
const Descriptor* path_elem_type = type; |
||||
DescriptorVector message_path; |
||||
do { |
||||
message_path.push_back(path_elem_type); |
||||
path_elem_type = path_elem_type->containing_type(); |
||||
} while (path_elem_type); // implicit nullptr comparison; don't be explicit
|
||||
grpc::string file_name = type->file()->name(); |
||||
static const int proto_suffix_length = strlen(".proto"); |
||||
if (!(file_name.size() > static_cast<size_t>(proto_suffix_length) && |
||||
file_name.find_last_of(".proto") == file_name.size() - 1)) { |
||||
return false; |
||||
} |
||||
|
||||
grpc::string module; |
||||
if (generator_file_name != file_name || generate_in_pb2_grpc) { |
||||
module = ModuleAlias(file_name, import_prefix) + "."; |
||||
} else { |
||||
module = ""; |
||||
} |
||||
grpc::string message_type; |
||||
for (DescriptorVector::reverse_iterator path_iter = message_path.rbegin(); |
||||
path_iter != message_path.rend(); ++path_iter) { |
||||
message_type += (*path_iter)->name() + "."; |
||||
} |
||||
// no pop_back prior to C++11
|
||||
message_type.resize(message_type.size() - 1); |
||||
*out = module + message_type; |
||||
return true; |
||||
} |
||||
|
||||
template <typename DescriptorType> |
||||
StringVector get_all_comments(const DescriptorType* descriptor) { |
||||
StringVector comments; |
||||
grpc_generator::GetComment( |
||||
descriptor, grpc_generator::COMMENTTYPE_LEADING_DETACHED, &comments); |
||||
grpc_generator::GetComment(descriptor, grpc_generator::COMMENTTYPE_LEADING, |
||||
&comments); |
||||
grpc_generator::GetComment(descriptor, grpc_generator::COMMENTTYPE_TRAILING, |
||||
&comments); |
||||
return comments; |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
} // namespace grpc_python_generator
|
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_HELPERS_H
|
||||
@ -0,0 +1,99 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_PYTHON_PRIVATE_GENERATOR_H |
||||
#define GRPC_INTERNAL_COMPILER_PYTHON_PRIVATE_GENERATOR_H |
||||
|
||||
#include <iostream> |
||||
#include <vector> |
||||
|
||||
#include "src/compiler/python_generator.h" |
||||
#include "src/compiler/schema_interface.h" |
||||
|
||||
namespace grpc_python_generator { |
||||
|
||||
namespace { |
||||
|
||||
// Tucks all generator state in an anonymous namespace away from
|
||||
// PythonGrpcGenerator and the header file, mostly to encourage future changes
|
||||
// to not require updates to the grpcio-tools C++ code part. Assumes that it is
|
||||
// only ever used from a single thread.
|
||||
struct PrivateGenerator { |
||||
const GeneratorConfiguration& config; |
||||
const grpc_generator::File* file; |
||||
|
||||
bool generate_in_pb2_grpc; |
||||
|
||||
PrivateGenerator(const GeneratorConfiguration& config, |
||||
const grpc_generator::File* file); |
||||
|
||||
std::pair<bool, grpc::string> GetGrpcServices(); |
||||
|
||||
private: |
||||
bool PrintPreamble(grpc_generator::Printer* out); |
||||
bool PrintBetaPreamble(grpc_generator::Printer* out); |
||||
bool PrintGAServices(grpc_generator::Printer* out); |
||||
bool PrintBetaServices(grpc_generator::Printer* out); |
||||
|
||||
bool PrintAddServicerToServer( |
||||
const grpc::string& package_qualified_service_name, |
||||
const grpc_generator::Service* service, grpc_generator::Printer* out); |
||||
bool PrintServicer(const grpc_generator::Service* service, |
||||
grpc_generator::Printer* out); |
||||
bool PrintStub(const grpc::string& package_qualified_service_name, |
||||
const grpc_generator::Service* service, |
||||
grpc_generator::Printer* out); |
||||
|
||||
bool PrintBetaServicer(const grpc_generator::Service* service, |
||||
grpc_generator::Printer* out); |
||||
bool PrintBetaServerFactory( |
||||
const grpc::string& package_qualified_service_name, |
||||
const grpc_generator::Service* service, grpc_generator::Printer* out); |
||||
bool PrintBetaStub(const grpc_generator::Service* service, |
||||
grpc_generator::Printer* out); |
||||
bool PrintBetaStubFactory(const grpc::string& package_qualified_service_name, |
||||
const grpc_generator::Service* service, |
||||
grpc_generator::Printer* out); |
||||
|
||||
// Get all comments (leading, leading_detached, trailing) and print them as a
|
||||
// docstring. Any leading space of a line will be removed, but the line
|
||||
// wrapping will not be changed.
|
||||
void PrintAllComments(std::vector<grpc::string> comments, |
||||
grpc_generator::Printer* out); |
||||
}; |
||||
|
||||
} // namespace
|
||||
|
||||
} // namespace grpc_python_generator
|
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_PYTHON_PRIVATE_GENERATOR_H
|
||||
@ -0,0 +1,126 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_SCHEMA_INTERFACE_H |
||||
#define GRPC_INTERNAL_COMPILER_SCHEMA_INTERFACE_H |
||||
|
||||
#include "src/compiler/config.h" |
||||
|
||||
#include <memory> |
||||
#include <vector> |
||||
|
||||
#ifndef GRPC_CUSTOM_STRING |
||||
#include <string> |
||||
#define GRPC_CUSTOM_STRING std::string |
||||
#endif |
||||
|
||||
namespace grpc { |
||||
|
||||
typedef GRPC_CUSTOM_STRING string; |
||||
|
||||
} // namespace grpc
|
||||
|
||||
namespace grpc_generator { |
||||
|
||||
// A common interface for objects having comments in the source.
|
||||
// Return formatted comments to be inserted in generated code.
|
||||
struct CommentHolder { |
||||
virtual ~CommentHolder() {} |
||||
virtual grpc::string GetLeadingComments(const grpc::string prefix) const = 0; |
||||
virtual grpc::string GetTrailingComments(const grpc::string prefix) const = 0; |
||||
virtual std::vector<grpc::string> GetAllComments() const = 0; |
||||
}; |
||||
|
||||
// An abstract interface representing a method.
|
||||
struct Method : public CommentHolder { |
||||
virtual ~Method() {} |
||||
|
||||
virtual grpc::string name() const = 0; |
||||
|
||||
virtual grpc::string input_type_name() const = 0; |
||||
virtual grpc::string output_type_name() const = 0; |
||||
|
||||
virtual bool get_module_and_message_path_input( |
||||
grpc::string *str, grpc::string generator_file_name, |
||||
bool generate_in_pb2_grpc, grpc::string import_prefix) const = 0; |
||||
virtual bool get_module_and_message_path_output( |
||||
grpc::string *str, grpc::string generator_file_name, |
||||
bool generate_in_pb2_grpc, grpc::string import_prefix) const = 0; |
||||
|
||||
virtual grpc::string get_input_type_name() const = 0; |
||||
virtual grpc::string get_output_type_name() const = 0; |
||||
virtual bool NoStreaming() const = 0; |
||||
virtual bool ClientStreaming() const = 0; |
||||
virtual bool ServerStreaming() const = 0; |
||||
virtual bool BidiStreaming() const = 0; |
||||
}; |
||||
|
||||
// An abstract interface representing a service.
|
||||
struct Service : public CommentHolder { |
||||
virtual ~Service() {} |
||||
|
||||
virtual grpc::string name() const = 0; |
||||
|
||||
virtual int method_count() const = 0; |
||||
virtual std::unique_ptr<const Method> method(int i) const = 0; |
||||
}; |
||||
|
||||
struct Printer { |
||||
virtual ~Printer() {} |
||||
|
||||
virtual void Print(const std::map<grpc::string, grpc::string> &vars, |
||||
const char *template_string) = 0; |
||||
virtual void Print(const char *string) = 0; |
||||
virtual void Indent() = 0; |
||||
virtual void Outdent() = 0; |
||||
}; |
||||
|
||||
// An interface that allows the source generated to be output using various
|
||||
// libraries/idls/serializers.
|
||||
struct File : public CommentHolder { |
||||
virtual ~File() {} |
||||
|
||||
virtual grpc::string filename() const = 0; |
||||
virtual grpc::string filename_without_ext() const = 0; |
||||
virtual grpc::string package() const = 0; |
||||
virtual std::vector<grpc::string> package_parts() const = 0; |
||||
virtual grpc::string additional_headers() const = 0; |
||||
|
||||
virtual int service_count() const = 0; |
||||
virtual std::unique_ptr<const Service> service(int i) const = 0; |
||||
|
||||
virtual std::unique_ptr<Printer> CreatePrinter(grpc::string *str) const = 0; |
||||
}; |
||||
} // namespace grpc_generator
|
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_SCHEMA_INTERFACE_H
|
||||
Loading…
Reference in new issue