blob: bd7c5e8663889cbc64ecdf18199a3d60c4eb8ebe [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"
27#include "scheduler.h"
Alexander Afanasyevabe952a2013-01-17 17:06:32 -080028
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080029#include <boost/function.hpp>
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080030
31class SyncCore
32{
33public:
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080034 typedef boost::function<void (const SyncStateMsgPtr & stateMsg) > StateMsgCallback;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080035
Zhenkai Zhue851b952013-01-13 22:29:57 -080036 static const int FRESHNESS = 2; // seconds
37 static const string RECOVER;
38 static const double WAIT; // seconds;
39 static const double RANDOM_PERCENT; // seconds;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080040
41public:
Zhenkai Zhub330aed2013-01-17 13:29:37 -080042 SyncCore(SyncLogPtr syncLog
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080043 , const Ccnx::Name &userName
44 , const Ccnx::Name &localPrefix // routable name used by the local user
45 , const Ccnx::Name &syncPrefix // the prefix for the sync collection
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080046 , const StateMsgCallback &callback // callback when state change is detected
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080047 , Ccnx::CcnxWrapperPtr ccnx
48 , SchedulerPtr scheduler);
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080049 ~SyncCore();
50
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080051 void
Alexander Afanasyevcbda9922013-01-22 11:21:12 -080052 updateLocalPrefix (const Ccnx::Name &localPrefix);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080053
54 void
Alexander Afanasyevcbda9922013-01-22 11:21:12 -080055 localStateChanged ();
Zhenkai Zhu8224bb62013-01-06 15:58:02 -080056
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080057 void
58 updateLocalState (sqlite3_int64);
59
60// ------------------ only used in test -------------------------
61public:
62 HashPtr
63 root() const { return m_rootHash; }
64
65 sqlite3_int64
66 seq (const Ccnx::Name &name);
67
68private:
69 void
70 handleInterest(const Ccnx::Name &name);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080071
72 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080073 handleSyncData(const Ccnx::Name &name, Ccnx::PcoPtr content);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080074
75 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080076 handleRecoverData(const Ccnx::Name &name, Ccnx::PcoPtr content);
77
78 Ccnx::Closure::TimeoutCallbackReturnValue
79 handleSyncInterestTimeout(const Ccnx::Name &name);
80
81 Ccnx::Closure::TimeoutCallbackReturnValue
82 handleRecoverInterestTimeout(const Ccnx::Name &name);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080083
84 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080085 deregister(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -080086
87 void
88 recover(const HashPtr &hash);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080089
Alexander Afanasyevc507ac22013-01-21 16:01:58 -080090private:
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -080091 void
92 sendSyncInterest();
93
Zhenkai Zhue851b952013-01-13 22:29:57 -080094 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080095 handleSyncInterest(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -080096
97 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -080098 handleRecoverInterest(const Ccnx::Name &name);
Zhenkai Zhue851b952013-01-13 22:29:57 -080099
100 void
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800101 handleStateData(const Ccnx::Bytes &content);
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800102
Alexander Afanasyevc507ac22013-01-21 16:01:58 -0800103private:
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800104 Ccnx::CcnxWrapperPtr m_ccnx;
105
Zhenkai Zhub330aed2013-01-17 13:29:37 -0800106 SyncLogPtr m_log;
Zhenkai Zhu66dc5a92013-01-08 21:41:15 -0800107 SchedulerPtr m_scheduler;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800108 StateMsgCallback m_stateMsgCallback;
Alexander Afanasyev49a30d02013-01-21 21:38:48 -0800109
110 Ccnx::Name m_syncPrefix;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800111 HashPtr m_rootHash;
Zhenkai Zhu74dd53c2013-01-10 23:39:57 -0800112
Zhenkai Zhue573ae82013-01-15 13:15:52 -0800113 IntervalGeneratorPtr m_recoverWaitGenerator;
Zhenkai Zhu8224bb62013-01-06 15:58:02 -0800114};
115
116#endif // SYNC_CORE_H