DB2Proto

最新推荐文章于 2024-10-12 07:00:00 发布

原创最新推荐文章于 2024-10-12 07:00:00 发布 · 251 阅读

0 ·

CC 4.0 BY-SA版权

C++ 专栏收录该内容

22 篇文章

订阅专栏

记录一下

定义person.proto文件，

package tutorial;
message Person {
required int32 id = 1;
optional string name = 2;
optional string email = 3;
}

#include "ProtoImporter.h"

using namespace google::protobuf;

void reflectionFill(Message* message, const FieldDescriptor* descriptor, const Reflection* reflection, std::map<string, string> row)
{
   assert(descriptor != NULL);
   //current not support for repeated label
   if (descriptor->label() == FieldDescriptor::LABEL_REPEATED) {
       return;
   }

const char* name = descriptor->name().c_str();

   switch (descriptor->type()) {
   case FieldDescriptor::TYPE_FIXED64:
   case FieldDescriptor::TYPE_INT64:
       reflection->SetInt64(message, descriptor, (long long)row[name].c_str()); break;
   case FieldDescriptor::TYPE_UINT64:
       reflection->SetInt64(message, descriptor, (unsigned long long)row[name].c_str()); break;

   case FieldDescriptor::TYPE_FIXED32:
   case FieldDescriptor::TYPE_INT32:
       reflection->SetInt32(message, descriptor, (int)row[name].c_str()); break;
   case FieldDescriptor::TYPE_UINT32:
       reflection->SetInt32(message, descriptor, (unsigned int)row[name].c_str()); break;

case FieldDescriptor::TYPE_STRING:
reflection->SetString(message, descriptor, (std::string)row[name]); break;

   case FieldDescriptor::TYPE_DOUBLE:
       reflection->SetDouble(message, descriptor, (double)atof(row[name].c_str())); break;
   case FieldDescriptor::TYPE_FLOAT:
       reflection->SetFloat(message, descriptor, (float)atof(row[name].c_str())); break;
   case FieldDescriptor::TYPE_BOOL:
       reflection->SetBool(message, descriptor, (bool)row[name].c_str()); break;
   default: std::cerr << "not support type " << descriptor->type() << std::endl;
   }
}

void convert(std::map<string, string> rowInfo, std::string& data)
{
google::protobuf::Message* message = sProtoImporter.createDynamicMessage("Person");

   //Message* message = createMessage(message_type);
   const Reflection* reflection = message->GetReflection();
   const Descriptor* descriptor = message->GetDescriptor();

   for (int i = 0; i < descriptor->field_count(); ++i)
   {
       reflectionFill(message, descriptor->field(i), reflection, rowInfo);
   }
   message->SerializeToString(&data);

delete message;
}

void TestProto()
{
   // sql::row rowInfo;
   std::map<string, string> rowInfo;
   rowInfo["id"] = "20001";
   rowInfo["name"] = "testname";
   rowInfo["email"] = "aaa@163.com";

   std::string data;
   if (sProtoImporter.Import("person.proto"))
   {
       convert(rowInfo, data);
   }

printf("get: %s", data.c_str());

return;
}