mirror of https://github.com/aria2/aria2
DefaultBtMessageDispatcher: Remove unused members peerStorage_, pieceStorage_
parent
cf8cfeefbc
commit
7a57ecc5f8
|
@ -64,8 +64,6 @@ namespace aria2 {
|
||||||
DefaultBtMessageDispatcher::DefaultBtMessageDispatcher()
|
DefaultBtMessageDispatcher::DefaultBtMessageDispatcher()
|
||||||
: cuid_{0},
|
: cuid_{0},
|
||||||
downloadContext_{nullptr},
|
downloadContext_{nullptr},
|
||||||
peerStorage_{nullptr},
|
|
||||||
pieceStorage_{nullptr},
|
|
||||||
peerConnection_{nullptr},
|
peerConnection_{nullptr},
|
||||||
messageFactory_{nullptr},
|
messageFactory_{nullptr},
|
||||||
requestGroupMan_{nullptr},
|
requestGroupMan_{nullptr},
|
||||||
|
@ -315,16 +313,6 @@ void DefaultBtMessageDispatcher::setDownloadContext
|
||||||
downloadContext_ = downloadContext;
|
downloadContext_ = downloadContext;
|
||||||
}
|
}
|
||||||
|
|
||||||
void DefaultBtMessageDispatcher::setPieceStorage(PieceStorage* pieceStorage)
|
|
||||||
{
|
|
||||||
pieceStorage_ = pieceStorage;
|
|
||||||
}
|
|
||||||
|
|
||||||
void DefaultBtMessageDispatcher::setPeerStorage(PeerStorage* peerStorage)
|
|
||||||
{
|
|
||||||
peerStorage_ = peerStorage;
|
|
||||||
}
|
|
||||||
|
|
||||||
void DefaultBtMessageDispatcher::setBtMessageFactory(BtMessageFactory* factory)
|
void DefaultBtMessageDispatcher::setBtMessageFactory(BtMessageFactory* factory)
|
||||||
{
|
{
|
||||||
messageFactory_ = factory;
|
messageFactory_ = factory;
|
||||||
|
|
|
@ -45,8 +45,6 @@
|
||||||
namespace aria2 {
|
namespace aria2 {
|
||||||
|
|
||||||
class DownloadContext;
|
class DownloadContext;
|
||||||
class PeerStorage;
|
|
||||||
class PieceStorage;
|
|
||||||
class BtMessage;
|
class BtMessage;
|
||||||
class BtMessageFactory;
|
class BtMessageFactory;
|
||||||
class Peer;
|
class Peer;
|
||||||
|
@ -60,8 +58,6 @@ private:
|
||||||
std::deque<std::unique_ptr<BtMessage> > messageQueue_;
|
std::deque<std::unique_ptr<BtMessage> > messageQueue_;
|
||||||
std::deque<std::unique_ptr<RequestSlot>> requestSlots_;
|
std::deque<std::unique_ptr<RequestSlot>> requestSlots_;
|
||||||
DownloadContext* downloadContext_;
|
DownloadContext* downloadContext_;
|
||||||
PeerStorage* peerStorage_;
|
|
||||||
PieceStorage* pieceStorage_;
|
|
||||||
PeerConnection* peerConnection_;
|
PeerConnection* peerConnection_;
|
||||||
BtMessageFactory* messageFactory_;
|
BtMessageFactory* messageFactory_;
|
||||||
std::shared_ptr<Peer> peer_;
|
std::shared_ptr<Peer> peer_;
|
||||||
|
@ -135,10 +131,6 @@ public:
|
||||||
|
|
||||||
void setDownloadContext(DownloadContext* downloadContext);
|
void setDownloadContext(DownloadContext* downloadContext);
|
||||||
|
|
||||||
void setPieceStorage(PieceStorage* pieceStorage);
|
|
||||||
|
|
||||||
void setPeerStorage(PeerStorage* peerStorage);
|
|
||||||
|
|
||||||
void setBtMessageFactory(BtMessageFactory* factory);
|
void setBtMessageFactory(BtMessageFactory* factory);
|
||||||
|
|
||||||
void setRequestGroupMan(RequestGroupMan* rgman);
|
void setRequestGroupMan(RequestGroupMan* rgman);
|
||||||
|
|
|
@ -180,8 +180,6 @@ PeerInteractionCommand::PeerInteractionCommand
|
||||||
dispatcher->setCuid(cuid);
|
dispatcher->setCuid(cuid);
|
||||||
dispatcher->setPeer(getPeer());
|
dispatcher->setPeer(getPeer());
|
||||||
dispatcher->setDownloadContext(requestGroup_->getDownloadContext().get());
|
dispatcher->setDownloadContext(requestGroup_->getDownloadContext().get());
|
||||||
dispatcher->setPieceStorage(pieceStorage.get());
|
|
||||||
dispatcher->setPeerStorage(peerStorage.get());
|
|
||||||
dispatcher->setRequestTimeout(getOption()->
|
dispatcher->setRequestTimeout(getOption()->
|
||||||
getAsInt(PREF_BT_REQUEST_TIMEOUT));
|
getAsInt(PREF_BT_REQUEST_TIMEOUT));
|
||||||
dispatcher->setBtMessageFactory(factory.get());
|
dispatcher->setBtMessageFactory(factory.get());
|
||||||
|
|
|
@ -6,8 +6,6 @@
|
||||||
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
#include "Exception.h"
|
#include "Exception.h"
|
||||||
#include "MockPieceStorage.h"
|
|
||||||
#include "MockPeerStorage.h"
|
|
||||||
#include "MockBtMessage.h"
|
#include "MockBtMessage.h"
|
||||||
#include "MockBtMessageFactory.h"
|
#include "MockBtMessageFactory.h"
|
||||||
#include "prefs.h"
|
#include "prefs.h"
|
||||||
|
@ -44,8 +42,6 @@ private:
|
||||||
std::shared_ptr<DownloadContext> dctx_;
|
std::shared_ptr<DownloadContext> dctx_;
|
||||||
std::shared_ptr<Peer> peer;
|
std::shared_ptr<Peer> peer;
|
||||||
std::unique_ptr<DefaultBtMessageDispatcher> btMessageDispatcher;
|
std::unique_ptr<DefaultBtMessageDispatcher> btMessageDispatcher;
|
||||||
std::unique_ptr<MockPeerStorage> peerStorage;
|
|
||||||
std::unique_ptr<MockPieceStorage> pieceStorage;
|
|
||||||
std::unique_ptr<MockBtMessageFactory> messageFactory_;
|
std::unique_ptr<MockBtMessageFactory> messageFactory_;
|
||||||
std::unique_ptr<RequestGroupMan> rgman_;
|
std::unique_ptr<RequestGroupMan> rgman_;
|
||||||
std::shared_ptr<Option> option_;
|
std::shared_ptr<Option> option_;
|
||||||
|
@ -106,21 +102,6 @@ public:
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
class MockPieceStorage2 : public MockPieceStorage {
|
|
||||||
private:
|
|
||||||
std::shared_ptr<Piece> piece;
|
|
||||||
public:
|
|
||||||
virtual std::shared_ptr<Piece> getPiece(size_t index) CXX11_OVERRIDE
|
|
||||||
{
|
|
||||||
return piece;
|
|
||||||
}
|
|
||||||
|
|
||||||
void setPiece(const std::shared_ptr<Piece>& piece)
|
|
||||||
{
|
|
||||||
this->piece = piece;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
class MockBtMessageFactory2 : public MockBtMessageFactory {
|
class MockBtMessageFactory2 : public MockBtMessageFactory {
|
||||||
public:
|
public:
|
||||||
virtual std::unique_ptr<BtCancelMessage>
|
virtual std::unique_ptr<BtCancelMessage>
|
||||||
|
@ -146,9 +127,6 @@ public:
|
||||||
peer = std::make_shared<Peer>("192.168.0.1", 6969);
|
peer = std::make_shared<Peer>("192.168.0.1", 6969);
|
||||||
peer->allocateSessionResource
|
peer->allocateSessionResource
|
||||||
(dctx_->getPieceLength(), dctx_->getTotalLength());
|
(dctx_->getPieceLength(), dctx_->getTotalLength());
|
||||||
peerStorage = make_unique<MockPeerStorage>();
|
|
||||||
pieceStorage = make_unique<MockPieceStorage>();
|
|
||||||
|
|
||||||
messageFactory_ = make_unique<MockBtMessageFactory2>();
|
messageFactory_ = make_unique<MockBtMessageFactory2>();
|
||||||
|
|
||||||
rgman_ = make_unique<RequestGroupMan>
|
rgman_ = make_unique<RequestGroupMan>
|
||||||
|
@ -157,8 +135,6 @@ public:
|
||||||
btMessageDispatcher = make_unique<DefaultBtMessageDispatcher>();
|
btMessageDispatcher = make_unique<DefaultBtMessageDispatcher>();
|
||||||
btMessageDispatcher->setPeer(peer);
|
btMessageDispatcher->setPeer(peer);
|
||||||
btMessageDispatcher->setDownloadContext(dctx_.get());
|
btMessageDispatcher->setDownloadContext(dctx_.get());
|
||||||
btMessageDispatcher->setPieceStorage(pieceStorage.get());
|
|
||||||
btMessageDispatcher->setPeerStorage(peerStorage.get());
|
|
||||||
btMessageDispatcher->setBtMessageFactory(messageFactory_.get());
|
btMessageDispatcher->setBtMessageFactory(messageFactory_.get());
|
||||||
btMessageDispatcher->setCuid(1);
|
btMessageDispatcher->setCuid(1);
|
||||||
btMessageDispatcher->setRequestGroupMan(rgman_.get());
|
btMessageDispatcher->setRequestGroupMan(rgman_.get());
|
||||||
|
@ -233,11 +209,7 @@ void DefaultBtMessageDispatcherTest::testCheckRequestSlotAndDoNecessaryThing()
|
||||||
CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index));
|
CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index));
|
||||||
CPPUNIT_ASSERT_EQUAL((size_t)0, index);
|
CPPUNIT_ASSERT_EQUAL((size_t)0, index);
|
||||||
|
|
||||||
auto pieceStorage = make_unique<MockPieceStorage2>();
|
|
||||||
pieceStorage->setPiece(piece);
|
|
||||||
|
|
||||||
btMessageDispatcher->setRequestTimeout(60);
|
btMessageDispatcher->setRequestTimeout(60);
|
||||||
btMessageDispatcher->setPieceStorage(pieceStorage.get());
|
|
||||||
btMessageDispatcher->addOutstandingRequest
|
btMessageDispatcher->addOutstandingRequest
|
||||||
(make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece));
|
(make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece));
|
||||||
|
|
||||||
|
@ -256,11 +228,7 @@ testCheckRequestSlotAndDoNecessaryThing_timeout() {
|
||||||
CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index));
|
CPPUNIT_ASSERT(piece->getMissingUnusedBlockIndex(index));
|
||||||
CPPUNIT_ASSERT_EQUAL((size_t)0, index);
|
CPPUNIT_ASSERT_EQUAL((size_t)0, index);
|
||||||
|
|
||||||
auto pieceStorage = make_unique<MockPieceStorage2>();
|
|
||||||
pieceStorage->setPiece(piece);
|
|
||||||
|
|
||||||
btMessageDispatcher->setRequestTimeout(60);
|
btMessageDispatcher->setRequestTimeout(60);
|
||||||
btMessageDispatcher->setPieceStorage(pieceStorage.get());
|
|
||||||
auto slot = make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece);
|
auto slot = make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece);
|
||||||
// make this slot timeout
|
// make this slot timeout
|
||||||
slot->setDispatchedTime(0);
|
slot->setDispatchedTime(0);
|
||||||
|
@ -279,11 +247,7 @@ void DefaultBtMessageDispatcherTest::
|
||||||
testCheckRequestSlotAndDoNecessaryThing_completeBlock() {
|
testCheckRequestSlotAndDoNecessaryThing_completeBlock() {
|
||||||
auto piece = std::make_shared<Piece>(0, MY_PIECE_LENGTH);
|
auto piece = std::make_shared<Piece>(0, MY_PIECE_LENGTH);
|
||||||
piece->completeBlock(0);
|
piece->completeBlock(0);
|
||||||
auto pieceStorage = make_unique<MockPieceStorage2>();
|
|
||||||
pieceStorage->setPiece(piece);
|
|
||||||
|
|
||||||
btMessageDispatcher->setRequestTimeout(60);
|
btMessageDispatcher->setRequestTimeout(60);
|
||||||
btMessageDispatcher->setPieceStorage(pieceStorage.get());
|
|
||||||
btMessageDispatcher->addOutstandingRequest
|
btMessageDispatcher->addOutstandingRequest
|
||||||
(make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece));
|
(make_unique<RequestSlot>(0, 0, MY_PIECE_LENGTH, 0, piece));
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue