From 43179f56c32c07d8f107b5f022ae608463817eac Mon Sep 17 00:00:00 2001
From: cheliequan <liequanche@126.com>
Date: 星期五, 13 一月 2023 09:49:41 +0800
Subject: [PATCH] 使用模板处理protobuf数组转换
---
rule.cpp | 9 ++++
util.h | 100 +++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 107 insertions(+), 2 deletions(-)
diff --git a/rule.cpp b/rule.cpp
index 04cf8e5..50f8f6e 100644
--- a/rule.cpp
+++ b/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;
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