Add cache for MessageHistory Database

Signed-off-by: Dev4Mod <[email protected]>

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/core/db/MessageHistory.java

@@ -5,6 +5,7 @@
 import android.database.Cursor;
 import android.database.sqlite.SQLiteDatabase;
 import android.database.sqlite.SQLiteOpenHelper;
+import android.util.LruCache;
 
 import androidx.annotation.Nullable;
 
@@ -25,13 +26,23 @@ public class MessageHistory extends SQLiteOpenHelper {
     private static MessageHistory mInstance;
     private SQLiteDatabase dbWrite;
 
+    private static final int MESSAGE_CACHE_SIZE = 100;
+    private static final int SEEN_MESSAGE_CACHE_SIZE = 200;
+    private static final int SEEN_MESSAGES_LIST_CACHE_SIZE = 50;
+    private final LruCache<Long, ArrayList<MessageItem>> messagesCache;
+    private final LruCache<String, MessageSeenItem> seenMessageCache;
+    private final LruCache<String, List<MessageSeenItem>> seenMessagesListCache;
+
     public enum MessageType {
         MESSAGE_TYPE,
         VIEW_ONCE_TYPE
     }
 
     public MessageHistory(Context context) {
         super(context, "MessageHistory.db", null, 2);
+        messagesCache = new LruCache<>(MESSAGE_CACHE_SIZE);
+        seenMessageCache = new LruCache<>(SEEN_MESSAGE_CACHE_SIZE);
+        seenMessagesListCache = new LruCache<>(SEEN_MESSAGES_LIST_CACHE_SIZE);
     }
 
     public static MessageHistory getInstance() {
@@ -51,10 +62,20 @@ public final void insertMessage(long id, String message, long timestamp) {
             contentValues0.put("text_data", message);
             contentValues0.put("editTimestamp", timestamp);
             dbWrite.insert("MessageHistory", null, contentValues0);
+
+            // Invalidate cache for this message ID
+            messagesCache.remove(id);
         }
     }
 
     public ArrayList<MessageItem> getMessages(long v) {
+        // Check cache first
+        ArrayList<MessageItem> cachedMessages = messagesCache.get(v);
+        if (cachedMessages != null) {
+            return cachedMessages;
+        }
+
+        // If not in cache, query database
         Cursor history = dbWrite.query("MessageHistory", new String[]{"_id", "row_id", "text_data", "editTimestamp"}, "row_id=?", new String[]{String.valueOf(v)}, null, null, null);
         if (!history.moveToFirst()) {
             history.close();
@@ -69,6 +90,9 @@ public ArrayList<MessageItem> getMessages(long v) {
         }
         while (history.moveToNext());
         history.close();
+
+        // Store in cache
+        messagesCache.put(v, messages);
         return messages;
     }
 
@@ -82,6 +106,11 @@ public final void insertHideSeenMessage(String jid, String message_id, MessageTy
             content.put("message_id", message_id);
             content.put("type", type.ordinal());
             dbWrite.insert("hide_seen_messages", null, content);
+
+            // Invalidate caches
+            String cacheKey = createSeenMessageCacheKey(jid, message_id, type);
+            seenMessageCache.remove(cacheKey);
+            invalidateSeenMessagesListCache(jid, type);
         }
     }
 
@@ -95,12 +124,28 @@ public boolean updateViewedMessage(String jid, String message_id, MessageType ty
             ContentValues content = new ContentValues();
             content.put("viewed", viewed ? 1 : 0);
             dbWrite.update("hide_seen_messages", content, "_id=?", new String[]{cursor.getString(cursor.getColumnIndexOrThrow("_id"))});
+
+            // Update cache or invalidate
+            String cacheKey = createSeenMessageCacheKey(jid, message_id, type);
+            MessageSeenItem cachedItem = seenMessageCache.get(cacheKey);
+            if (cachedItem != null && cachedItem.viewed != viewed) {
+                seenMessageCache.remove(cacheKey);
+            }
+            invalidateSeenMessagesListCache(jid, type);
         }
         cursor.close();
         return true;
     }
 
     public MessageSeenItem getHideSeenMessage(String jid, String message_id, MessageType type) {
+        // Check cache first
+        String cacheKey = createSeenMessageCacheKey(jid, message_id, type);
+        MessageSeenItem cachedItem = seenMessageCache.get(cacheKey);
+        if (cachedItem != null) {
+            return cachedItem;
+        }
+
+        // If not in cache, query database
         Cursor cursor = dbWrite.query("hide_seen_messages", new String[]{"viewed"}, "jid=? AND message_id=? AND type=?", new String[]{jid, message_id, String.valueOf(type.ordinal())}, null, null, null);
         if (!cursor.moveToFirst()) {
             cursor.close();
@@ -109,10 +154,21 @@ public MessageSeenItem getHideSeenMessage(String jid, String message_id, Message
         var viewed = cursor.getInt(cursor.getColumnIndexOrThrow("viewed")) == 1;
         var message = new MessageSeenItem(jid, message_id, viewed);
         cursor.close();
+
+        // Store in cache
+        seenMessageCache.put(cacheKey, message);
         return message;
     }
 
     public List<MessageSeenItem> getHideSeenMessages(String jid, MessageType type, boolean viewed) {
+        // Check cache first
+        String cacheKey = createSeenMessagesListCacheKey(jid, type, viewed);
+        List<MessageSeenItem> cachedList = seenMessagesListCache.get(cacheKey);
+        if (cachedList != null) {
+            return cachedList;
+        }
+
+        // If not in cache, query database
         Cursor cursor = dbWrite.query("hide_seen_messages", new String[]{"jid", "message_id", "viewed"}, "jid=? AND type=? AND viewed=?", new String[]{jid, String.valueOf(type.ordinal()), viewed ? "1" : "0"}, null, null, null);
         if (!cursor.moveToFirst()) {
             cursor.close();
@@ -123,12 +179,18 @@ public List<MessageSeenItem> getHideSeenMessages(String jid, MessageType type, b
             var message_id = cursor.getString(cursor.getColumnIndexOrThrow("message_id"));
             var message = new MessageSeenItem(jid, message_id, viewed);
             messages.add(message);
+
+            // Also cache individual messages
+            String msgCacheKey = createSeenMessageCacheKey(jid, message_id, type);
+            seenMessageCache.put(msgCacheKey, message);
         } while (cursor.moveToNext());
         cursor.close();
+
+        // Store in cache
+        seenMessagesListCache.put(cacheKey, messages);
         return messages;
     }
 
-
     @Override
     public void onCreate(SQLiteDatabase sqLiteDatabase) {
         sqLiteDatabase.execSQL("create table MessageHistory(_id INTEGER PRIMARY KEY AUTOINCREMENT, row_id INTEGER NOT NULL, text_data TEXT NOT NULL, editTimestamp BIGINT DEFAULT 0 );");
@@ -142,6 +204,25 @@ public void onUpgrade(SQLiteDatabase sqLiteDatabase, int oldVersion, int newVers
         }
     }
 
+    private String createSeenMessageCacheKey(String jid, String message_id, MessageType type) {
+        return jid + "_" + message_id + "_" + type.ordinal();
+    }
+
+    private String createSeenMessagesListCacheKey(String jid, MessageType type, boolean viewed) {
+        return jid + "_" + type.ordinal() + "_" + (viewed ? "1" : "0");
+    }
+
+    private void invalidateSeenMessagesListCache(String jid, MessageType type) {
+        seenMessagesListCache.remove(createSeenMessagesListCacheKey(jid, type, true));
+        seenMessagesListCache.remove(createSeenMessagesListCacheKey(jid, type, false));
+    }
+
+    public void clearCaches() {
+        messagesCache.evictAll();
+        seenMessageCache.evictAll();
+        seenMessagesListCache.evictAll();
+    }
+
     @AllArgsConstructor
     public static class MessageItem {
         public long id;
@@ -170,9 +251,5 @@ public FMessageWpp getFMessage() {
             }
             return fMessageWpp;
         }
-
-
     }
-
-
 }