blob: 1be6a27d439f094fb63e345013b76e4d825a241c [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 Afanasyevff8d9dc2013-01-26 00:45:08 -080028#include "executor.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
Alexander Afanasyevff8d9dc2013-01-26 00:45:08 -080061 const Ccnx::Name &
62 GetForwardingHint () const { return m_forwardingHint; }
63
64 const Ccnx::Name &
65 GetName () const { return m_name; }
66
67
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080068private:
69 void
Alexander Afanasyev50547892013-01-19 22:03:45 -080070 FillPipeline ();
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080071
72 void
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080073 OnData (uint64_t seqno, const Ccnx::Name &name, Ccnx::PcoPtr data);
Alexander Afanasyev50547892013-01-19 22:03:45 -080074
75 Ccnx::Closure::TimeoutCallbackReturnValue
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080076 OnTimeout (uint64_t seqno, const Ccnx::Name &name);
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080077
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080078public:
79 boost::intrusive::list_member_hook<> m_managerListHook;
80
Alexander Afanasyevdfe58192013-01-17 17:34:04 -080081private:
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080082 Ccnx::CcnxWrapperPtr m_ccnx;
83
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080084 SegmentCallback m_segmentCallback;
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080085 OnFetchCompleteCallback m_onFetchComplete;
86 OnFetchFailedCallback m_onFetchFailed;
87
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080088 FinishCallback m_finishCallback;
89
Alexander Afanasyev83531a42013-01-19 16:21:54 -080090 bool m_active;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080091
Alexander Afanasyev49a18522013-01-18 17:49:04 -080092 Ccnx::Name m_name;
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080093 Ccnx::Name m_deviceName;
Alexander Afanasyev49a18522013-01-18 17:49:04 -080094 Ccnx::Name m_forwardingHint;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -080095
Alexander Afanasyev21a166e2013-01-20 16:04:41 -080096 boost::posix_time::time_duration m_maximumNoActivityPeriod;
97
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -080098 int64_t m_minSendSeqNo;
99 int64_t m_maxInOrderRecvSeqNo;
100 std::set<int64_t> m_outOfOrderRecvSeqNo;
Alexander Afanasyev50547892013-01-19 22:03:45 -0800101
Zhenkai Zhu3d1beca2013-01-23 14:55:32 -0800102 int64_t m_minSeqNo;
103 int64_t m_maxSeqNo;
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800104
Alexander Afanasyevf9978f82013-01-23 16:30:31 -0800105 uint32_t m_pipeline;
106 uint32_t m_activePipeline;
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -0800107
Alexander Afanasyev21a166e2013-01-20 16:04:41 -0800108 boost::posix_time::ptime m_lastPositiveActivity;
Alexander Afanasyevff8d9dc2013-01-26 00:45:08 -0800109 Executor m_executor;
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800110};
111
Alexander Afanasyevd6c2a902013-01-19 21:24:30 -0800112typedef boost::error_info<struct tag_errmsg, std::string> errmsg_info_str;
Alexander Afanasyev49a18522013-01-18 17:49:04 -0800113
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800114namespace Error {
115struct Fetcher : virtual boost::exception, virtual std::exception { };
116}
117
118typedef boost::shared_ptr<Fetcher> FetcherPtr;
119
Alexander Afanasyev21a166e2013-01-20 16:04:41 -0800120bool
121Fetcher::IsActive () const
122{
123 return m_active;
124}
125
Alexander Afanasyevdfe58192013-01-17 17:34:04 -0800126
127#endif // FETCHER_H