12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- #include "clientsourcegenerator.h"
- #include <google/protobuf/io/zero_copy_stream.h>
- #include "utils.h"
- #include "templates.h"
- using namespace qtprotobuf::generator;
- using namespace ::google::protobuf;
- using namespace ::google::protobuf::compiler;
- ClientSourceGenerator::ClientSourceGenerator(const google::protobuf::ServiceDescriptor *service,
- std::unique_ptr<google::protobuf::io::ZeroCopyOutputStream> out) :
- ClassSourceGeneratorBase(service->full_name(), std::move(out))
- , mService(service)
- {
- mClassName += "Client";
- }
- void ClientSourceGenerator::printMethods()
- {
- for (int i = 0; i < mService->method_count(); i++) {
- const MethodDescriptor* method = mService->method(i);
- std::string inputTypeName = method->input_type()->full_name();
- std::string outputTypeName = method->output_type()->full_name();
- utils::replace(inputTypeName, ".", "::");
- utils::replace(outputTypeName, ".", "::");
- std::map<std::string, std::string> parameters = {{"classname", mClassName},
- {"return_type", outputTypeName},
- {"method_name", method->name()},
- {"param_type", inputTypeName},
- {"param_name", "arg"},
- {"return_name", "ret"}
- };
- mPrinter.Print(parameters, Templates::ClientMethodDefinitionSyncTemplate);
- mPrinter.Print(parameters, Templates::ClientMethodDefinitionAsyncTemplate);
- }
- }
- void ClientSourceGenerator::printConstructor()
- {
- mPrinter.Print({ {"classname", mClassName},
- {"parent_class", "AbstractClient"},
- {"service_name", mService->full_name()} }, Templates::ConstructorDefinitionSyncTemplate);
- }
|