Speed up session lookups

Signed-off-by: applenick <applenick@users.noreply.github.com>

Author: applenick

core/src/main/java/dev/pgm/community/sessions/services/SQLSessionService.java

@@ -5,6 +5,7 @@
 import com.google.common.cache.CacheLoader;
 import com.google.common.cache.LoadingCache;
 import dev.pgm.community.Community;
+import dev.pgm.community.database.Query;
 import dev.pgm.community.feature.SQLFeatureBase;
 import dev.pgm.community.sessions.Session;
 import dev.pgm.community.sessions.SessionQuery;
@@ -31,6 +32,12 @@ public SessionData load(@NotNull SessionQuery key) {
         });
   }
 
+  @Override
+  public void createTable() {
+    super.createTable();
+    DB.executeUpdateAsync(Query.createTable(LATEST_TABLE_NAME, LATEST_TABLE_FIELDS));
+  }
+
   @Override
   public void save(Session session) {
     SessionData query = sessionCache.getUnchecked(new SessionQuery(session.getPlayerId(), false));
@@ -47,20 +54,50 @@ public void save(Session session) {
         session.getServerName(),
         session.getStartDate().toEpochMilli(),
         session.getEndDate() == null ? null : session.getEndDate().toEpochMilli());
+
+    DB.executeUpdateAsync(
+        UPSERT_LATEST_SESSION_QUERY,
+        session.getPlayerId().toString(),
+        false,
+        session.getSessionId().toString(),
+        session.isDisguised(),
+        session.getServerName(),
+        session.getStartDate().toEpochMilli(),
+        session.getEndDate() == null ? null : session.getEndDate().toEpochMilli());
+
+    if (!session.isDisguised()) {
+      DB.executeUpdateAsync(
+          UPSERT_LATEST_SESSION_QUERY,
+          session.getPlayerId().toString(),
+          true,
+          session.getSessionId().toString(),
+          session.isDisguised(),
+          session.getServerName(),
+          session.getStartDate().toEpochMilli(),
+          session.getEndDate() == null ? null : session.getEndDate().toEpochMilli());
+    }
   }
 
   public void updateSessionEndTime(Session session) {
     DB.executeUpdateAsync(
         UPDATE_SESSION_ENDTIME_QUERY,
         session.getEndDate() == null ? null : session.getEndDate().toEpochMilli(),
         session.getSessionId().toString());
+    DB.executeUpdateAsync(
+        UPDATE_LATEST_ENDTIME_QUERY,
+        session.getEndDate() == null ? null : session.getEndDate().toEpochMilli(),
+        session.getSessionId().toString());
   }
 
   public void endOngoingSessions() {
     DB.executeUpdateAsync(
         UPDATE_ONGOING_SESSION_ENDING_QUERY,
         Instant.now().toEpochMilli(),
         Community.get().getServerId());
+    DB.executeUpdateAsync(
+        UPDATE_LATEST_ONGOING_SESSION_ENDING_QUERY,
+        Instant.now().toEpochMilli(),
+        Community.get().getServerId());
   }
 
   @Override
@@ -76,11 +113,12 @@ public CompletableFuture<Session> query(SessionQuery target) {
       return CompletableFuture.completedFuture(data.getSession());
     } else {
       return DB.getFirstRowAsync(
-              target.ignoreDisguised() ? SELECT_DISGUISED_SESSION_QUERY : SELECT_SESSION_QUERY,
-              target.getPlayerId().toString())
+              SELECT_LATEST_SESSION_QUERY,
+              target.getPlayerId().toString(),
+              target.ignoreDisguised())
           .thenApplyAsync(result -> {
             if (result != null) {
-              String id = result.getString("id");
+              String id = result.getString("session_id");
 
               String player = result.getString("player");
               boolean disguised = DatabaseUtils.parseBoolean(result, "disguised");

core/src/main/java/dev/pgm/community/sessions/services/SessionDataQuery.java

@@ -6,19 +6,31 @@ public interface SessionDataQuery {
   static final String TABLE_FIELDS =
       "(id VARCHAR(36) PRIMARY KEY, player VARCHAR(36), disguised BOOL, server VARCHAR(32), start_time BIGINT, end_time BIGINT)";
 
-  static final String INSERT_SESSION_QUERY = "INSERT INTO "
-      + TABLE_NAME
+  static final String INSERT_SESSION_QUERY = "INSERT INTO " + TABLE_NAME
       + "(id, player, disguised, server, start_time, end_time) VALUES (?, ?, ?, ?, ?, ?)";
 
-  static final String SELECT_DISGUISED_SESSION_QUERY = "SELECT * FROM " + TABLE_NAME
-      + " WHERE player = ? AND disguised = 0 ORDER BY end_time DESC LIMIT 1";
-
-  static final String SELECT_SESSION_QUERY =
-      "SELECT * FROM " + TABLE_NAME + " WHERE player = ? ORDER BY end_time DESC LIMIT 1";
-
   static final String UPDATE_SESSION_ENDTIME_QUERY =
       "UPDATE " + TABLE_NAME + " SET end_time = ? WHERE id = ?";
 
   static final String UPDATE_ONGOING_SESSION_ENDING_QUERY =
       "UPDATE " + TABLE_NAME + " SET end_time = ? WHERE server = ? AND end_time IS NULL";
+
+  static final String LATEST_TABLE_NAME = "latest_sessions";
+  static final String LATEST_TABLE_FIELDS =
+      "(player VARCHAR(36), ignore_disguised BOOL, session_id VARCHAR(36), disguised BOOL, server VARCHAR(32), start_time BIGINT, end_time BIGINT, PRIMARY KEY (player, ignore_disguised))";
+
+  static final String UPSERT_LATEST_SESSION_QUERY = "INSERT INTO " + LATEST_TABLE_NAME
+      + "(player, ignore_disguised, session_id, disguised, server, start_time, end_time)"
+      + " VALUES (?, ?, ?, ?, ?, ?, ?) ON DUPLICATE KEY UPDATE "
+      + "session_id = VALUES(session_id), disguised = VALUES(disguised), "
+      + "server = VALUES(server), start_time = VALUES(start_time), end_time = VALUES(end_time)";
+
+  static final String SELECT_LATEST_SESSION_QUERY =
+      "SELECT * FROM " + LATEST_TABLE_NAME + " WHERE player = ? AND ignore_disguised = ? LIMIT 1";
+
+  static final String UPDATE_LATEST_ENDTIME_QUERY =
+      "UPDATE " + LATEST_TABLE_NAME + " SET end_time = ? WHERE session_id = ?";
+
+  static final String UPDATE_LATEST_ONGOING_SESSION_ENDING_QUERY =
+      "UPDATE " + LATEST_TABLE_NAME + " SET end_time = ? WHERE server = ? AND end_time IS NULL";
 }