mirror of https://github.com/grpc/grpc.git
Harsh Vardhan
8 years ago
parent
739cecb0bc
commit
5b8fa87282
8 changed files with 353 additions and 263 deletions
@ -0,0 +1,146 @@ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H |
||||
#define GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H |
||||
|
||||
#include "src/compiler/config.h" |
||||
#include "src/compiler/schema_interface.h" |
||||
#include "src/compiler/cpp_generator_helpers.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); |
||||
} |
||||
|
||||
bool NoStreaming() const { |
||||
return !method_->client_streaming() && !method_->server_streaming(); |
||||
} |
||||
|
||||
bool ClientStreaming() const { |
||||
return method_->client_streaming() && !method_->server_streaming(); |
||||
} |
||||
|
||||
bool ServerStreaming() const { |
||||
return !method_->client_streaming() && 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); |
||||
} |
||||
|
||||
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); |
||||
} |
||||
|
||||
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); |
||||
} |
||||
|
||||
private: |
||||
const grpc::protobuf::FileDescriptor *file_; |
||||
}; |
||||
|
||||
|
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_PROTOBUF_PLUGIN_H
|
||||
@ -0,0 +1,114 @@ |
||||
/*
|
||||
* |
||||
* 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 <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; |
||||
}; |
||||
|
||||
// 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 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