mirror of
https://github.com/QuasarApp/Heart.git
synced 2025-04-28 10:44:39 +00:00
158 lines
3.6 KiB
C++
158 lines
3.6 KiB
C++
/*
|
|
* Copyright (C) 2018-2023 QuasarApp.
|
|
* Distributed under the lgplv3 software license, see the accompanying
|
|
* Everyone is permitted to copy and distribute verbatim copies
|
|
* of this license document, but changing it is not allowed.
|
|
*/
|
|
|
|
#include "bigdatatest.h"
|
|
|
|
#include <abstractnode.h>
|
|
|
|
#define LOCAL_TEST_PORT TEST_PORT + 4
|
|
|
|
class BigPackage: public QH::PKG::AbstractData {
|
|
QH_PACKAGE_AUTO(BigPackage)
|
|
|
|
public:
|
|
BigPackage(){
|
|
data = {};
|
|
};
|
|
QByteArray data;
|
|
|
|
// StreamBase interface
|
|
protected:
|
|
QDataStream &fromStream(QDataStream &stream) override{
|
|
stream >> data;
|
|
|
|
return stream;
|
|
};
|
|
|
|
QDataStream &toStream(QDataStream &stream) const override{
|
|
stream << data;
|
|
|
|
return stream;
|
|
};
|
|
};
|
|
|
|
|
|
class BigDataTestParser: public QH::iParser {
|
|
public:
|
|
BigDataTestParser(QH::AbstractNode* parentNode): QH::iParser(parentNode) {
|
|
registerPackageType<BigPackage>();
|
|
data = new BigPackage();
|
|
|
|
}
|
|
|
|
// iParser interface
|
|
public:
|
|
bool sendRequest(const QByteArray& data) {
|
|
BigPackage pkg;
|
|
pkg.data = data;
|
|
|
|
return sendData(&pkg, QH::HostAddress(TEST_LOCAL_HOST, LOCAL_TEST_PORT));
|
|
}
|
|
|
|
const BigPackage* getData() const {
|
|
QMutexLocker locker(&_mData);
|
|
return data;
|
|
}
|
|
|
|
QH::ParserResult parsePackage(const QSharedPointer<QH::PKG::AbstractData> &pkg,
|
|
const QH::Header &,
|
|
QH::AbstractNodeInfo *sender) override {
|
|
|
|
if (pkg->cmd() == BigPackage::command()) {
|
|
|
|
_mData.lock();
|
|
data->copy<BigPackage>(*pkg);
|
|
_mData.unlock();
|
|
sendData(data, sender);
|
|
return QH::ParserResult::Processed;
|
|
}
|
|
|
|
return QH::ParserResult::NotProcessed;
|
|
};
|
|
int version() const override {return 0;};
|
|
QString parserId() const override {return "BigDataTestParser";};
|
|
|
|
private:
|
|
mutable QMutex _mData;
|
|
BigPackage *data = nullptr;
|
|
};
|
|
|
|
class TestingClientBigData: public QH::AbstractNode {
|
|
|
|
// AbstractNode interface
|
|
public:
|
|
TestingClientBigData() {
|
|
_parser = addApiParser<BigDataTestParser>();
|
|
}
|
|
|
|
NodeType nodeType() const override {
|
|
return NodeType::Node;
|
|
};
|
|
|
|
const QSharedPointer<QH::iParser>& parser() const {
|
|
return _parser;
|
|
}
|
|
|
|
private:
|
|
QSharedPointer<QH::iParser> _parser;
|
|
};
|
|
|
|
BigDataTest::BigDataTest() {
|
|
_nodeA = new TestingClientBigData();
|
|
_nodeAParser = static_cast<TestingClientBigData*>(_nodeA)->parser();
|
|
|
|
_nodeB = new TestingClientBigData();
|
|
_nodeBParser = static_cast<TestingClientBigData*>(_nodeB)->parser();
|
|
|
|
}
|
|
|
|
BigDataTest::~BigDataTest() {
|
|
_nodeA->softDelete();
|
|
_nodeB->softDelete();
|
|
}
|
|
|
|
void BigDataTest::test() {
|
|
//#ifdef Q_OS_LINUX
|
|
|
|
QVERIFY(connectTest());
|
|
QVERIFY(sendDataTest());
|
|
//#endif
|
|
}
|
|
|
|
bool BigDataTest::connectTest() {
|
|
|
|
if (!_nodeA->run(TEST_LOCAL_HOST, LOCAL_TEST_PORT)) {
|
|
return false;
|
|
}
|
|
|
|
return connectFunc(_nodeB, TEST_LOCAL_HOST, LOCAL_TEST_PORT);
|
|
}
|
|
|
|
bool BigDataTest::sendDataTest() {
|
|
|
|
|
|
QByteArray testData;
|
|
for (int i = 0; i < 1024 * 1024 * 10; i++) {
|
|
testData.push_back(rand());
|
|
}
|
|
|
|
auto request = [this, testData](){
|
|
return _nodeBParser.staticCast<BigDataTestParser>()->sendRequest(testData);
|
|
};
|
|
|
|
auto client = _nodeBParser.staticCast<BigDataTestParser>();
|
|
auto server = _nodeAParser.staticCast<BigDataTestParser>();
|
|
|
|
auto check = [client, server, testData](){
|
|
return client->getData() && server->getData()->data == testData;
|
|
};
|
|
|
|
return funcPrivateConnect(request, check);
|
|
}
|
|
|
|
|