Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
Junxiao Shi | 047a6fb | 2017-06-08 16:16:05 +0000 | [diff] [blame^] | 3 | * Copyright (c) 2014-2017, Regents of the University of California. |
Alexander Afanasyev | e1e6f2a | 2014-04-25 11:28:12 -0700 | [diff] [blame] | 4 | * |
| 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 Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 18 | */ |
| 19 | |
Alexander Afanasyev | 39d9807 | 2014-05-04 12:46:29 -0700 | [diff] [blame] | 20 | #ifndef REPO_REPO_HPP |
| 21 | #define REPO_REPO_HPP |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 22 | |
Weiqi Shi | f0330d5 | 2014-07-09 10:54:27 -0700 | [diff] [blame] | 23 | //#include "storage/repo_storage.hpp" |
| 24 | #include "storage/sqlite-storage.hpp" |
| 25 | #include "storage/repo-storage.hpp" |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 26 | |
Alexander Afanasyev | 39d9807 | 2014-05-04 12:46:29 -0700 | [diff] [blame] | 27 | #include "handles/read-handle.hpp" |
| 28 | #include "handles/write-handle.hpp" |
Weiqi Shi | 098f91c | 2014-07-23 17:41:35 -0700 | [diff] [blame] | 29 | #include "handles/watch-handle.hpp" |
Alexander Afanasyev | 39d9807 | 2014-05-04 12:46:29 -0700 | [diff] [blame] | 30 | #include "handles/delete-handle.hpp" |
| 31 | #include "handles/tcp-bulk-insert-handle.hpp" |
| 32 | |
Shuo Chen | 028dcd3 | 2014-06-21 16:36:44 +0800 | [diff] [blame] | 33 | #include "common.hpp" |
Alexander Afanasyev | 39d9807 | 2014-05-04 12:46:29 -0700 | [diff] [blame] | 34 | |
Junxiao Shi | 047a6fb | 2017-06-08 16:16:05 +0000 | [diff] [blame^] | 35 | #include <ndn-cxx/security/validator-null.hpp> |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 36 | #include <boost/property_tree/ptree.hpp> |
| 37 | #include <boost/property_tree/info_parser.hpp> |
| 38 | |
| 39 | namespace repo { |
| 40 | |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 41 | struct RepoConfig |
| 42 | { |
Wentao Shang | a8f3c40 | 2014-10-30 14:03:27 -0700 | [diff] [blame] | 43 | std::string repoConfigPath; |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 44 | //StorageMethod storageMethod; This will be implemtented if there is other method. |
| 45 | std::string dbPath; |
Wentao Shang | a8f3c40 | 2014-10-30 14:03:27 -0700 | [diff] [blame] | 46 | std::vector<ndn::Name> dataPrefixes; |
| 47 | std::vector<ndn::Name> repoPrefixes; |
| 48 | std::vector<std::pair<std::string, std::string> > tcpBulkInsertEndpoints; |
Shuo Chen | 6e81812 | 2015-05-04 10:40:03 +0800 | [diff] [blame] | 49 | uint64_t nMaxPackets; |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 50 | boost::property_tree::ptree validatorNode; |
| 51 | }; |
| 52 | |
| 53 | RepoConfig |
| 54 | parseConfig(const std::string& confPath); |
| 55 | |
| 56 | class Repo : noncopyable |
| 57 | { |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 58 | public: |
| 59 | class Error : public std::runtime_error |
| 60 | { |
| 61 | public: |
| 62 | explicit |
| 63 | Error(const std::string& what) |
| 64 | : std::runtime_error(what) |
| 65 | { |
| 66 | } |
| 67 | }; |
| 68 | |
| 69 | public: |
| 70 | Repo(boost::asio::io_service& ioService, const RepoConfig& config); |
| 71 | |
Shuo Chen | a12f528 | 2014-08-01 15:18:30 +0800 | [diff] [blame] | 72 | //@brief rebuild index from storage file when repo starts. |
| 73 | void |
| 74 | initializeStorage(); |
| 75 | |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 76 | void |
| 77 | enableListening(); |
| 78 | |
Shuo Chen | 028dcd3 | 2014-06-21 16:36:44 +0800 | [diff] [blame] | 79 | void |
| 80 | enableValidation(); |
| 81 | |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 82 | private: |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 83 | RepoConfig m_config; |
| 84 | ndn::Scheduler m_scheduler; |
| 85 | ndn::Face m_face; |
Wentao Shang | a8f3c40 | 2014-10-30 14:03:27 -0700 | [diff] [blame] | 86 | std::shared_ptr<Storage> m_store; |
Weiqi Shi | f0330d5 | 2014-07-09 10:54:27 -0700 | [diff] [blame] | 87 | RepoStorage m_storageHandle; |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 88 | KeyChain m_keyChain; |
Junxiao Shi | 047a6fb | 2017-06-08 16:16:05 +0000 | [diff] [blame^] | 89 | ndn::ValidatorNull m_validator; |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 90 | ReadHandle m_readHandle; |
| 91 | WriteHandle m_writeHandle; |
Weiqi Shi | 098f91c | 2014-07-23 17:41:35 -0700 | [diff] [blame] | 92 | WatchHandle m_watchHandle; |
Shuo Chen | 478204c | 2014-03-18 18:27:04 -0700 | [diff] [blame] | 93 | DeleteHandle m_deleteHandle; |
| 94 | TcpBulkInsertHandle m_tcpBulkInsertHandle; |
| 95 | }; |
| 96 | |
| 97 | } // namespace repo |
| 98 | |
Alexander Afanasyev | 39d9807 | 2014-05-04 12:46:29 -0700 | [diff] [blame] | 99 | #endif // REPO_REPO_HPP |