From c9089c5d510f2d45c93c90c6d91b5f846e0719c0 Mon Sep 17 00:00:00 2001
From: heavyrain2012 <heavyrain.lee@gmail.com>
Date: Sun, 24 Nov 2019 17:27:55 +0800
Subject: [PATCH] =?UTF-8?q?=E6=80=A7=E8=83=BD=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 broker/migrate/h2/V18__add_friend_request_index.sql   |  1 +
 .../migrate/mysql/V23__add_friend_request_index.sql   |  1 +
 .../java/io/moquette/persistence/DatabaseStore.java   | 11 +++++++++--
 3 files changed, 11 insertions(+), 2 deletions(-)
 create mode 100755 broker/migrate/h2/V18__add_friend_request_index.sql
 create mode 100755 broker/migrate/mysql/V23__add_friend_request_index.sql

diff --git a/broker/migrate/h2/V18__add_friend_request_index.sql b/broker/migrate/h2/V18__add_friend_request_index.sql
new file mode 100755
index 00000000..a7109787
--- /dev/null
+++ b/broker/migrate/h2/V18__add_friend_request_index.sql
@@ -0,0 +1 @@
+alter table t_friend_request add index `friend_request_uid_index` (`_friend_uid`);
diff --git a/broker/migrate/mysql/V23__add_friend_request_index.sql b/broker/migrate/mysql/V23__add_friend_request_index.sql
new file mode 100755
index 00000000..a7109787
--- /dev/null
+++ b/broker/migrate/mysql/V23__add_friend_request_index.sql
@@ -0,0 +1 @@
+alter table t_friend_request add index `friend_request_uid_index` (`_friend_uid`);
diff --git a/broker/src/main/java/io/moquette/persistence/DatabaseStore.java b/broker/src/main/java/io/moquette/persistence/DatabaseStore.java
index 66693764..14fdaa68 100644
--- a/broker/src/main/java/io/moquette/persistence/DatabaseStore.java
+++ b/broker/src/main/java/io/moquette/persistence/DatabaseStore.java
@@ -2217,13 +2217,20 @@ public class DatabaseStore {
         ResultSet rs = null;
         try {
             connection = DBUtil.getConnection();
-            String sql = "select   `_uid`" +
+            String sql = "select  `_uid`" +
                 ", `_friend_uid`" +
                 ", `_reason`" +
                 ", `_status`" +
                 ", `_dt`" +
                 ", `_from_read_status`" +
-                ", `_to_read_status` from t_friend_request where `_uid` = ? or `_friend_uid` = ?";
+                ", `_to_read_status` from t_friend_request where `_uid` = ? UNION ALL " +
+                "select   `_uid`" +
+                ", `_friend_uid`" +
+                ", `_reason`" +
+                ", `_status`" +
+                ", `_dt`" +
+                ", `_from_read_status`" +
+                ", `_to_read_status` from t_friend_request where `_friend_uid` = ?";
             statement = connection.prepareStatement(sql);
 
             int index = 1;