1 /****************************************************************************
2 **
3 ** Copyright (C) 2015 The Qt Company Ltd.
4 ** Contact: http://www.qt.io/licensing/
5 **
6 ** This file is part of the examples of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:BSD$
9 ** You may use this file under the terms of the BSD license as follows:
10 **
11 ** "Redistribution and use in source and binary forms, with or without
12 ** modification, are permitted provided that the following conditions are
13 ** met:
14 **   * Redistributions of source code must retain the above copyright
15 **     notice, this list of conditions and the following disclaimer.
16 **   * Redistributions in binary form must reproduce the above copyright
17 **     notice, this list of conditions and the following disclaimer in
18 **     the documentation and/or other materials provided with the
19 **     distribution.
20 **   * Neither the name of The Qt Company Ltd nor the names of its
21 **     contributors may be used to endorse or promote products derived
22 **     from this software without specific prior written permission.
23 **
24 **
25 ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26 ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27 ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
28 ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
29 ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31 ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
35 ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
36 **
37 ** $QT_END_LICENSE$
38 **
39 ****************************************************************************/
40 
41 #ifndef RATECONTROLLER_H
42 #define RATECONTROLLER_H
43 
44 #include <QObject>
45 #include <QSet>
46 #include <QTime>
47 
48 class PeerWireClient;
49 
50 class RateController : public QObject
51 {
52     Q_OBJECT
53 
54 public:
55     inline RateController(QObject *parent = 0)
QObject(parent)56         : QObject(parent), transferScheduled(false) { }
57     static RateController *instance();
58 
59     void addSocket(PeerWireClient *socket);
60     void removeSocket(PeerWireClient *socket);
61 
uploadLimit()62     inline int uploadLimit() const { return upLimit; }
downloadLimit()63     inline int downloadLimit() const { return downLimit; }
setUploadLimit(int bytesPerSecond)64     inline void setUploadLimit(int bytesPerSecond) { upLimit = bytesPerSecond; }
65     void setDownloadLimit(int bytesPerSecond);
66 
67 public slots:
68     void transfer();
69     void scheduleTransfer();
70 
71 private:
72     QTime stopWatch;
73     QSet<PeerWireClient *> sockets;
74     int upLimit;
75     int downLimit;
76     bool transferScheduled;
77 };
78 
79 #endif
80