blob: e417960cbd60d52974c3440c9ca22f6bc57eefcf [file] [log] [blame]
Shuo Chen478204c2014-03-18 18:27:04 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Alexander Afanasyevc0e26582017-08-13 21:16:49 -04002/*
weijia yuan82cf9142018-10-21 12:25:02 -07003 * Copyright (c) 2014-2018, Regents of the University of California.
Alexander Afanasyeve1e6f2a2014-04-25 11:28:12 -07004 *
5 * This file is part of NDN repo-ng (Next generation of NDN repository).
6 * See AUTHORS.md for complete list of repo-ng authors and contributors.
7 *
8 * repo-ng is free software: you can redistribute it and/or modify it under the terms
9 * of the GNU General Public License as published by the Free Software Foundation,
10 * either version 3 of the License, or (at your option) any later version.
11 *
12 * repo-ng is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
13 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
14 * PURPOSE. See the GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along with
17 * repo-ng, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
Shuo Chen478204c2014-03-18 18:27:04 -070018 */
19
Alexander Afanasyev39d98072014-05-04 12:46:29 -070020#ifndef REPO_REPO_HPP
21#define REPO_REPO_HPP
Shuo Chen478204c2014-03-18 18:27:04 -070022
Weiqi Shif0330d52014-07-09 10:54:27 -070023#include "storage/repo-storage.hpp"
weijia yuan82cf9142018-10-21 12:25:02 -070024#include "storage/sqlite-storage.hpp"
Shuo Chen478204c2014-03-18 18:27:04 -070025
Alexander Afanasyev39d98072014-05-04 12:46:29 -070026#include "handles/delete-handle.hpp"
weijia yuan82cf9142018-10-21 12:25:02 -070027#include "handles/read-handle.hpp"
Alexander Afanasyev39d98072014-05-04 12:46:29 -070028#include "handles/tcp-bulk-insert-handle.hpp"
weijia yuan82cf9142018-10-21 12:25:02 -070029#include "handles/watch-handle.hpp"
30#include "handles/write-handle.hpp"
Alexander Afanasyev39d98072014-05-04 12:46:29 -070031
Shuo Chen028dcd32014-06-21 16:36:44 +080032#include "common.hpp"
Alexander Afanasyev39d98072014-05-04 12:46:29 -070033
weijia yuan82cf9142018-10-21 12:25:02 -070034#include <ndn-cxx/mgmt/dispatcher.hpp>
Alexander Afanasyevc0e26582017-08-13 21:16:49 -040035#include <ndn-cxx/security/validator-config.hpp>
weijia yuan82cf9142018-10-21 12:25:02 -070036
Shuo Chen478204c2014-03-18 18:27:04 -070037#include <boost/property_tree/info_parser.hpp>
weijia yuan82cf9142018-10-21 12:25:02 -070038#include <boost/property_tree/ptree.hpp>
Shuo Chen478204c2014-03-18 18:27:04 -070039
40namespace repo {
41
Shuo Chen478204c2014-03-18 18:27:04 -070042struct RepoConfig
43{
Nick Gordon190e4dc2017-10-04 16:54:10 -050044 static const size_t DISABLED_SUBSET_LENGTH = -1;
45
Wentao Shanga8f3c402014-10-30 14:03:27 -070046 std::string repoConfigPath;
Shuo Chen478204c2014-03-18 18:27:04 -070047 //StorageMethod storageMethod; This will be implemtented if there is other method.
48 std::string dbPath;
Wentao Shanga8f3c402014-10-30 14:03:27 -070049 std::vector<ndn::Name> dataPrefixes;
Nick Gordon190e4dc2017-10-04 16:54:10 -050050 size_t registrationSubset = DISABLED_SUBSET_LENGTH;
Wentao Shanga8f3c402014-10-30 14:03:27 -070051 std::vector<ndn::Name> repoPrefixes;
52 std::vector<std::pair<std::string, std::string> > tcpBulkInsertEndpoints;
Shuo Chen6e818122015-05-04 10:40:03 +080053 uint64_t nMaxPackets;
Shuo Chen478204c2014-03-18 18:27:04 -070054 boost::property_tree::ptree validatorNode;
55};
56
57RepoConfig
58parseConfig(const std::string& confPath);
59
60class Repo : noncopyable
61{
Shuo Chen478204c2014-03-18 18:27:04 -070062public:
63 class Error : public std::runtime_error
64 {
65 public:
66 explicit
67 Error(const std::string& what)
68 : std::runtime_error(what)
69 {
70 }
71 };
72
73public:
74 Repo(boost::asio::io_service& ioService, const RepoConfig& config);
75
Shuo Chena12f5282014-08-01 15:18:30 +080076 //@brief rebuild index from storage file when repo starts.
77 void
78 initializeStorage();
79
Shuo Chen478204c2014-03-18 18:27:04 -070080 void
81 enableListening();
82
Shuo Chen028dcd32014-06-21 16:36:44 +080083 void
84 enableValidation();
85
Shuo Chen478204c2014-03-18 18:27:04 -070086private:
Shuo Chen478204c2014-03-18 18:27:04 -070087 RepoConfig m_config;
88 ndn::Scheduler m_scheduler;
89 ndn::Face m_face;
weijia yuan82cf9142018-10-21 12:25:02 -070090 ndn::mgmt::Dispatcher m_dispatcher;
Wentao Shanga8f3c402014-10-30 14:03:27 -070091 std::shared_ptr<Storage> m_store;
Weiqi Shif0330d52014-07-09 10:54:27 -070092 RepoStorage m_storageHandle;
Shuo Chen478204c2014-03-18 18:27:04 -070093 KeyChain m_keyChain;
Alexander Afanasyevc0e26582017-08-13 21:16:49 -040094 ValidatorConfig m_validator;
weijia yuan82cf9142018-10-21 12:25:02 -070095
Shuo Chen478204c2014-03-18 18:27:04 -070096 ReadHandle m_readHandle;
97 WriteHandle m_writeHandle;
Weiqi Shi098f91c2014-07-23 17:41:35 -070098 WatchHandle m_watchHandle;
Shuo Chen478204c2014-03-18 18:27:04 -070099 DeleteHandle m_deleteHandle;
100 TcpBulkInsertHandle m_tcpBulkInsertHandle;
101};
102
103} // namespace repo
104
Alexander Afanasyev39d98072014-05-04 12:46:29 -0700105#endif // REPO_REPO_HPP