79 lines
2.4 KiB
C++
79 lines
2.4 KiB
C++
#ifndef NET_GRPC_COMPILER_JAVA_GENERATOR_H_
|
|
#define NET_GRPC_COMPILER_JAVA_GENERATOR_H_
|
|
|
|
#include <stdlib.h> // for abort()
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include <google/protobuf/io/zero_copy_stream.h>
|
|
#include <google/protobuf/descriptor.h>
|
|
|
|
class LogHelper {
|
|
std::ostream* os;
|
|
|
|
public:
|
|
LogHelper(std::ostream* os) : os(os) {}
|
|
~LogHelper() {
|
|
*os << std::endl;
|
|
::abort();
|
|
}
|
|
std::ostream& get_os() {
|
|
return *os;
|
|
}
|
|
};
|
|
|
|
// Abort the program after logging the mesage if the given condition is not
|
|
// true. Otherwise, do nothing.
|
|
#define GRPC_CODEGEN_CHECK(x) !(x) && LogHelper(&std::cerr).get_os() \
|
|
<< "CHECK FAILED: " << __FILE__ << ":" \
|
|
<< __LINE__ << ": "
|
|
|
|
// Abort the program after logging the mesage.
|
|
#define GRPC_CODEGEN_FAIL GRPC_CODEGEN_CHECK(false)
|
|
|
|
using namespace std;
|
|
|
|
namespace java_grpc_generator {
|
|
|
|
enum ProtoFlavor {
|
|
NORMAL, LITE, NANO
|
|
};
|
|
|
|
// Returns the package name of the gRPC services defined in the given file.
|
|
string ServiceJavaPackage(const google::protobuf::FileDescriptor* file, bool nano);
|
|
|
|
// Returns the name of the outer class that wraps in all the generated code for
|
|
// the given service.
|
|
string ServiceClassName(const google::protobuf::ServiceDescriptor* service);
|
|
|
|
// Writes the generated service interface into the given ZeroCopyOutputStream
|
|
void GenerateService(const google::protobuf::ServiceDescriptor* service,
|
|
google::protobuf::io::ZeroCopyOutputStream* out,
|
|
ProtoFlavor flavor,
|
|
bool disable_version);
|
|
|
|
} // namespace java_grpc_generator
|
|
|
|
namespace java_dubbo_generator {
|
|
|
|
enum ProtoFlavor {
|
|
NORMAL
|
|
};
|
|
|
|
// Returns the package name of the Dubbo services defined in the given file.
|
|
string ServiceJavaPackage(const google::protobuf::FileDescriptor* file, bool nano);
|
|
|
|
// Returns the name of the outer class that wraps in all the generated code for
|
|
// the given service.
|
|
string ServiceClassName(const google::protobuf::ServiceDescriptor* service);
|
|
|
|
// Writes the generated service interface into the given ZeroCopyOutputStream
|
|
void GenerateService(const google::protobuf::ServiceDescriptor* service,
|
|
google::protobuf::io::ZeroCopyOutputStream* out,
|
|
ProtoFlavor flavor,
|
|
bool disable_version);
|
|
|
|
} // namespace java_dubbo_generator
|
|
|
|
#endif // NET_GRPC_COMPILER_JAVA_GENERATOR_H_
|