1 /* <!-- copyright */
2 /*
3 * aria2 - The high speed download utility
4 *
5 * Copyright (C) 2006 Tatsuhiro Tsujikawa
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 *
21 * In addition, as a special exception, the copyright holders give
22 * permission to link the code of portions of this program with the
23 * OpenSSL library under certain conditions as described in each
24 * individual source file, and distribute linked combinations
25 * including the two.
26 * You must obey the GNU General Public License in all respects
27 * for all of the code used other than OpenSSL. If you modify
28 * file(s) with this exception, you may extend this exception to your
29 * version of the file(s), but you are not obligated to do so. If you
30 * do not wish to do so, delete this exception statement from your
31 * version. If you delete this exception statement from all source
32 * files in the program, then also delete it here.
33 */
34 /* copyright --> */
35 #include "DHTBucketRefreshCommand.h"
36 #include "DHTRoutingTable.h"
37 #include "DHTTaskQueue.h"
38 #include "DHTTaskFactory.h"
39 #include "DHTTask.h"
40 #include "DownloadEngine.h"
41 #include "RequestGroupMan.h"
42
43 namespace aria2 {
44
DHTBucketRefreshCommand(cuid_t cuid,DownloadEngine * e,std::chrono::seconds interval)45 DHTBucketRefreshCommand::DHTBucketRefreshCommand(cuid_t cuid, DownloadEngine* e,
46 std::chrono::seconds interval)
47 : TimeBasedCommand{cuid, e, std::move(interval)},
48 routingTable_{nullptr},
49 taskQueue_{nullptr},
50 taskFactory_{nullptr}
51 {
52 }
53
54 DHTBucketRefreshCommand::~DHTBucketRefreshCommand() = default;
55
preProcess()56 void DHTBucketRefreshCommand::preProcess()
57 {
58 if (getDownloadEngine()->getRequestGroupMan()->downloadFinished() ||
59 getDownloadEngine()->isHaltRequested()) {
60 enableExit();
61 }
62 }
63
process()64 void DHTBucketRefreshCommand::process()
65 {
66 taskQueue_->addPeriodicTask1(taskFactory_->createBucketRefreshTask());
67 }
68
setRoutingTable(DHTRoutingTable * routingTable)69 void DHTBucketRefreshCommand::setRoutingTable(DHTRoutingTable* routingTable)
70 {
71 routingTable_ = routingTable;
72 }
73
setTaskQueue(DHTTaskQueue * taskQueue)74 void DHTBucketRefreshCommand::setTaskQueue(DHTTaskQueue* taskQueue)
75 {
76 taskQueue_ = taskQueue;
77 }
78
setTaskFactory(DHTTaskFactory * taskFactory)79 void DHTBucketRefreshCommand::setTaskFactory(DHTTaskFactory* taskFactory)
80 {
81 taskFactory_ = taskFactory;
82 }
83
84 } // namespace aria2
85