|  | Home · All Namespaces · All Classes · Main Classes · Grouped Classes · Modules · Functions |

# ratecontroller.cpp Example File
network/torrent/ratecontroller.cpp

     /****************************************************************************
     **
     ** Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies).
     ** Contact: Qt Software Information (qt-info@nokia.com)
     **
     ** This file is part of the example classes of the Qt Toolkit.
     **
     ** Commercial Usage
     ** Licensees holding valid Qt Commercial licenses may use this file in
     ** accordance with the Qt Commercial License Agreement provided with the
     ** Software or, alternatively, in accordance with the terms contained in
     ** a written agreement between you and Nokia.
     **
     **
     ** GNU General Public License Usage
     ** Alternatively, this file may be used under the terms of the GNU
     ** General Public License versions 2.0 or 3.0 as published by the Free
     ** Software Foundation and appearing in the file LICENSE.GPL included in
     ** the packaging of this file.  Please review the following information
     ** to ensure GNU General Public Licensing requirements will be met:
     ** http://www.fsf.org/licensing/licenses/info/GPLv2.html and
     ** http://www.gnu.org/copyleft/gpl.html.  In addition, as a special
     ** exception, Nokia gives you certain additional rights. These rights
     ** are described in the Nokia Qt GPL Exception version 1.3, included in
     ** the file GPL_EXCEPTION.txt in this package.
     **
     ** Qt for Windows(R) Licensees
     ** As a special exception, Nokia, as the sole copyright holder for Qt
     ** Designer, grants users of the Qt/Eclipse Integration plug-in the
     ** right for the Qt/Eclipse Integration to link to functionality
     ** provided by Qt Designer and its related libraries.
     **
     ** If you are unsure which license is appropriate for your use, please
     ** contact the sales department at qt-sales@nokia.com.
     **
     ****************************************************************************/

     #include "peerwireclient.h"
     #include "ratecontroller.h"

     #include <QtCore>

     Q_GLOBAL_STATIC(RateController, rateController)

     RateController *RateController::instance()
     {
         return rateController();
     }

     void RateController::addSocket(PeerWireClient *socket)
     {
         connect(socket, SIGNAL(readyToTransfer()), this, SLOT(scheduleTransfer()));
         socket->setReadBufferSize(downLimit * 4);
         sockets << socket;
         scheduleTransfer();
     }

     void RateController::removeSocket(PeerWireClient *socket)
     {
         disconnect(socket, SIGNAL(readyToTransfer()), this, SLOT(scheduleTransfer()));
         socket->setReadBufferSize(0);
         sockets.remove(socket);
     }

     void RateController::setDownloadLimit(int bytesPerSecond)
     {
         downLimit = bytesPerSecond;
         foreach (PeerWireClient *socket, sockets)
             socket->setReadBufferSize(downLimit * 4);
     }

     void RateController::scheduleTransfer()
     {
         if (transferScheduled)
             return;
         transferScheduled = true;
         QTimer::singleShot(50, this, SLOT(transfer()));
     }

     void RateController::transfer()
     {
         transferScheduled = false;
         if (sockets.isEmpty())
             return;

         int msecs = 1000;
         if (!stopWatch.isNull())
             msecs = qMin(msecs, stopWatch.elapsed());

         qint64 bytesToWrite = (upLimit * msecs) / 1000;
         qint64 bytesToRead = (downLimit * msecs) / 1000;
         if (bytesToWrite == 0 && bytesToRead == 0) {
             scheduleTransfer();
             return;
         }

         QSet<PeerWireClient *> pendingSockets;
         foreach (PeerWireClient *client, sockets) {
             if (client->canTransferMore())
                 pendingSockets << client;
         }
         if (pendingSockets.isEmpty())
             return;

         stopWatch.start();

         bool canTransferMore;
         do {
             canTransferMore = false;
             qint64 writeChunk = qMax<qint64>(1, bytesToWrite / pendingSockets.size());
             qint64 readChunk = qMax<qint64>(1, bytesToRead / pendingSockets.size());

             QSetIterator<PeerWireClient *> it(pendingSockets);
             while (it.hasNext() && (bytesToWrite > 0 || bytesToRead > 0)) {
                 PeerWireClient *socket = it.next();
                 if (socket->state() != QAbstractSocket::ConnectedState) {
                     pendingSockets.remove(socket);
                     continue;
                 }

                 bool dataTransferred = false;
                 qint64 available = qMin<qint64>(socket->socketBytesAvailable(), readChunk);
                 if (available > 0) {
                     qint64 readBytes = socket->readFromSocket(qMin<qint64>(available, bytesToRead));
                     if (readBytes > 0) {
                         bytesToRead -= readBytes;
                         dataTransferred = true;
                     }
                 }

                 if (upLimit * 2 > socket->bytesToWrite()) {
                     qint64 chunkSize = qMin<qint64>(writeChunk, bytesToWrite);
                     qint64 toWrite = qMin(upLimit * 2 - socket->bytesToWrite(), chunkSize);
                     if (toWrite > 0) {
                         qint64 writtenBytes = socket->writeToSocket(toWrite);
                         if (writtenBytes > 0) {
                             bytesToWrite -= writtenBytes;
                             dataTransferred = true;
                         }
                     }
                 }

                 if (dataTransferred && socket->canTransferMore())
                     canTransferMore = true;
                 else
                     pendingSockets.remove(socket);
             }
         } while (canTransferMore && (bytesToWrite > 0 || bytesToRead > 0) && !pendingSockets.isEmpty());

         if (canTransferMore || bytesToWrite == 0 || bytesToRead == 0)
             scheduleTransfer();
     }