code

Plugins/MqttClient/Source/DTMqtt/MqttCPP/iclient_persistence.cpp

@@ -0,0 +1,157 @@
+// Copyright 2023 Dexter.Wan. All Rights Reserved. 
+// EMail: 45141961@qq.com
+
+#include "iclient_persistence.h"
+#include "types.h"
+#include <cstring>
+#include <cstdlib>
+#include <vector>
+
+#ifdef _MSC_VER
+#pragma warning(disable:4996)
+#endif
+
+using namespace std;
+
+namespace mqtt {
+
+/////////////////////////////////////////////////////////////////////////////
+// Functions to transition C persistence calls to the C++ persistence object.
+
+// Upon the call to persistence_open(), the 'context' has the address of the
+// C++ persistence object, which is reassigned to the 'handle'. Subsequent
+// calls have the object address as the handle.
+
+int iclient_persistence::persistence_open(void** handle, const char* clientID, 
+										  const char* serverURI, void* context)
+{
+	try {
+		if (handle && clientID && serverURI && context) {
+			static_cast<iclient_persistence*>(context)->open(clientID, serverURI);
+			*handle = context;
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_close(void* handle)
+{
+	try {
+		if (handle) {
+			static_cast<iclient_persistence*>(handle)->close();
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_put(void* handle, char* key, int bufcount, 
+										 char* buffers[], int buflens[])
+{
+	try {
+		if (handle && bufcount > 0 && buffers && buflens) {
+			std::vector<string_view> vec;
+			for (int i=0; i<bufcount; ++i)
+				vec.push_back(string_view(buffers[i], buflens[i]));
+			static_cast<iclient_persistence*>(handle)->put(key, vec);
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_get(void* handle, char* key, 
+										 char** buffer, int* buflen)
+{
+	try {
+		if (handle && key && buffer && buflen) {
+			auto s = static_cast<iclient_persistence*>(handle)->get(key);
+			size_t n = s.length();
+			*buffer = static_cast<char*>(MQTTAsync_malloc(n));
+			memcpy(*buffer, s.data(), n);
+			*buflen = int(n);
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_remove(void* handle, char* key)
+{
+	try {
+		if (handle && key) {
+			static_cast<iclient_persistence*>(handle)->remove(key);
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_keys(void* handle, char*** keys, int* nkeys)
+{
+	try {
+		if (handle && keys && nkeys) {
+			auto k = static_cast<iclient_persistence*>(handle)->keys();
+			size_t n = k.size();
+			*nkeys = int(n);
+			if (n == 0) {
+				*keys = nullptr;
+			}
+			else {
+				*keys = static_cast<char**>(MQTTAsync_malloc(n*sizeof(char*)));
+				for (size_t i=0; i<n; ++i) {
+					auto sz = k[i].size();
+					char* buf = static_cast<char*>(MQTTAsync_malloc(sz+1));
+					strncpy(buf, k[i].c_str(), sz+1);
+					buf[sz] = '\0';
+					(*keys)[i] = buf;
+				}
+			}
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_clear(void* handle)
+{
+	try {
+		if (handle) {
+			static_cast<iclient_persistence*>(handle)->clear();
+			return MQTTASYNC_SUCCESS;
+		}
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+int iclient_persistence::persistence_containskey(void* handle, char* key)
+{
+	try {
+		if (handle && key &&
+				static_cast<iclient_persistence*>(handle)->contains_key(key))
+			return MQTTASYNC_SUCCESS;
+	}
+	catch (...) {}
+
+	return MQTTCLIENT_PERSISTENCE_ERROR;
+}
+
+/////////////////////////////////////////////////////////////////////////////
+// end namespace mqtt
+}
+