github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity

DefaultPeerStorage: Code cleanup

pull/150/head
Tatsuhiro Tsujikawa 2013-10-26 21:26:41 +09:00
parent 234dd76cd4
commit 6d2dd2ab18
2 changed files with 20 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/DefaultPeerStorage.cc
View File

@ -60,15 +60,13 @@ const size_t MAX_PEER_LIST_UPDATE = 100;
DefaultPeerStorage::DefaultPeerStorage()
: maxPeerListSize_(MAX_PEER_LIST_SIZE),
seederStateChoke_(new BtSeederStateChoke()),
leecherStateChoke_(new BtLeecherStateChoke()),
seederStateChoke_(make_unique<BtSeederStateChoke>()),
leecherStateChoke_(make_unique<BtLeecherStateChoke>()),
lastTransferStatMapUpdated_(0)
{}
DefaultPeerStorage::~DefaultPeerStorage()
{
delete seederStateChoke_;
delete leecherStateChoke_;
assert(uniqPeers_.size() == unusedPeers_.size() + usedPeers_.size());
}
@ -112,13 +110,14 @@ bool DefaultPeerStorage::addPeer(const std::shared_ptr<Peer>& peer)
return true;
}
void DefaultPeerStorage::addPeer(const std::vector<std::shared_ptr<Peer> >& peers)
void DefaultPeerStorage::addPeer
(const std::vector<std::shared_ptr<Peer>>& peers)
{
size_t added = 0;
size_t addMax = std::min(maxPeerListSize_, MAX_PEER_LIST_UPDATE);
for(auto itr = peers.begin(),
eoi = peers.end(); itr != eoi && added < addMax; ++itr) {
const std::shared_ptr<Peer>& peer = *itr;
for(auto itr = std::begin(peers), eoi = std::end(peers);
itr != eoi && added < addMax; ++itr) {
auto& peer = *itr;
if(isPeerAlreadyAdded(peer)) {
A2_LOG_DEBUG(fmt("Adding %s:%u is rejected because it has been already"
" added.",
@ -148,8 +147,8 @@ void DefaultPeerStorage::addDroppedPeer(const std::shared_ptr<Peer>& peer)
{
// Make sure that no duplicated peer exists in droppedPeers_. If
// exists, erase older one.
for(auto i = droppedPeers_.begin(),
eoi = droppedPeers_.end(); i != eoi; ++i) {
for(auto i = std::begin(droppedPeers_), eoi = std::end(droppedPeers_);
i != eoi; ++i) {
if((*i)->getIPAddress() == peer->getIPAddress() &&
(*i)->getPort() == peer->getPort()) {
droppedPeers_.erase(i);
@ -229,7 +228,7 @@ std::shared_ptr<Peer> DefaultPeerStorage::checkoutPeer(cuid_t cuid)
if(!isPeerAvailable()) {
return nullptr;
}
std::shared_ptr<Peer> peer = unusedPeers_.front();
auto peer = unusedPeers_.front();
unusedPeers_.pop_front();
if(peer->usedBy() != 0) {
A2_LOG_WARN(fmt("CUID#%" PRId64 " is already set for peer %s:%u",

4
src/DefaultPeerStorage.h
View File

@ -66,8 +66,8 @@ private:
std::deque<std::shared_ptr<Peer>> droppedPeers_;
BtSeederStateChoke* seederStateChoke_;
BtLeecherStateChoke* leecherStateChoke_;
std::unique_ptr<BtSeederStateChoke> seederStateChoke_;
std::unique_ptr<BtLeecherStateChoke> leecherStateChoke_;
Timer lastTransferStatMapUpdated_;