blob: 9fba688ac3d6a30542df02d88514c4ccf5711d38 [file] [log] [blame]
Alexander Afanasyevdfe58192013-01-17 17:34:04 -08001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
2/*
3 * Copyright (c) 2012-2013 University of California, Los Angeles
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation;
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
18 * Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
19 * Zhenkai Zhu <zhenkai@cs.ucla.edu>
20 */
21
22#ifndef FETCHER_H
23#define FETCHER_H
24
Alexander Afanasyev49a18522013-01-18 17:49:04 -080025#include "ccnx-wrapper.h"
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080026#include "ccnx-name.h"
27
Alexander Afanasyev49a18522013-01-18 17:49:04 -080028#include "scheduler.h"
Alexander Afanasyev83531a42013-01-19 16:21:54 -080029#include <boost/intrusive/list.hpp>
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080030#include <boost/date_time/posix_time/posix_time_types.hpp>
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080031
Alexander Afanasyev76d4fa42013-01-19 14:51:21 -080032class FetchManager;
33
Alexander Afanasyev83531a42013-01-19 16:21:54 -080034class Fetcher
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080035{
36public:
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080037 typedef boost::function<void(Ccnx::Name &deviceName, Ccnx::Name &baseName, uint64_t seq, Ccnx::PcoPtr pco)> SegmentCallback;
38 typedef boost::function<void(Ccnx::Name &deviceName, Ccnx::Name &baseName)> FinishCallback;
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080039 typedef boost::function<void (Fetcher &)> OnFetchCompleteCallback;
40 typedef boost::function<void (Fetcher &)> OnFetchFailedCallback;
41
42 Fetcher (Ccnx::CcnxWrapperPtr ccnx,
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080043 const SegmentCallback &segmentCallback, // callback passed by caller of FetchManager
44 const FinishCallback &finishCallback, // callback passed by caller of FetchManager
45 OnFetchCompleteCallback onFetchComplete, OnFetchFailedCallback onFetchFailed, // callbacks provided by FetchManager
46 const Ccnx::Name &deviceName, const Ccnx::Name &name, int64_t minSeqNo, int64_t maxSeqNo,
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080047 boost::posix_time::time_duration timeout = boost::posix_time::seconds (30), // this time is not precise, but sets min bound
48 // actual time depends on how fast Interests timeout
Alexander Afanasyev49a18522013-01-18 17:49:04 -080049 const Ccnx::Name &forwardingHint = Ccnx::Name ());
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080050 virtual ~Fetcher ();
Alexander Afanasyev76d4fa42013-01-19 14:51:21 -080051
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080052 inline bool
53 IsActive () const;
54
Alexander Afanasyev76d4fa42013-01-19 14:51:21 -080055 void
Alexander Afanasyev83531a42013-01-19 16:21:54 -080056 RestartPipeline ();
Alexander Afanasyev76d4fa42013-01-19 14:51:21 -080057
58 void
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080059 SetForwardingHint (const Ccnx::Name &forwardingHint);
60
61private:
62 void
Alexander Afanasyev50547892013-01-19 22:03:45 -080063 FillPipeline ();
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080064
65 void
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080066 OnData (uint64_t seqno, const Ccnx::Name &name, Ccnx::PcoPtr data);
Alexander Afanasyev50547892013-01-19 22:03:45 -080067
68 Ccnx::Closure::TimeoutCallbackReturnValue
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080069 OnTimeout (uint64_t seqno, const Ccnx::Name &name);
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080070
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080071public:
72 boost::intrusive::list_member_hook<> m_managerListHook;
73
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080074private:
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080075 Ccnx::CcnxWrapperPtr m_ccnx;
76
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080077 SegmentCallback m_segmentCallback;
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080078 OnFetchCompleteCallback m_onFetchComplete;
79 OnFetchFailedCallback m_onFetchFailed;
80
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080081 FinishCallback m_finishCallback;
82
Alexander Afanasyev83531a42013-01-19 16:21:54 -080083 bool m_active;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080084
Alexander Afanasyev49a18522013-01-18 17:49:04 -080085 Ccnx::Name m_name;
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080086 Ccnx::Name m_deviceName;
Alexander Afanasyev49a18522013-01-18 17:49:04 -080087 Ccnx::Name m_forwardingHint;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080088
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080089 boost::posix_time::time_duration m_maximumNoActivityPeriod;
90
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080091 int64_t m_minSendSeqNo;
92 int64_t m_maxInOrderRecvSeqNo;
93 std::set<int64_t> m_outOfOrderRecvSeqNo;
Alexander Afanasyev50547892013-01-19 22:03:45 -080094
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080095 int64_t m_minSeqNo;
96 int64_t m_maxSeqNo;
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080097
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080098 uint64_t m_pipeline;
99 uint64_t m_activePipeline;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -0800100
Alexander Afanasyev21a166e2013-01-20 16:04:41 -0800101 boost::posix_time::ptime m_lastPositiveActivity;
102
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800103};
104
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -0800105typedef boost::error_info<struct tag_errmsg, std::string> errmsg_info_str;
Alexander Afanasyev49a18522013-01-18 17:49:04 -0800106
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800107namespace Error {
108struct Fetcher : virtual boost::exception, virtual std::exception { };
109}
110
111typedef boost::shared_ptr<Fetcher> FetcherPtr;
112
Alexander Afanasyev21a166e2013-01-20 16:04:41 -0800113bool
114Fetcher::IsActive () const
115{
116 return m_active;
117}
118
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800119
120#endif // FETCHER_H