Display and update message processing progress

[quassel.git] / src / qtui / qtuimessageprocessor.cpp

diff --git a/src/qtui/qtuimessageprocessor.cpp b/src/qtui/qtuimessageprocessor.cpp
index 203397d..a9209ae 100644 (file)
--- a/src/qtui/qtuimessageprocessor.cpp
+++ b/src/qtui/qtuimessageprocessor.cpp
@@ -26,13 +26,25 @@
 #include "messagemodel.h"
 #include "network.h"
 
+const int progressUpdateDelay = 100;  // ms between progress signal updates
+
 QtUiMessageProcessor::QtUiMessageProcessor(QObject *parent) : AbstractMessageProcessor(parent) {
+  _msgsProcessed = _msgCount = 0;
   _processing = false;
   _processMode = TimerBased;
   _processTimer.setInterval(0);
   connect(&_processTimer, SIGNAL(timeout()), this, SLOT(processNextMessage()));
 }
 
+void QtUiMessageProcessor::reset() {
+  if(processMode() == TimerBased) {
+    if(_processTimer.isActive()) _processTimer.stop();
+    _processing = false;
+    _currentBatch.clear();
+    _processQueue.clear();
+  }
+}
+
 void QtUiMessageProcessor::process(Message &msg) {
   checkForHighlight(msg);
   Client::messageModel()->insertMessage(msg);
@@ -41,13 +53,19 @@ void QtUiMessageProcessor::process(Message &msg) {
 
 void QtUiMessageProcessor::process(QList<Message> &msgs) {
   _processQueue.append(msgs);
+  _msgCount += msgs.count();
   if(!isProcessing()) startProcessing();
+  else updateProgress();
 }
 
 void QtUiMessageProcessor::startProcessing() {
   if(processMode() == TimerBased) {
     if(_currentBatch.isEmpty() && _processQueue.isEmpty()) return;
     _processing = true;
+    _msgsProcessed = 0;
+    _msgCount = _currentBatch.count();
+    foreach(QList<Message> msglist, _processQueue) _msgCount += msglist.count();
+    updateProgress();
     if(!_processTimer.isActive()) _processTimer.start();
   }
 }
@@ -57,12 +75,28 @@ void QtUiMessageProcessor::processNextMessage() {
     if(_processQueue.isEmpty()) {
       _processTimer.stop();
       _processing = false;
+      _msgsProcessed = _msgCount = 0;
+      updateProgress();
       return;
     }
     _currentBatch = _processQueue.takeFirst();
   }
   Message msg = _currentBatch.takeFirst();
   process(msg);
+  _msgsProcessed++;
+  updateProgress();
+}
+
+void QtUiMessageProcessor::updateProgress(bool start) {
+  if(start) {
+    _progressTimer.start();
+    emit progressUpdated(_msgsProcessed, _msgCount);
+  } else {
+    if(_msgCount == 0 || _progressTimer.elapsed() >= progressUpdateDelay) {
+      _progressTimer.restart();
+      emit progressUpdated(_msgsProcessed, _msgCount);
+    }
+  }
 }
 
 // TODO optimize checkForHighlight