blob: 549d759976e58b54300919fe16f7905af8c3d110 [file] [log] [blame]
Alexander Afanasyev71b43e72012-12-27 01:03:43 -08001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
2/*
Alexander Afanasyeva199f972013-01-02 19:37:26 -08003 * Copyright (c) 2013 University of California, Los Angeles
Alexander Afanasyev71b43e72012-12-27 01:03:43 -08004 *
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 SYNC_LOG_H
23#define SYNC_LOG_H
24
Alexander Afanasyeva199f972013-01-02 19:37:26 -080025#include "db-helper.h"
Alexander Afanasyev6f70a0f2013-01-02 20:44:09 -080026#include <sync-state.pb.h>
Alexander Afanasyevd09871f2013-01-04 22:36:37 -080027#include <ccnx-name.h>
Zhenkai Zhue851b952013-01-13 22:29:57 -080028#include <map>
Alexander Afanasyevbeee0b42013-01-16 18:25:08 -080029#include <boost/thread/shared_mutex.hpp>
Alexander Afanasyev6f70a0f2013-01-02 20:44:09 -080030
31typedef boost::shared_ptr<SyncStateMsg> SyncStateMsgPtr;
Alexander Afanasyev71b43e72012-12-27 01:03:43 -080032
Alexander Afanasyeva199f972013-01-02 19:37:26 -080033class SyncLog : public DbHelper
34{
35public:
Alexander Afanasyev68f2a952013-01-08 14:34:16 -080036 SyncLog (const boost::filesystem::path &path, const std::string &localName);
Alexander Afanasyev71b43e72012-12-27 01:03:43 -080037
Zhenkai Zhue851b952013-01-13 22:29:57 -080038 // fill in the map with device-name <-> locator pairs
39 void
40 initYP(map<Ccnx::Name, Ccnx::Name> &yp);
41
Alexander Afanasyev433ecda2013-01-02 22:13:45 -080042 sqlite3_int64
43 GetNextLocalSeqNo (); // side effect: local seq_no will be increased
44
Alexander Afanasyeva199f972013-01-02 19:37:26 -080045 // done
46 void
Zhenkai Zhue851b952013-01-13 22:29:57 -080047 UpdateDeviceSeqNo (const Ccnx::Name &name, sqlite3_int64 seqNo);
Alexander Afanasyev71b43e72012-12-27 01:03:43 -080048
Zhenkai Zhue851b952013-01-13 22:29:57 -080049 Ccnx::Name
50 LookupLocator (const Ccnx::Name &deviceName);
Alexander Afanasyev71b43e72012-12-27 01:03:43 -080051
Zhenkai Zhue851b952013-01-13 22:29:57 -080052 void
53 UpdateLocator (const Ccnx::Name &deviceName, const Ccnx::Name &locator);
Alexander Afanasyev71b43e72012-12-27 01:03:43 -080054
Alexander Afanasyeva199f972013-01-02 19:37:26 -080055 // done
56 /**
57 * Create an entry in SyncLog and SyncStateNodes corresponding to the current state of SyncNodes
58 */
59 HashPtr
60 RememberStateInStateLog ();
61
62 // done
63 sqlite3_int64
64 LookupSyncLog (const std::string &stateHash);
65
66 // done
67 sqlite3_int64
68 LookupSyncLog (const Hash &stateHash);
69
70 // How difference is exposed will be determined later by the actual protocol
Alexander Afanasyev6f70a0f2013-01-02 20:44:09 -080071 SyncStateMsgPtr
Zhenkai Zhu085aae72013-01-17 21:09:01 -080072 FindStateDifferences (const std::string &oldHash, const std::string &newHash, bool includeOldSeq = false);
Alexander Afanasyeva199f972013-01-02 19:37:26 -080073
Alexander Afanasyev6f70a0f2013-01-02 20:44:09 -080074 SyncStateMsgPtr
Zhenkai Zhu085aae72013-01-17 21:09:01 -080075 FindStateDifferences (const Hash &oldHash, const Hash &newHash, bool includeOldSeq = false);
Alexander Afanasyev433ecda2013-01-02 22:13:45 -080076
Zhenkai Zhu9501b8b2013-01-17 12:37:00 -080077 //-------- only used in test -----------------
78 sqlite3_int64
79 SeqNo(const Ccnx::Name &name);
80
Alexander Afanasyev433ecda2013-01-02 22:13:45 -080081protected:
Zhenkai Zhue851b952013-01-13 22:29:57 -080082 void
83 UpdateDeviceSeqNo (sqlite3_int64 deviceId, sqlite3_int64 seqNo);
84
85
86protected:
Alexander Afanasyevd09871f2013-01-04 22:36:37 -080087 // std::string m_localName;
88 Ccnx::Name m_localName;
Alexander Afanasyev433ecda2013-01-02 22:13:45 -080089
Alexander Afanasyevd09871f2013-01-04 22:36:37 -080090 sqlite3_int64 m_localDeviceId;
Alexander Afanasyevbeee0b42013-01-16 18:25:08 -080091
92 typedef boost::mutex Mutex;
93 typedef boost::unique_lock<Mutex> WriteLock;
94
95 Mutex m_stateUpdateMutex;
Alexander Afanasyeva199f972013-01-02 19:37:26 -080096};
97
Zhenkai Zhub330aed2013-01-17 13:29:37 -080098typedef boost::shared_ptr<SyncLog> SyncLogPtr;
99
Alexander Afanasyev71b43e72012-12-27 01:03:43 -0800100
101#endif // SYNC_LOG_H