From 43179f56c32c07d8f107b5f022ae608463817eac Mon Sep 17 00:00:00 2001 From: cheliequan <liequanche@126.com> Date: 星期五, 13 一月 2023 09:49:41 +0800 Subject: [PATCH] 使用模板处理protobuf数组转换 --- util.h | 100 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 99 insertions(+), 1 deletions(-) diff --git a/util.h b/util.h index 77298c1..fbc6a6e 100644 --- a/util.h +++ b/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 \ No newline at end of file +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 -- Gitblit v1.8.0