blob: bd1dac21fafcbc957d07494372a101a585bc35cb [file] [log] [blame]
Junxiao Shic041ca32014-02-25 20:01:15 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Teng Liang952d6fd2018-05-29 21:09:52 -07002/*
Davide Pesavento2bdf60c2019-02-19 18:23:45 -05003 * Copyright (c) 2014-2019, Regents of the University of California,
Alexander Afanasyev5d57f972015-02-11 21:04:29 -08004 * Arizona Board of Regents,
5 * Colorado State University,
6 * University Pierre & Marie Curie, Sorbonne University,
7 * Washington University in St. Louis,
8 * Beijing Institute of Technology,
9 * The University of Memphis.
Alexander Afanasyev9bcbc7c2014-04-06 19:37:37 -070010 *
11 * This file is part of NFD (Named Data Networking Forwarding Daemon).
12 * See AUTHORS.md for complete list of NFD authors and contributors.
13 *
14 * NFD is free software: you can redistribute it and/or modify it under the terms
15 * of the GNU General Public License as published by the Free Software Foundation,
16 * either version 3 of the License, or (at your option) any later version.
17 *
18 * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
19 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
20 * PURPOSE. See the GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License along with
23 * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
Alexander Afanasyev5d57f972015-02-11 21:04:29 -080024 */
Junxiao Shic041ca32014-02-25 20:01:15 -070025
Davide Pesavento3dade002019-03-19 11:29:56 -060026#include "daemon/global.hpp"
Junxiao Shic041ca32014-02-25 20:01:15 -070027
28namespace nfd {
29
Davide Pesavento2bdf60c2019-02-19 18:23:45 -050030static thread_local unique_ptr<boost::asio::io_service> g_ioService;
Davide Pesavento3dade002019-03-19 11:29:56 -060031static thread_local unique_ptr<Scheduler> g_scheduler;
Teng Liangf59e58f2018-09-07 16:41:54 -070032static boost::asio::io_service* g_mainIoService = nullptr;
Teng Liang952d6fd2018-05-29 21:09:52 -070033static boost::asio::io_service* g_ribIoService = nullptr;
Junxiao Shic041ca32014-02-25 20:01:15 -070034
35boost::asio::io_service&
36getGlobalIoService()
37{
Davide Pesavento2bdf60c2019-02-19 18:23:45 -050038 if (g_ioService == nullptr) {
39 g_ioService = make_unique<boost::asio::io_service>();
Junxiao Shic041ca32014-02-25 20:01:15 -070040 }
41 return *g_ioService;
42}
43
Davide Pesavento3dade002019-03-19 11:29:56 -060044Scheduler&
45getScheduler()
46{
47 if (g_scheduler == nullptr) {
48 g_scheduler = make_unique<Scheduler>(getGlobalIoService());
49 }
50 return *g_scheduler;
51}
Davide Pesavento2bdf60c2019-02-19 18:23:45 -050052
Davide Pesavento3dade002019-03-19 11:29:56 -060053#ifdef WITH_TESTS
Junxiao Shic041ca32014-02-25 20:01:15 -070054void
55resetGlobalIoService()
56{
Davide Pesavento3dade002019-03-19 11:29:56 -060057 g_scheduler.reset();
Junxiao Shic041ca32014-02-25 20:01:15 -070058 g_ioService.reset();
59}
Davide Pesavento2bdf60c2019-02-19 18:23:45 -050060#endif
Junxiao Shic041ca32014-02-25 20:01:15 -070061
Teng Liang952d6fd2018-05-29 21:09:52 -070062boost::asio::io_service&
Teng Liangf59e58f2018-09-07 16:41:54 -070063getMainIoService()
64{
65 BOOST_ASSERT(g_mainIoService != nullptr);
66 return *g_mainIoService;
67}
68
69boost::asio::io_service&
Teng Liang952d6fd2018-05-29 21:09:52 -070070getRibIoService()
71{
72 BOOST_ASSERT(g_ribIoService != nullptr);
73 return *g_ribIoService;
74}
75
76void
Davide Pesavento3dade002019-03-19 11:29:56 -060077setMainIoService(boost::asio::io_service* mainIo)
78{
79 g_mainIoService = mainIo;
80}
81
82void
83setRibIoService(boost::asio::io_service* ribIo)
84{
85 g_ribIoService = ribIo;
86}
87
88void
Teng Liangf59e58f2018-09-07 16:41:54 -070089runOnMainIoService(const std::function<void()>& f)
90{
91 getMainIoService().post(f);
92}
93
94void
Teng Liang952d6fd2018-05-29 21:09:52 -070095runOnRibIoService(const std::function<void()>& f)
96{
97 getRibIoService().post(f);
98}
99
Junxiao Shic041ca32014-02-25 20:01:15 -0700100} // namespace nfd