158 lines
3.6 KiB
C++
158 lines
3.6 KiB
C++
// 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
|
|
}
|
|
|