blob: b804ac5c76f0ca9ae5dbbe6b4b7ef11d208b8d1e [file] [log] [blame]
Zhenkai Zhu8224bb62013-01-06 15:58:02 -08001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
2/*
3 * Copyright (c) 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 *
Alexander Afanasyev8e2104a2013-01-22 10:56:18 -080018 * Author: Zhenkai Zhu <zhenkai@cs.ucla.edu>
19 * Alexander Afanasyev <alexander.afanasyev@ucla.edu>
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080020 */
21
22#ifndef SYNC_CORE_H
23#define SYNC_CORE_H
24
25#include "sync-log.h"
Alexander Afanasyevabe952a2013-01-17 17:06:32 -080026#include "ccnx-wrapper.h"
Zhenkai Zhuff4fa8a2013-01-28 22:02:40 -080027#include "ccnx-selectors.h"
Alexander Afanasyevabe952a2013-01-17 17:06:32 -080028#include "scheduler.h"
Zhenkai Zhu3b406592013-01-25 21:07:49 -080029#include "task.h"
Alexander Afanasyevabe952a2013-01-17 17:06:32 -080030
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080031#include <boost/function.hpp>
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080032
33class SyncCore
34{
35public:
Alexander Afanasyevfc720362013-01-24 21:49:48 -080036 typedef boost::function<void (SyncStateMsgPtr stateMsg) > StateMsgCallback;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080037
Zhenkai Zhue851b952013-01-13 22:29:57 -080038 static const int FRESHNESS = 2; // seconds
39 static const string RECOVER;
40 static const double WAIT; // seconds;
41 static const double RANDOM_PERCENT; // seconds;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080042
43public:
Zhenkai Zhub330aed2013-01-17 13:29:37 -080044 SyncCore(SyncLogPtr syncLog
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080045 , const Ccnx::Name &userName
46 , const Ccnx::Name &localPrefix // routable name used by the local user
47 , const Ccnx::Name &syncPrefix // the prefix for the sync collection
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080048 , const StateMsgCallback &callback // callback when state change is detected
Zhenkai Zhu95160102013-01-25 21:54:57 -080049 , Ccnx::CcnxWrapperPtr ccnx
50 , double syncInterestInterval = -1.0);
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080051 ~SyncCore();
52
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080053 void
Alexander Afanasyevcbda9922013-01-22 11:21:12 -080054 updateLocalPrefix (const Ccnx::Name &localPrefix);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080055
56 void
Alexander Afanasyevcbda9922013-01-22 11:21:12 -080057 localStateChanged ();
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080058
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080059 void
60 updateLocalState (sqlite3_int64);
61
62// ------------------ only used in test -------------------------
63public:
64 HashPtr
65 root() const { return m_rootHash; }
66
67 sqlite3_int64
68 seq (const Ccnx::Name &name);
69
70private:
71 void
72 handleInterest(const Ccnx::Name &name);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080073
74 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080075 handleSyncData(const Ccnx::Name &name, Ccnx::PcoPtr content);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080076
77 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080078 handleRecoverData(const Ccnx::Name &name, Ccnx::PcoPtr content);
79
Zhenkai Zhuff4fa8a2013-01-28 22:02:40 -080080 void
81 handleSyncInterestTimeout(const Ccnx::Name &name, const Ccnx::Closure &closure, Ccnx::Selectors selectors);
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080082
Zhenkai Zhuff4fa8a2013-01-28 22:02:40 -080083 void
84 handleRecoverInterestTimeout(const Ccnx::Name &name, const Ccnx::Closure &closure, Ccnx::Selectors selectors);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080085
86 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080087 deregister(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -080088
89 void
Alexander Afanasyev50526282013-01-26 13:43:57 -080090 recover(HashPtr hash);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080091
Alexander Afanasyevc507ac22013-01-21 16:01:58 -080092private:
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080093 void
94 sendSyncInterest();
95
Zhenkai Zhue851b952013-01-13 22:29:57 -080096 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080097 handleSyncInterest(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -080098
99 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800100 handleRecoverInterest(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -0800101
102 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800103 handleStateData(const Ccnx::Bytes &content);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800104
Alexander Afanasyevc507ac22013-01-21 16:01:58 -0800105private:
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800106 Ccnx::CcnxWrapperPtr m_ccnx;
107
Zhenkai Zhub330aed2013-01-17 13:29:37 -0800108 SyncLogPtr m_log;
Zhenkai Zhu66dc5a92013-01-08 21:41:15 -0800109 SchedulerPtr m_scheduler;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800110 StateMsgCallback m_stateMsgCallback;
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800111
112 Ccnx::Name m_syncPrefix;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800113 HashPtr m_rootHash;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800114
Zhenkai Zhue573ae82013-01-15 13:15:52 -0800115 IntervalGeneratorPtr m_recoverWaitGenerator;
Zhenkai Zhu3b406592013-01-25 21:07:49 -0800116
117 TaskPtr m_sendSyncInterestTask;
Zhenkai Zhu95160102013-01-25 21:54:57 -0800118
119 double m_syncInterestInterval;
Zhenkai Zhu8224bb62013-01-06 15:58:02 -0800120};
121
122#endif // SYNC_CORE_H