Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 1 | /* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */ |
| 2 | /* |
| 3 | * Copyright (c) 2012 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: Zhenkai Zhu <zhenkai@cs.ucla.edu> |
| 19 | * 卞超轶 Chaoyi Bian <bcy@pku.edu.cn> |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 20 | * Alexander Afanasyev <alexander.afanasyev@ucla.edu> |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 21 | */ |
| 22 | |
Chaoyi Bian | 11f294f | 2012-03-08 14:28:06 -0800 | [diff] [blame] | 23 | #ifndef SYNC_LOGIC_H |
| 24 | #define SYNC_LOGIC_H |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 25 | #include <boost/shared_ptr.hpp> |
| 26 | #include <boost/function.hpp> |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 27 | #include "boost/date_time/posix_time/posix_time_types.hpp" |
| 28 | |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 29 | #include "sync-ccnx-wrapper.h" |
| 30 | #include "sync-interest-table.h" |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 31 | #include "sync-diff-state.h" |
| 32 | #include "sync-full-state.h" |
Chaoyi Bian | 4194b74 | 2012-03-08 17:21:35 -0800 | [diff] [blame] | 33 | #include "sync-std-name-info.h" |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 34 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 35 | #include "sync-diff-state-container.h" |
| 36 | |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 37 | namespace Sync { |
| 38 | |
| 39 | /** |
| 40 | * \ingroup sync |
Zhenkai Zhu | aae8152 | 2012-03-06 11:05:44 -0800 | [diff] [blame] | 41 | * @brief A wrapper for SyncApp, which handles ccnx related things (process |
| 42 | * interests and data) |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 43 | */ |
Chaoyi Bian | 11f294f | 2012-03-08 14:28:06 -0800 | [diff] [blame] | 44 | class SyncLogic |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 45 | { |
| 46 | public: |
Chaoyi Bian | 89ee2dc | 2012-03-09 14:06:01 -0800 | [diff] [blame] | 47 | typedef boost::function<void (const std::string &, uint32_t, uint32_t)> LogicCallback; |
| 48 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 49 | /** |
Alexander Afanasyev | 172d2b7 | 2012-03-08 23:43:39 -0800 | [diff] [blame] | 50 | * @brief Constructor |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 51 | * @param syncPrefix the name prefix to use for the Sync Interest |
| 52 | * @param fetch the fetch function, which will be called to actually fetch |
Alexander Afanasyev | 172d2b7 | 2012-03-08 23:43:39 -0800 | [diff] [blame] | 53 | * @param ccnxHandle ccnx handle |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 54 | * the app data when new remote names are learned |
| 55 | */ |
Alexander Afanasyev | 172d2b7 | 2012-03-08 23:43:39 -0800 | [diff] [blame] | 56 | SyncLogic (const std::string &syncPrefix, LogicCallback fetch, CcnxWrapperPtr ccnxHandle); |
Zhenkai Zhu | 46b26a1 | 2012-03-06 13:51:16 -0800 | [diff] [blame] | 57 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 58 | ~SyncLogic (); |
Chaoyi Bian | 89ee2dc | 2012-03-09 14:06:01 -0800 | [diff] [blame] | 59 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 60 | /** |
| 61 | * a wrapper for the same func in SyncApp |
| 62 | */ |
| 63 | void addLocalNames (const std::string &prefix, uint32_t session, uint32_t seq); |
Zhenkai Zhu | 46b26a1 | 2012-03-06 13:51:16 -0800 | [diff] [blame] | 64 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 65 | /** |
| 66 | * @brief respond to the Sync Interest; a lot of logic needs to go in here |
| 67 | * @param interest the Sync Interest in string format |
| 68 | */ |
| 69 | void respondSyncInterest (const std::string &interest); |
Zhenkai Zhu | 46b26a1 | 2012-03-06 13:51:16 -0800 | [diff] [blame] | 70 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 71 | /** |
| 72 | * @brief process the fetched sync data |
Chaoyi Bian | 633255f | 2012-03-09 21:25:38 -0800 | [diff] [blame] | 73 | * @param name the data name |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 74 | * @param dataBuffer the sync data |
| 75 | */ |
| 76 | void processSyncData (const std::string &name, const std::string &dataBuffer); |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 77 | |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 78 | #ifdef _DEBUG |
| 79 | size_t |
| 80 | getListChecksSize () |
| 81 | { |
| 82 | boost::lock_guard<boost::mutex> lock (m_listChecksMutex); |
| 83 | return m_listChecks.size (); |
| 84 | } |
| 85 | #endif |
| 86 | |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 87 | private: |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 88 | void delayedChecksLoop (); |
| 89 | |
| 90 | void |
| 91 | processSyncInterest (DigestConstPtr digest, const std::string &interestname, bool timedProcessing=false); |
| 92 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 93 | void sendSyncInterest (); |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 94 | // void checkAgain (const std::string &interest, DigestPtr digest); |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 95 | |
| 96 | private: |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 97 | typedef std::list< boost::tuple< boost::system_time, boost::function< void ( ) > > > DelayedChecksList; |
| 98 | |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 99 | FullState m_state; |
Chaoyi Bian | 89ee2dc | 2012-03-09 14:06:01 -0800 | [diff] [blame] | 100 | DiffStateContainer m_log; |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 101 | SyncInterestTable m_syncInterestTable; |
| 102 | |
| 103 | std::string m_syncPrefix; |
Alexander Afanasyev | 172d2b7 | 2012-03-08 23:43:39 -0800 | [diff] [blame] | 104 | LogicCallback m_fetch; |
Alexander Afanasyev | c103019 | 2012-03-08 22:21:28 -0800 | [diff] [blame] | 105 | CcnxWrapperPtr m_ccnxHandle; |
Chaoyi Bian | 89ee2dc | 2012-03-09 14:06:01 -0800 | [diff] [blame] | 106 | |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 107 | boost::thread m_delayedCheckThread; |
| 108 | bool m_delayedCheckThreadRunning; |
| 109 | DelayedChecksList m_listChecks; |
| 110 | boost::condition_variable m_listChecksCondition; |
| 111 | boost::mutex m_listChecksMutex; |
Chaoyi Bian | 633255f | 2012-03-09 21:25:38 -0800 | [diff] [blame] | 112 | |
Alexander Afanasyev | 387ac95 | 2012-03-11 23:49:27 -0700 | [diff] [blame^] | 113 | static const boost::posix_time::time_duration m_delayedCheckTime; |
Chaoyi Bian | 89ee2dc | 2012-03-09 14:06:01 -0800 | [diff] [blame] | 114 | static const int m_syncResponseFreshness = 2; |
Zhenkai Zhu | 8d935c8 | 2012-03-06 10:44:12 -0800 | [diff] [blame] | 115 | }; |
| 116 | |
| 117 | |
| 118 | } // Sync |
| 119 | |
Zhenkai Zhu | 1ac6f80 | 2012-03-06 17:40:27 -0800 | [diff] [blame] | 120 | #endif // SYNC_APP_WRAPPER_H |