Clean up AssertClient

This commit is contained in:
Ryan Huffman 2015-08-27 09:31:17 -07:00
parent 3731091d5e
commit b72e7e356d

View file

@ -16,6 +16,7 @@
#include "AssetRequest.h"
#include "NodeList.h"
#include "PacketReceiver.h"
#include "AssetUtils.h"
MessageID AssetClient::_currentID = 0;
@ -37,7 +38,7 @@ AssetRequest* AssetClient::create(QString hash) {
return req;
}
if (hash.length() != 32) {
if (hash.length() != HASH_HEX_LENGTH) {
qDebug() << "Invalid hash size";
return nullptr;
}
@ -69,7 +70,7 @@ bool AssetClient::getAsset(QString hash, DataOffset start, DataOffset end, Recei
auto messageID = ++_currentID;
packet->writePrimitive(messageID);
packet->write(hash.toLatin1().constData(), 32);
packet->write(hash.toLatin1().constData(), HASH_HEX_LENGTH);
packet->writePrimitive(start);
packet->writePrimitive(end);
@ -92,7 +93,7 @@ bool AssetClient::getAssetInfo(QString hash, GetInfoCallback callback) {
auto messageID = ++_currentID;
packet->writePrimitive(messageID);
packet->write(hash.toLatin1().constData(), 32);
packet->write(hash.toLatin1().constData(), HASH_HEX_LENGTH);
nodeList->sendPacket(std::move(packet), *assetServer);
@ -112,15 +113,15 @@ void AssetClient::handleAssetGetInfoReply(QSharedPointer<NLPacket> packet, Share
AssetServerError error;
packet->readPrimitive(&error);
AssetInfo info;
AssetInfo info { assetHash, 0 };
if (!error) {
if (error == NO_ERROR) {
packet->readPrimitive(&info.size);
}
if (_pendingInfoRequests.contains(messageID)) {
auto callback = _pendingInfoRequests.take(messageID);
callback(error != NO_ERROR, info);
callback(error == NO_ERROR, info);
}
}
@ -149,7 +150,7 @@ void AssetClient::handleAssetGetReply(QSharedPointer<NLPacketList> packetList, S
if (_pendingRequests.contains(messageID)) {
auto callback = _pendingRequests.take(messageID);
callback(!error, data);
callback(error == NO_ERROR, data);
}
}
@ -159,7 +160,7 @@ bool AssetClient::uploadAsset(QByteArray data, QString extension, UploadResultCa
if (assetServer) {
auto packetList = std::unique_ptr<NLPacketList>(new NLPacketList(PacketType::AssetUpload, QByteArray(), true, true));
auto messageID = _currentID++;
auto messageID = ++_currentID;
packetList->writePrimitive(messageID);
packetList->writePrimitive(static_cast<uint8_t>(extension.length()));
@ -185,23 +186,23 @@ void AssetClient::handleAssetUploadReply(QSharedPointer<NLPacket> packet, Shared
MessageID messageID;
packet->readPrimitive(&messageID);
bool success;
packet->readPrimitive(&success);
AssetServerError error;
packet->readPrimitive(&error);
QString hashString { "" };
if (success) {
if (error) {
qDebug() << "Error uploading file to asset server";
} else {
auto hashData = packet->read(HASH_HEX_LENGTH);
hashString = QString(hashData);
qDebug() << "Successfully uploaded asset to asset-server - SHA256 hash is " << hashString;
} else {
qDebug() << "Error uploading file to asset server";
}
if (_pendingUploads.contains(messageID)) {
auto callback = _pendingUploads.take(messageID);
callback(success, hashString);
callback(error == NO_ERROR, hashString);
}
}