使用模板处理protobuf数组转换

cheliequan

2023-01-13 43179f56c32c07d8f107b5f022ae608463817eac

使用模板处理protobuf数组转换

 rule.cpp

@@ -4,7 +4,14 @@

using namespace protomsg;

bool dbapi_get_time_rules(void *handle, std::vector<protomsg::CameraTimerule> & list) {
bool dbapi_get_time_rules(void *handle, std::vector<protomsg::CameraTimerule> & list){
    const char *topic = DATA_URL_PREFIX("/cameraTimerule/findAll");	

    bool ret = bus_dbapi_get_list(handle, topic, list);
    return true;
}

bool dbapi_get_time_rules1(void *handle, std::vector<protomsg::CameraTimerule> & list) {
    const char *topic = DATA_URL_PREFIX("/cameraTimerule/findAll");

    char * ptr_value = NULL;

 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);
#endif
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