120 lines
3.1 KiB
C++
120 lines
3.1 KiB
C++
#include "boltdb.h"
|
|
|
|
BoltDB::BoltDB()
|
|
{
|
|
|
|
}
|
|
|
|
BoltDB* BoltDB::createFrom(QString filePath, QString &errorOut)
|
|
{
|
|
auto opts = ::Bolt_Options_New_Readonly();
|
|
|
|
QByteArray filePathBytes(filePath.toUtf8());
|
|
GoString filePathGS = Interop::toGoString_WeakRef(&filePathBytes);
|
|
|
|
auto open_ret = ::Bolt_Open(filePathGS, 0444, opts);
|
|
if (open_ret.r2 != 0) {
|
|
errorOut = QString::fromUtf8(open_ret.r1, open_ret.r2);
|
|
free(open_ret.r1);
|
|
return nullptr;
|
|
}
|
|
|
|
BoltDB *ret = new BoltDB();
|
|
ret->gmsDbRef = open_ret.r0;
|
|
return ret;
|
|
}
|
|
|
|
static const int ERROR_AND_STOP_CALLING = 100;
|
|
static const int ERROR_AND_KEEP_CALLING = 101;
|
|
static const int FINISHED_OK = 102;
|
|
static const int REAL_MESSAGE = 103;
|
|
|
|
bool BoltDB::listBucketsAtRoot(QString& errorOut, NameReciever cb)
|
|
{
|
|
auto listJob = ::Bolt_ListBucketsAtRoot(this->gmsDbRef);
|
|
|
|
errorOut.clear();
|
|
|
|
for(;;) {
|
|
auto gnr = ::GetNext(listJob);
|
|
|
|
if (gnr.r0 == ERROR_AND_STOP_CALLING) {
|
|
errorOut.append(QString::fromUtf8(gnr.r1, gnr.r2)); // log error
|
|
free(gnr.r1);
|
|
break; // done
|
|
|
|
} else if (gnr.r0 == ERROR_AND_KEEP_CALLING) {
|
|
errorOut.append(QString::fromUtf8(gnr.r1, gnr.r2)); // log error
|
|
free(gnr.r1);
|
|
continue;
|
|
|
|
} else if (gnr.r0 == FINISHED_OK) {
|
|
// Once we hit this, the go-side will clean up the channel / associated goroutines
|
|
break;
|
|
|
|
} else if (gnr.r0 == REAL_MESSAGE) {
|
|
cb(QByteArray(gnr.r1, gnr.r2));
|
|
free(gnr.r1);
|
|
continue;
|
|
}
|
|
}
|
|
|
|
return (errorOut.length() == 0);
|
|
}
|
|
|
|
bool BoltDB::getStatsJSON(std::function<void(QString)> onSuccess, std::function<void(QString)> onError)
|
|
{
|
|
auto statresp = Bolt_DBStats(this->gmsDbRef);
|
|
|
|
if (statresp.r0 == ERROR_AND_STOP_CALLING) {
|
|
QString err = QString::fromUtf8(statresp.r1, statresp.r2);
|
|
free(statresp.r1);
|
|
onError(err);
|
|
return false;
|
|
|
|
} else if (statresp.r0 == REAL_MESSAGE) {
|
|
QString json = QString::fromUtf8(statresp.r1, statresp.r2);
|
|
free(statresp.r1);
|
|
onSuccess(json);
|
|
return true;
|
|
|
|
} else {
|
|
// ?? shouldn't be reachable
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool BoltDB::getBucketStatsJSON(QStringList bucketPath, std::function<void(QString)> onSuccess, std::function<void(QString)> onError)
|
|
{
|
|
GoSliceManagedWrapper sliceWrapper(&bucketPath);
|
|
auto statresp = Bolt_BucketStats(this->gmsDbRef, sliceWrapper.slice);
|
|
|
|
if (statresp.r0 == ERROR_AND_STOP_CALLING) {
|
|
QString err = QString::fromUtf8(statresp.r1, statresp.r2);
|
|
free(statresp.r1);
|
|
onError(err);
|
|
return false;
|
|
|
|
} else if (statresp.r0 == REAL_MESSAGE) {
|
|
QString json = QString::fromUtf8(statresp.r1, statresp.r2);
|
|
free(statresp.r1);
|
|
onSuccess(json);
|
|
return true;
|
|
|
|
} else {
|
|
// ?? shouldn't be reachable
|
|
return false;
|
|
}
|
|
}
|
|
|
|
BoltDB::~BoltDB()
|
|
{
|
|
auto err = ::Bolt_Close(this->gmsDbRef);
|
|
if (err.r1 != 0) {
|
|
// Error closing database!
|
|
// Need to display an alert... somewhere
|
|
|
|
free(err.r0);
|
|
}
|
|
}
|