diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2010-06-12 13:49:39 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2010-06-12 13:49:39 +0000 |
| commit | 8d6378b675bef77b8f0bdd60f951b3d132dbf70b (patch) | |
| tree | 2bda56f819e75e8daf2a8e38bf5d1788650a5226 | |
| parent | 95f5c589036c95f58b83441513d9c7c3f16f27bc (diff) | |
2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Renamed member variables in SegmentMan
* src/RequestGroup.cc
* src/RequestGroupMan.cc
* src/SegmentMan.cc
* src/SegmentMan.h
* src/XmlRpcMethodImpl.cc
* test/BtDependencyTest.cc
| -rw-r--r-- | ChangeLog | 10 | ||||
| -rw-r--r-- | src/RequestGroup.cc | 1 | ||||
| -rw-r--r-- | src/RequestGroupMan.cc | 1 | ||||
| -rw-r--r-- | src/SegmentMan.cc | 102 | ||||
| -rw-r--r-- | src/SegmentMan.h | 19 | ||||
| -rw-r--r-- | src/XmlRpcMethodImpl.cc | 1 | ||||
| -rw-r--r-- | test/BtDependencyTest.cc | 1 |
7 files changed, 71 insertions, 64 deletions
@@ -1,5 +1,15 @@ 2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net> + Renamed member variables in SegmentMan + * src/RequestGroup.cc + * src/RequestGroupMan.cc + * src/SegmentMan.cc + * src/SegmentMan.h + * src/XmlRpcMethodImpl.cc + * test/BtDependencyTest.cc + +2010-06-12 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net> + Renamed member variables. * src/RequestSlot.cc * src/RequestSlot.h diff --git a/src/RequestGroup.cc b/src/RequestGroup.cc index 3cfbca5a..0c0ce2f2 100644 --- a/src/RequestGroup.cc +++ b/src/RequestGroup.cc @@ -77,6 +77,7 @@ #include "SocketCore.h" #include "SimpleRandomizer.h" #include "ServerStatMan.h" +#include "Segment.h" #ifdef ENABLE_MESSAGE_DIGEST # include "CheckIntegrityCommand.h" #endif // ENABLE_MESSAGE_DIGEST diff --git a/src/RequestGroupMan.cc b/src/RequestGroupMan.cc index 029a317b..b622c2ec 100644 --- a/src/RequestGroupMan.cc +++ b/src/RequestGroupMan.cc @@ -69,6 +69,7 @@ #include "StringFormat.h" #include "FileAllocationEntry.h" #include "CheckIntegrityEntry.h" +#include "Segment.h" namespace aria2 { diff --git a/src/SegmentMan.cc b/src/SegmentMan.cc index 8b42b632..f7a62264 100644 --- a/src/SegmentMan.cc +++ b/src/SegmentMan.cc @@ -55,16 +55,11 @@ namespace aria2 { -SegmentEntry::SegmentEntry(cuid_t cuid, const SharedHandle<Segment>& segment): - cuid(cuid), segment(segment) {} - -SegmentEntry::~SegmentEntry() {} - SegmentMan::SegmentMan(const Option* option, const SharedHandle<DownloadContext>& downloadContext, const PieceStorageHandle& pieceStorage): _option(option), - logger(LogFactory::getInstance()), + _logger(LogFactory::getInstance()), _downloadContext(downloadContext), _pieceStorage(pieceStorage), _lastPeerStatDlspdMapUpdated(0), @@ -88,7 +83,8 @@ bool SegmentMan::downloadFinished() const void SegmentMan::init() { - // TODO Do we have to do something about DownloadContext and PieceStorage here? + // TODO Do we have to do something about DownloadContext and + // PieceStorage here? } uint64_t SegmentMan::getTotalLength() const @@ -117,9 +113,9 @@ SharedHandle<Segment> SegmentMan::checkoutSegment if(piece.isNull()) { return SharedHandle<Segment>(); } - if(logger->debug()) { - logger->debug("Attach segment#%d to CUID#%s.", - piece->getIndex(), util::itos(cuid).c_str()); + if(_logger->debug()) { + _logger->debug("Attach segment#%d to CUID#%s.", + piece->getIndex(), util::itos(cuid).c_str()); } SharedHandle<Segment> segment; if(piece->getLength() == 0) { @@ -128,22 +124,22 @@ SharedHandle<Segment> SegmentMan::checkoutSegment segment.reset(new PiecedSegment(_downloadContext->getPieceLength(), piece)); } SegmentEntryHandle entry(new SegmentEntry(cuid, segment)); - usedSegmentEntries.push_back(entry); - if(logger->debug()) { - logger->debug("index=%d, length=%d, segmentLength=%d, writtenLength=%d", - segment->getIndex(), - segment->getLength(), - segment->getSegmentLength(), - segment->getWrittenLength()); + _usedSegmentEntries.push_back(entry); + if(_logger->debug()) { + _logger->debug("index=%d, length=%d, segmentLength=%d, writtenLength=%d", + segment->getIndex(), + segment->getLength(), + segment->getSegmentLength(), + segment->getWrittenLength()); } if(piece->getLength() > 0) { std::map<size_t, size_t>::iterator positr = _segmentWrittenLengthMemo.find(segment->getIndex()); if(positr != _segmentWrittenLengthMemo.end()) { const size_t writtenLength = (*positr).second; - if(logger->debug()) { - logger->debug("writtenLength(in memo)=%d, writtenLength=%d", - writtenLength, segment->getWrittenLength()); + if(_logger->debug()) { + _logger->debug("writtenLength(in memo)=%d, writtenLength=%d", + writtenLength, segment->getWrittenLength()); } // If the difference between cached writtenLength and segment's // writtenLength is less than one block, we assume that these @@ -160,8 +156,8 @@ SharedHandle<Segment> SegmentMan::checkoutSegment void SegmentMan::getInFlightSegment (std::vector<SharedHandle<Segment> >& segments, cuid_t cuid) { - for(SegmentEntries::const_iterator itr = usedSegmentEntries.begin(), - eoi = usedSegmentEntries.end(); itr != eoi; ++itr) { + for(SegmentEntries::const_iterator itr = _usedSegmentEntries.begin(), + eoi = _usedSegmentEntries.end(); itr != eoi; ++itr) { const SegmentEntryHandle& segmentEntry = *itr; if(segmentEntry->cuid == cuid) { segments.push_back(segmentEntry->segment); @@ -220,8 +216,8 @@ SharedHandle<Segment> SegmentMan::getCleanSegmentIfOwnerIsIdle if(index > 0 && _downloadContext->getNumPieces() <= index) { return SharedHandle<Segment>(); } - for(SegmentEntries::const_iterator itr = usedSegmentEntries.begin(), - eoi = usedSegmentEntries.end(); itr != eoi; ++itr) { + for(SegmentEntries::const_iterator itr = _usedSegmentEntries.begin(), + eoi = _usedSegmentEntries.end(); itr != eoi; ++itr) { const SharedHandle<SegmentEntry>& segmentEntry = *itr; if(segmentEntry->segment->getIndex() == index) { if(segmentEntry->cuid == cuid) { @@ -245,24 +241,24 @@ SharedHandle<Segment> SegmentMan::getCleanSegmentIfOwnerIsIdle void SegmentMan::cancelSegment(const SharedHandle<Segment>& segment) { - if(logger->debug()) { - logger->debug("Canceling segment#%d", segment->getIndex()); + if(_logger->debug()) { + _logger->debug("Canceling segment#%d", segment->getIndex()); } _pieceStorage->cancelPiece(segment->getPiece()); _segmentWrittenLengthMemo[segment->getIndex()] = segment->getWrittenLength(); - if(logger->debug()) { - logger->debug("Memorized segment index=%u, writtenLength=%u", - segment->getIndex(), segment->getWrittenLength()); + if(_logger->debug()) { + _logger->debug("Memorized segment index=%u, writtenLength=%u", + segment->getIndex(), segment->getWrittenLength()); } } void SegmentMan::cancelSegment(cuid_t cuid) { - for(SegmentEntries::iterator itr = usedSegmentEntries.begin(), - eoi = usedSegmentEntries.end(); itr != eoi;) { + for(SegmentEntries::iterator itr = _usedSegmentEntries.begin(), + eoi = _usedSegmentEntries.end(); itr != eoi;) { if((*itr)->cuid == cuid) { cancelSegment((*itr)->segment); - itr = usedSegmentEntries.erase(itr); - eoi = usedSegmentEntries.end(); + itr = _usedSegmentEntries.erase(itr); + eoi = _usedSegmentEntries.end(); } else { ++itr; } @@ -272,12 +268,12 @@ void SegmentMan::cancelSegment(cuid_t cuid) { void SegmentMan::cancelSegment (cuid_t cuid, const SharedHandle<Segment>& segment) { - for(SegmentEntries::iterator itr = usedSegmentEntries.begin(), - eoi = usedSegmentEntries.end(); itr != eoi;) { + for(SegmentEntries::iterator itr = _usedSegmentEntries.begin(), + eoi = _usedSegmentEntries.end(); itr != eoi;) { if((*itr)->cuid == cuid && (*itr)->segment == segment) { cancelSegment((*itr)->segment); - itr = usedSegmentEntries.erase(itr); - //eoi = usedSegmentEntries.end(); + itr = _usedSegmentEntries.erase(itr); + //eoi = _usedSegmentEntries.end(); break; } else { ++itr; @@ -288,11 +284,11 @@ void SegmentMan::cancelSegment void SegmentMan::cancelAllSegments() { for(std::deque<SharedHandle<SegmentEntry> >::iterator itr = - usedSegmentEntries.begin(), eoi = usedSegmentEntries.end(); + _usedSegmentEntries.begin(), eoi = _usedSegmentEntries.end(); itr != eoi; ++itr) { cancelSegment((*itr)->segment); } - usedSegmentEntries.clear(); + _usedSegmentEntries.clear(); } void SegmentMan::eraseSegmentWrittenLengthMemo() @@ -316,13 +312,13 @@ bool SegmentMan::completeSegment (cuid_t cuid, const SharedHandle<Segment>& segment) { _pieceStorage->completePiece(segment->getPiece()); _pieceStorage->advertisePiece(cuid, segment->getPiece()->getIndex()); - SegmentEntries::iterator itr = std::find_if(usedSegmentEntries.begin(), - usedSegmentEntries.end(), + SegmentEntries::iterator itr = std::find_if(_usedSegmentEntries.begin(), + _usedSegmentEntries.end(), FindSegmentEntry(segment)); - if(itr == usedSegmentEntries.end()) { + if(itr == _usedSegmentEntries.end()) { return false; } else { - usedSegmentEntries.erase(itr); + _usedSegmentEntries.erase(itr); return true; } } @@ -341,20 +337,20 @@ uint64_t SegmentMan::getDownloadLength() const { void SegmentMan::registerPeerStat(const SharedHandle<PeerStat>& peerStat) { - for(std::vector<SharedHandle<PeerStat> >::iterator i = peerStats.begin(), - eoi = peerStats.end(); i != eoi; ++i) { + for(std::vector<SharedHandle<PeerStat> >::iterator i = _peerStats.begin(), + eoi = _peerStats.end(); i != eoi; ++i) { if((*i)->getStatus() == PeerStat::IDLE) { *i = peerStat; return; } } - peerStats.push_back(peerStat); + _peerStats.push_back(peerStat); } SharedHandle<PeerStat> SegmentMan::getPeerStat(cuid_t cuid) const { for(std::vector<SharedHandle<PeerStat> >::const_iterator i = - peerStats.begin(), eoi = peerStats.end(); i != eoi; ++i) { + _peerStats.begin(), eoi = _peerStats.end(); i != eoi; ++i) { if((*i)->getCuid() == cuid) { return *i; } @@ -401,7 +397,7 @@ unsigned int SegmentMan::calculateDownloadSpeed() _lastPeerStatDlspdMapUpdated = global::wallclock; _peerStatDlspdMap.clear(); for(std::vector<SharedHandle<PeerStat> >::const_iterator i = - peerStats.begin(), eoi = peerStats.end(); i != eoi; ++i) { + _peerStats.begin(), eoi = _peerStats.end(); i != eoi; ++i) { if((*i)->getStatus() == PeerStat::ACTIVE) { unsigned int s = (*i)->calculateDownloadSpeed(); _peerStatDlspdMap[(*i)->getCuid()] = s; @@ -455,11 +451,11 @@ size_t SegmentMan::countFreePieceFrom(size_t index) const void SegmentMan::ignoreSegmentFor(const SharedHandle<FileEntry>& fileEntry) { - if(logger->debug()) { - logger->debug("ignoring segment for path=%s, offset=%s, length=%s", - fileEntry->getPath().c_str(), - util::itos(fileEntry->getOffset()).c_str(), - util::uitos(fileEntry->getLength()).c_str()); + if(_logger->debug()) { + _logger->debug("ignoring segment for path=%s, offset=%s, length=%s", + fileEntry->getPath().c_str(), + util::itos(fileEntry->getOffset()).c_str(), + util::uitos(fileEntry->getLength()).c_str()); } _ignoreBitfield.addFilter(fileEntry->getOffset(), fileEntry->getLength()); } diff --git a/src/SegmentMan.h b/src/SegmentMan.h index b9e8f694..9f3bc13e 100644 --- a/src/SegmentMan.h +++ b/src/SegmentMan.h @@ -57,14 +57,11 @@ class PieceStorage; class Piece; class FileEntry; -class SegmentEntry { -public: +struct SegmentEntry { cuid_t cuid; SharedHandle<Segment> segment; -public: - SegmentEntry(cuid_t cuid, const SharedHandle<Segment>& segment); - - ~SegmentEntry(); + SegmentEntry(cuid_t cuid, const SharedHandle<Segment>& segment): + cuid(cuid), segment(segment) {} }; typedef SharedHandle<SegmentEntry> SegmentEntryHandle; @@ -77,20 +74,20 @@ class SegmentMan { private: const Option* _option; - Logger* logger; + Logger* _logger; SharedHandle<DownloadContext> _downloadContext; SharedHandle<PieceStorage> _pieceStorage; - SegmentEntries usedSegmentEntries; + SegmentEntries _usedSegmentEntries; // Remember writtenLength for each segment. The key is an index of a // segment. The value is writtenLength for that segment. std::map<size_t, size_t> _segmentWrittenLengthMemo; // Used for calculating download speed. - std::vector<SharedHandle<PeerStat> > peerStats; + std::vector<SharedHandle<PeerStat> > _peerStats; // Keep track of fastest PeerStat for each server std::vector<SharedHandle<PeerStat> > _fastestPeerStats; @@ -207,14 +204,14 @@ public: uint64_t getDownloadLength() const; - // If there is inactive PeerStat in peerStats, it is replaced with + // If there is inactive PeerStat in _peerStats, it is replaced with // given peerStat. If no such PeerStat exist, the given peerStat is // inserted. void registerPeerStat(const SharedHandle<PeerStat>& peerStat); const std::vector<SharedHandle<PeerStat> >& getPeerStats() const { - return peerStats; + return _peerStats; } SharedHandle<PeerStat> getPeerStat(cuid_t cuid) const; diff --git a/src/XmlRpcMethodImpl.cc b/src/XmlRpcMethodImpl.cc index ccb724da..2dbffb1d 100644 --- a/src/XmlRpcMethodImpl.cc +++ b/src/XmlRpcMethodImpl.cc @@ -67,6 +67,7 @@ #include "ServerStatMan.h" #include "FileAllocationEntry.h" #include "CheckIntegrityEntry.h" +#include "Segment.h" #ifdef ENABLE_BITTORRENT # include "bittorrent_helper.h" # include "BtRegistry.h" diff --git a/test/BtDependencyTest.cc b/test/BtDependencyTest.cc index 856e7835..22e8a823 100644 --- a/test/BtDependencyTest.cc +++ b/test/BtDependencyTest.cc @@ -10,6 +10,7 @@ #include "Option.h" #include "Exception.h" #include "SegmentMan.h" +#include "Segment.h" #include "FileEntry.h" #include "PieceSelector.h" #include "bittorrent_helper.h" |
