Implemented set to monitor pending interests in manager.
Change-Id: I4a11db79053e69a19ceba0e0501b93911131d1d7
diff --git a/src/sequential-data-fetcher.cpp b/src/sequential-data-fetcher.cpp
index 9a5ae5f..472cd71 100644
--- a/src/sequential-data-fetcher.cpp
+++ b/src/sequential-data-fetcher.cpp
@@ -26,12 +26,14 @@
namespace ndn {
namespace ntorrent {
-SequentialDataFetcher::SequentialDataFetcher(const ndn::Name& torrentFileName,
- const std::string& dataPath)
+SequentialDataFetcher::SequentialDataFetcher(const ndn::Name& torrentFileName,
+ const std::string& dataPath,
+ bool seed)
: m_dataPath(dataPath)
, m_torrentFileName(torrentFileName)
+ , m_seedFlag(seed)
{
- m_manager = make_shared<TorrentManager>(m_torrentFileName, m_dataPath);
+ m_manager = make_shared<TorrentManager>(m_torrentFileName, m_dataPath, seed);
}
SequentialDataFetcher::~SequentialDataFetcher()
@@ -121,6 +123,9 @@
}
else {
LOG_INFO << "All data complete" << std::endl;
+ if (!m_seedFlag) {
+ m_manager->shutdown();
+ }
}
}
}
@@ -136,6 +141,8 @@
void
SequentialDataFetcher::onTorrentFileSegmentReceived(const std::vector<Name>& manifestNames)
{
+ // TODO(msweatt) Add parameter for torrent file
+ LOG_INFO << "Torrent Segment Received: " << m_torrentFileName << std::endl;
this->downloadManifestFiles(manifestNames);
}