X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fclient%2Fbacklogrequester.h;h=c8a51e6afc4fa350509e9469da60c55cdd5ebe4a;hp=af7696afc98c9fb4d059aafb0f144125ed982a6e;hb=52209badc8e769e50aa3019b63689dda0e79e9d0;hpb=fb011e52bc6613198a65e037f0752f6361339aeb

diff --git a/src/client/backlogrequester.h b/src/client/backlogrequester.h
index af7696af..c8a51e6a 100644
--- a/src/client/backlogrequester.h
+++ b/src/client/backlogrequester.h
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-2019 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
@@ -15,11 +15,12 @@
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program; if not, write to the                         *
  *   Free Software Foundation, Inc.,                                       *
- *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+ *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  ***************************************************************************/
 
-#ifndef BACKLOGREQUESTER_H
-#define BACKLOGREQUESTER_H
+#pragma once
+
+#include <set>
 
 #include <QList>
 
@@ -30,51 +31,86 @@
 
 class ClientBacklogManager;
 
-class BacklogRequester {
+class BacklogRequester
+{
 public:
-  enum RequesterTypes {
-    InvalidRequester = 0,
-    GlobalUnread,
-    PerBufferUnread,
-    PerBufferFixed
-  };
+    enum RequesterType
+    {
+        InvalidRequester = 0,
+        PerBufferFixed,
+        PerBufferUnread,
+        GlobalUnread
+    };
+
+    BacklogRequester(bool buffering, RequesterType requesterType, ClientBacklogManager* backlogManger);
+    virtual ~BacklogRequester() = default;
+
+    inline bool isBuffering() { return _isBuffering; }
+    inline RequesterType type() { return _requesterType; }
+    inline const QList<Message>& bufferedMessages() { return _bufferedMessages; }
+
+    inline int buffersWaiting() const { return int(_buffersWaiting.size()); }
+    inline int totalBuffers() const { return _totalBuffers; }
 
-  BacklogRequester(bool buffering, ClientBacklogManager *backlogManger);
-  virtual inline ~BacklogRequester() {}
+    bool buffer(BufferId bufferId, const MessageList& messages);  //! returns false if it was the last missing backlogpart
 
-  inline bool isBuffering() { return _isBuffering; }
-  inline const QList<Message> &bufferedMessages() { return _bufferedMessages; }
+    virtual void requestBacklog(const BufferIdList& bufferIds) = 0;
+    virtual inline void requestInitialBacklog() { requestBacklog(allBufferIds()); }
 
-  //! returns false if it was the last missing backlogpart
-  bool buffer(BufferId bufferId, const MessageList &messages);
-  
-  virtual void requestBacklog() = 0;
+    virtual void flushBuffer();
 
 protected:
-  inline QList<BufferId> allBufferIds() const { return Client::networkModel()->allBufferIds(); }
-  inline void setWaitingBuffers(const QList<BufferId> &buffers) { _buffersWaiting = buffers.toSet(); }
-  inline void setWaitingBuffers(const QSet<BufferId> &buffers) { _buffersWaiting = buffers; }
-  inline void addWaitingBuffer(BufferId buffer) { _buffersWaiting << buffer; }
+    BufferIdList allBufferIds() const;
+    void setWaitingBuffers(const BufferIdList& buffers);
 
-  ClientBacklogManager *backlogManager;
+    ClientBacklogManager* backlogManager;
 
 private:
-  bool _isBuffering;
-  MessageList _bufferedMessages;
-  QSet<BufferId> _buffersWaiting;
+    bool _isBuffering;
+    RequesterType _requesterType;
+    MessageList _bufferedMessages;
+    int _totalBuffers;
+    std::set<BufferId> _buffersWaiting;
 };
 
 // ========================================
 //  FIXED BACKLOG REQUESTER
 // ========================================
-class FixedBacklogRequester : public BacklogRequester {
+class FixedBacklogRequester : public BacklogRequester
+{
+public:
+    FixedBacklogRequester(ClientBacklogManager* backlogManager);
+    void requestBacklog(const BufferIdList& bufferIds) override;
+
+private:
+    int _backlogCount;
+};
+
+// ========================================
+//  GLOBAL UNREAD BACKLOG REQUESTER
+// ========================================
+class GlobalUnreadBacklogRequester : public BacklogRequester
+{
 public:
-  FixedBacklogRequester(ClientBacklogManager *backlogManager);
-  virtual void requestBacklog();
+    GlobalUnreadBacklogRequester(ClientBacklogManager* backlogManager);
+    void requestInitialBacklog() override;
+    void requestBacklog(const BufferIdList&) override {}
 
 private:
-  int _backlogCount;
+    int _limit;
+    int _additional;
 };
 
+// ========================================
+//  PER BUFFER UNREAD BACKLOG REQUESTER
+// ========================================
+class PerBufferUnreadBacklogRequester : public BacklogRequester
+{
+public:
+    PerBufferUnreadBacklogRequester(ClientBacklogManager* backlogManager);
+    void requestBacklog(const BufferIdList& bufferIds) override;
 
-#endif //BACKLOGREQUESTER_H
+private:
+    int _limit;
+    int _additional;
+};