Mercurial > hg > orthanc-transfers

comparison Framework/HttpQueries/HttpQueriesQueue.h @ 0:95226b754d9e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial release
author Sebastien Jodogne <s.jodogne@gmail.com>
date Mon, 17 Sep 2018 11:34:55 +0200
parents
children 9bcd6eadcff5
comparison
equal deleted inserted replaced
-1:000000000000 0:95226b754d9e
1 /**
2 * Transfers accelerator plugin for Orthanc
3 * Copyright (C) 2018 Osimis, Belgium
4 *
5 * This program is free software: you can redistribute it and/or
6 * modify it under the terms of the GNU Affero General Public License
7 * as published by the Free Software Foundation, either version 3 of
8 * the License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Affero General Public License for more details.
14 *
15 * You should have received a copy of the GNU Affero General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 **/
18
19
20 #pragma once
21
22 #include "IHttpQuery.h"
23
24 #include <Plugins/Samples/Common/OrthancPluginCppWrapper.h>
25
26 #include <boost/thread/mutex.hpp>
27 #include <boost/thread/condition_variable.hpp>
28
29
30 namespace OrthancPlugins
31 {
32 class HttpQueriesQueue : public boost::noncopyable
33 {
34 public:
35 enum Status
36 {
37 Status_Running,
38 Status_Success,
39 Status_Failure
40 };
41
42 private:
43 OrthancPluginContext *context_;
44 OrthancPeers peers_;
45 boost::mutex mutex_;
46 boost::condition_variable completed_;
47 std::vector<IHttpQuery*> queries_;
48 unsigned int maxRetries_;
49
50 size_t position_;
51 uint64_t downloadedSize_; // GET answers + POST answers
52 uint64_t uploadedSize_; // PUT body + POST body
53 size_t successQueries_;
54 bool isFailure_;
55
56
57 Status GetStatusInternal() const;
58
59 public:
60 HttpQueriesQueue(OrthancPluginContext* context);
61
62 ~HttpQueriesQueue();
63
64 OrthancPeers& GetOrthancPeers()
65 {
66 return peers_;
67 }
68
69 unsigned int GetMaxRetries();
70
71 void SetMaxRetries(unsigned int maxRetries);
72
73 void Reserve(size_t size);
74
75 void Reset();
76
77 void Enqueue(IHttpQuery* query); // Takes ownership
78
79 bool ExecuteOneQuery(uint64_t& networkTraffic);
80
81 Status WaitComplete(unsigned int timeoutMS);
82
83 void WaitComplete();
84
85 void GetStatistics(size_t& scheduledQueriesCount,
86 size_t& successQueriesCount,
87 uint64_t& downloadedSize,
88 uint64_t& uploadedSize);
89 };
90 }