78 lines
2.5 KiB
C++
78 lines
2.5 KiB
C++
#include "networkaccessmanager.h"
|
|
|
|
#include <QNetworkRequest>
|
|
#include <QNetworkAccessManager>
|
|
#include <QNetworkReply>
|
|
#include <QNetworkDiskCache>
|
|
#include <QCoreApplication>
|
|
#include <QDir>
|
|
|
|
#include <QtDebug>
|
|
|
|
NetworkAccessManager::NetworkAccessManager(QObject* parent,
|
|
QNetworkAccessManager* injected)
|
|
: QObject(parent),
|
|
network_(injected ? injected : new QNetworkAccessManager(this)),
|
|
cache_(new QNetworkDiskCache(this))
|
|
{
|
|
cache_->setCacheDirectory(QString("%1/.config/%2/networkcache/")
|
|
.arg(QDir::homePath(), QCoreApplication::organizationName()));
|
|
network_->setCache(cache_);
|
|
}
|
|
|
|
void NetworkAccessManager::Get(const QUrl &url, QObject *receiver,
|
|
const char *slot, quint64 id, bool force_cache) {
|
|
connect(receiver, SIGNAL(destroyed(QObject*)), SLOT(ReceiverDestroyed(QObject*)));
|
|
QMetaObject::invokeMethod(
|
|
this, "RunGet", Qt::QueuedConnection,
|
|
Q_ARG(QUrl, url), Q_ARG(QObject*, receiver),
|
|
Q_ARG(const char*, slot),
|
|
Q_ARG(quint64, id), Q_ARG(bool, force_cache));
|
|
}
|
|
|
|
void NetworkAccessManager::RunGet(const QUrl &url, QObject *receiver,
|
|
const char *slot, quint64 id, bool force_cache) {
|
|
QNetworkRequest req = CreateRequest(url, force_cache);
|
|
QNetworkReply* reply = network_->get(req);
|
|
connect(reply, SIGNAL(finished()), SLOT(RequestFinished()));
|
|
|
|
Receiver r;
|
|
r.receiver = receiver;
|
|
r.slot = slot;
|
|
r.id = id;
|
|
|
|
pending_replies_.insert(reply, r);
|
|
}
|
|
|
|
void NetworkAccessManager::ReceiverDestroyed(QObject* receiver) {
|
|
foreach (QNetworkReply* key, pending_replies_.keys()) {
|
|
if (pending_replies_[key].receiver == receiver)
|
|
pending_replies_[key].receiver = NULL;
|
|
}
|
|
}
|
|
|
|
QNetworkRequest NetworkAccessManager::CreateRequest(const QUrl& url, bool force_cache) {
|
|
QNetworkRequest req(url);
|
|
req.setRawHeader("User-Agent", QString("%1 %2").arg(
|
|
QCoreApplication::applicationName(),
|
|
QCoreApplication::applicationVersion()).toUtf8());
|
|
|
|
if (force_cache) {
|
|
req.setAttribute(QNetworkRequest::CacheLoadControlAttribute,
|
|
QNetworkRequest::PreferCache);
|
|
}
|
|
return req;
|
|
}
|
|
|
|
void NetworkAccessManager::RequestFinished() {
|
|
QNetworkReply* reply = static_cast<QNetworkReply*>(sender());
|
|
Receiver r = pending_replies_.take(reply);
|
|
|
|
if (!r.receiver)
|
|
return;
|
|
|
|
QMetaObject::invokeMethod(r.receiver, r.slot,
|
|
Q_ARG(quint64, r.id),
|
|
Q_ARG(QNetworkReply*, reply));
|
|
}
|