src/client/clientbacklogmanager.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-09 by the Quassel Project                          *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "clientbacklogmanager.h"
  22
  23 #include "abstractmessageprocessor.h"
  24 #include "backlogsettings.h"
  25 #include "backlogrequester.h"
  26 #include "client.h"
  27
  28 #include <ctime>
  29
  30 #include <QDebug>
  31
  32 ClientBacklogManager::ClientBacklogManager(QObject *parent)
  33   : BacklogManager(parent),
  34     _requester(0)
  35 {
  36 }
  37
  38 void ClientBacklogManager::receiveBacklog(BufferId bufferId, MsgId first, MsgId last, int limit, int additional, QVariantList msgs) {
  39   Q_UNUSED(first) Q_UNUSED(last) Q_UNUSED(limit) Q_UNUSED(additional)
  40
  41   emit messagesReceived(bufferId, msgs.count());
  42
  43   MessageList msglist;
  44   foreach(QVariant v, msgs) {
  45     Message msg = v.value<Message>();
  46     msg.setFlags(msg.flags() | Message::Backlog);
  47     msglist << msg;
  48   }
  49
  50   if(isBuffering()) {
  51     bool lastPart = !_requester->buffer(bufferId, msglist);
  52     updateProgress(_requester->totalBuffers() - _requester->buffersWaiting(), _requester->totalBuffers());
  53     if(lastPart) {
  54       stopBuffering();
  55     }
  56   } else {
  57     dispatchMessages(msglist);
  58   }
  59 }
  60
  61 void ClientBacklogManager::receiveBacklogAll(MsgId first, MsgId last, int limit, int additional, QVariantList msgs) {
  62   Q_UNUSED(first) Q_UNUSED(last) Q_UNUSED(limit) Q_UNUSED(additional)
  63
  64   MessageList msglist;
  65   foreach(QVariant v, msgs) {
  66     Message msg = v.value<Message>();
  67     msg.setFlags(msg.flags() | Message::Backlog);
  68     msglist << msg;
  69   }
  70
  71   dispatchMessages(msglist);
  72 }
  73
  74 void ClientBacklogManager::requestInitialBacklog() {
  75   if(_requester) {
  76     qWarning() << "ClientBacklogManager::requestInitialBacklog() called twice in the same session! (Backlog has already been requested)";
  77     return;
  78   }
  79
  80   BacklogSettings settings;
  81   switch(settings.requesterType()) {
  82   case BacklogRequester::GlobalUnread:
  83     _requester = new GlobalUnreadBacklogRequester(this);
  84     break;
  85   case BacklogRequester::PerBufferUnread:
  86     _requester = new PerBufferUnreadBacklogRequester(this);
  87     break;
  88   case BacklogRequester::PerBufferFixed:
  89   default:
  90     _requester = new FixedBacklogRequester(this);
  91   };
  92
  93   _requester->requestBacklog();
  94   if(_requester->isBuffering()) {
  95     updateProgress(0, _requester->totalBuffers());
  96   }
  97 }
  98
  99 BufferIdList ClientBacklogManager::filterNewBufferIds(const BufferIdList &bufferIds) {
 100   BufferIdList newBuffers;
 101   foreach(BufferId bufferId, bufferIds) {
 102     if(!_buffersRequested.contains(bufferId))
 103       newBuffers << bufferId;
 104   }
 105   _buffersRequested += bufferIds.toSet();
 106   return newBuffers;
 107 }
 108
 109 void ClientBacklogManager::checkForBacklog(const QList<BufferId> &bufferIds) {
 110   if(!_requester) {
 111     // during client start up this message is to be expected in some situations.
 112     qDebug() << "ClientBacklogManager::checkForBacklog(): no active backlog requester (yet?).";
 113     return;
 114   }
 115   switch(_requester->type()) {
 116   case BacklogRequester::GlobalUnread:
 117     break;
 118   case BacklogRequester::PerBufferUnread:
 119   case BacklogRequester::PerBufferFixed:
 120   default:
 121     {
 122       BufferIdList buffers = filterNewBufferIds(bufferIds);
 123       if(!buffers.isEmpty())
 124         _requester->requestBacklog(buffers);
 125     }
 126   };
 127 }
 128
 129 void ClientBacklogManager::stopBuffering() {
 130   Q_ASSERT(_requester);
 131   dispatchMessages(_requester->bufferedMessages(), true);
 132 }
 133
 134 bool ClientBacklogManager::isBuffering() {
 135   return _requester && _requester->isBuffering();
 136 }
 137
 138 void ClientBacklogManager::dispatchMessages(const MessageList &messages, bool sort) {
 139   if(messages.isEmpty())
 140     return;
 141
 142   MessageList msgs = messages;
 143
 144   clock_t start_t = clock();
 145   if(sort)
 146     qSort(msgs);
 147   Client::messageProcessor()->process(msgs);
 148   clock_t end_t = clock();
 149
 150   emit messagesProcessed(tr("Processed %1 messages in %2 seconds.").arg(messages.count()).arg((float)(end_t - start_t) / CLOCKS_PER_SEC));
 151 }
 152
 153 void ClientBacklogManager::reset() {
 154   delete _requester;
 155   _requester = 0;
 156   _buffersRequested.clear();
 157 }