From 6e4af2ed0363d4021330e9792a62f1659bb84136 Mon Sep 17 00:00:00 2001
From: M66B <M66B@users.noreply.github.com>
Date: Tue, 29 Oct 2019 11:34:03 +0100
Subject: [PATCH] Batch search fix

---
 .../email/BoundaryCallbackMessages.java       | 36 ++++++++++---------
 1 file changed, 20 insertions(+), 16 deletions(-)

diff --git a/app/src/main/java/eu/faircode/email/BoundaryCallbackMessages.java b/app/src/main/java/eu/faircode/email/BoundaryCallbackMessages.java
index 6852e1ec2d..155b0c3a5a 100644
--- a/app/src/main/java/eu/faircode/email/BoundaryCallbackMessages.java
+++ b/app/src/main/java/eu/faircode/email/BoundaryCallbackMessages.java
@@ -181,20 +181,26 @@ public class BoundaryCallbackMessages extends PagedList.BoundaryCallback<TupleMe
             db.beginTransaction();
 
             while (found < pageSize && !state.destroyed) {
-                state.matches = db.message().matchMessages(
-                        folder,
-                        "%" + find + "%",
-                        seen, flagged, snoozed,
-                        SEARCH_LIMIT, state.offset);
-                Log.i("Boundary device folder=" + folder +
-                        " query=" + query +
-                        " seen=" + seen +
-                        " flagged=" + flagged +
-                        " snoozed=" + snoozed +
-                        " offset=" + state.offset +
-                        " size=" + state.matches.size());
-                if (state.matches.size() == 0)
-                    break;
+                if (state.matches == null ||
+                        (state.matches.size() > 0 && state.index >= state.matches.size())) {
+                    state.matches = db.message().matchMessages(
+                            folder,
+                            "%" + find + "%",
+                            seen, flagged, snoozed,
+                            SEARCH_LIMIT, state.offset);
+                    Log.i("Boundary device folder=" + folder +
+                            " query=" + query +
+                            " seen=" + seen +
+                            " flagged=" + flagged +
+                            " snoozed=" + snoozed +
+                            " offset=" + state.offset +
+                            " size=" + state.matches.size());
+                    if (state.matches.size() == 0)
+                        break;
+
+                    state.offset += Math.min(state.matches.size(), SEARCH_LIMIT);
+                    state.index = 0;
+                }
 
                 for (int i = state.index; i < state.matches.size() && found < pageSize && !state.destroyed; i++) {
                     state.index = i + 1;
@@ -223,8 +229,6 @@ public class BoundaryCallbackMessages extends PagedList.BoundaryCallback<TupleMe
                         db.message().setMessageFound(match.account, match.thread);
                     }
                 }
-
-                state.offset += SEARCH_LIMIT;
             }
 
             db.setTransactionSuccessful();