使用模板处理protobuf数组转换

cheliequan

2023-01-13 43179f56c32c07d8f107b5f022ae608463817eac

 util.h

@@ -2,8 +2,28 @@
#define C_BHOMEBUS_API_UTIL_H

#define DATA_URL_PREFIX(x) "/data/api-v" x

#include "google/protobuf/descriptor.h"
#include "google/protobuf/util/json_util.h"
#include "google/protobuf/util/type_resolver.h"
#include "google/protobuf/util/type_resolver_util.h"

#include "google/protobuf/message.h"
#include "google/protobuf/descriptor.h"
#include "google/protobuf/descriptor.pb.h"
#include "google/protobuf/dynamic_message.h"
#include "google/protobuf/compiler/importer.h"

#include "3rdparty/yyjson/yyjson.h"
#include "3rdparty/bus_client/cbhomeclient.h"
#include "3rdparty/bus_client/message.h"
#include "proto/x86_64/sysset.pb.h"
#include <vector>

using namespace protomsg;
using namespace std;
using google::protobuf::util::JsonStringToMessage;


bool json_to_proto(const std::string &json, google::protobuf::Message& message);
bool proto_to_json(const google::protobuf::Message& message, std::string& json);
@@ -11,4 +31,82 @@
bool bus_dbapi_get(void *handle, const char* topic, google::protobuf::Message& message);
bool bus_dbapi_get_topic_data(void *handle, const char* topic, crepmsg **pptr_repmsg, size_t len);
extern "C" int bus_dbapi_get_json_data(const char * json, char ** pptr_data, uint8_t *ptr_type);
template <class T>  bool bus_dbapi_get_list (void *handle, const char* topic, T & list)

{

   char * ptr_value = NULL;
   size_t data_len = 0;
    //get the topic json data
   yyjson_type type = YYJSON_TYPE_OBJ;
   crepmsg *repmsg = NULL;
    size_t len = 0;
   //T t;
   //typedef decltype(list) vec_type;
	
   using TMPT = typename std::remove_reference<decltype(*list.begin())>::type;
   TMPT t;
   //vec_type::value_type t;
   int ret = 0;
	
   ret = bus_dbapi_get_topic_data(handle, topic, &repmsg, len);
    if ( ret < 0 || NULL == repmsg)
    {
        return false;
    }

   //parse the json to get value of the key "data"  
    int msglen = strlen(repmsg->data);
    char* msgdata = (char *)malloc(msglen);
    memset(msgdata, 0 , msglen);

    if(0 == bus_dbapi_get_json_data((const char *)repmsg->data, &msgdata, &type))
    {
        printf("======>> protomsg: %s\n", msgdata);            
    }

    //iterate the array  to get the protobuf message
   if (YYJSON_TYPE_ARR == type)
   {

        printf("======>> deal the ARRAY\n"); 
      // Read JSON and get root
      yyjson_doc *doc = yyjson_read(msgdata, strlen(msgdata), 0);
      yyjson_val *arr = yyjson_doc_get_root(doc);

        //yyjson_val *arr = yyjson_obj_get(root, "data"); 
      yyjson_val *val;
      yyjson_arr_iter iter;
      yyjson_arr_iter_init(arr, &iter);
      while ((val = yyjson_arr_iter_next(&iter)))
      {		
         data_len = 0;
         ptr_value = yyjson_val_write(val, YYJSON_WRITE_NOFLAG, &data_len); 
         printf("data: %s\n", ptr_value);	
         printf("len: %lu\n", data_len);  

         std::string jsonString = ptr_value;			
         if (json_to_proto(jsonString, t)) 
         {
            printf("======>> json_to_proto done\n");
            list.push_back(t);
         } 
         else
         {
            printf("======>> json_to_proto fail\n");
             yyjson_doc_free(doc);
             free(msgdata);				
            return false;
         } 		
      }

      // Free the doc
        yyjson_doc_free(doc);
   }

    free(msgdata);
   free_reply_msg(repmsg);

    return true;
} 
#endif