From 8c8920d772cdfb9f7d23d36f283499d246f9bd57 Mon Sep 17 00:00:00 2001
From: Sylvain Tedoldi <sylvain.github@contact.tedoldi.fr>
Date: Mon, 11 Apr 2016 12:39:55 +0200
Subject: [PATCH] Heavily uses OrientDb indexes and SQL requests.

---
 .../tedoldi/lichess/game/retriever/game.clj   |   9 +-
 .../lichess/game/retriever/orientdb.clj       | 142 +++++++++++++++---
 2 files changed, 123 insertions(+), 28 deletions(-)

diff --git a/src/fr/tedoldi/lichess/game/retriever/game.clj b/src/fr/tedoldi/lichess/game/retriever/game.clj
index 729e41b..0ecfc87 100644
--- a/src/fr/tedoldi/lichess/game/retriever/game.clj
+++ b/src/fr/tedoldi/lichess/game/retriever/game.clj
@@ -15,12 +15,7 @@
 (declare username->games)
 
 (defn last-game-timestamp [dal username]
-  (->> username
-    (username->games dal)
-    (sort-by :timestamp)
-    reverse
-    first
-    :timestamp))
+  (:timestamp (dal/username->last-game dal username)))
 
 (defn update-user [dal url username]
   (-> (str "Updating user " username " from server.\n")
@@ -67,7 +62,7 @@
 
 (defn username->games
   ([dal username]
-   (dal/find-all-deep dal "game" {:userId username}))
+   (dal/username->games dal username))
 
   ([dal username custom-filter]
    (->> (username->games dal username)
diff --git a/src/fr/tedoldi/lichess/game/retriever/orientdb.clj b/src/fr/tedoldi/lichess/game/retriever/orientdb.clj
index b310bcd..06560db 100644
--- a/src/fr/tedoldi/lichess/game/retriever/orientdb.clj
+++ b/src/fr/tedoldi/lichess/game/retriever/orientdb.clj
@@ -9,11 +9,17 @@
            (com.orientechnologies.orient.core.sql OCommandSQL)
            (com.orientechnologies.orient.core.record.impl ODocument)
            (com.orientechnologies.orient.core.db.document ODatabaseDocumentPool ODatabaseDocumentTx)
+           (com.orientechnologies.orient.core.id ORecordId)
            (com.orientechnologies.orient.core.query OQuery)
            (com.orientechnologies.orient.core.sql.query OSQLSynchQuery)))
 
 
 
+(defprotocol LichessDal
+
+  (username->games [this username])
+  (username->last-game [this username]))
+
 
 
 
@@ -302,6 +308,72 @@
         (.close db)))))
 
 
+(def ^:private undefined-ORecordId
+  (ORecordId. -1 -1))
+
+(defn- -username->games-paginated
+  ([dal username] (-username->games-paginated dal
+                                              username
+                                              undefined-ORecordId))
+
+  ([dal username lower-rid]
+   (let [page-size 100
+         query     (OSQLSynchQuery. (str "SELECT FROM game"
+                                         " WHERE "
+                                         "       userId = ?"
+                                         "   AND @rid > " lower-rid
+                                         " LIMIT " page-size)
+                                    page-size)
+         db        (pool->db (:config dal))]
+     (try
+
+       (let [items (seq
+                     (-> db
+                         .activateOnCurrentThread
+
+                         (.query ^OSQLSynchQuery query (to-array [username]))))]
+         (when items
+           (cons items
+                 (-username->games-paginated
+                   dal
+                   username
+                   (.getIdentity (last items))))))
+
+       (catch java.lang.IllegalArgumentException e nil)
+
+       (finally
+         (.close db))))))
+
+(defn- -find-all-deep-paginated
+  ([dal collection] (-find-all-deep-paginated dal
+                                              collection
+                                              undefined-ORecordId))
+
+  ([dal collection lower-rid]
+   (let [page-size 100
+         query     (OSQLSynchQuery. (str "SELECT FROM " (kw->oclass-name collection)
+                                         " WHERE @rid > " lower-rid
+                                         " LIMIT " page-size)
+                                    page-size)
+         db        (pool->db (:config dal))]
+     (try
+
+       (let [items (seq
+                     (-> db
+                         .activateOnCurrentThread
+
+                         (.query ^OSQLSynchQuery query nil)))]
+         (when items
+           (cons items
+                 (-find-all-deep-paginated
+                   dal
+                   collection
+                   (.getIdentity (last items))))))
+
+       (catch java.lang.IllegalArgumentException e nil)
+
+       (finally
+         (.close db))))))
 
 
 
@@ -367,28 +439,17 @@
   (find-all-deep
     [this collection request]
 
-    (let [db (pool->db config)]
-      (try
-
-        (.activateOnCurrentThread db)
-
-        (->> collection
-             kw->oclass-name
-
-             (.browseClass db)
-             iterator-seq
-
-             (map ODocument->map)
-
-             (filter #(not (second (clojure.data/diff % request))))
-             (remove :sentinel)
-
-             (into []))
 
-        (catch java.lang.IllegalArgumentException e [])
+    (flatten
+      (lazy-cat
+        (map
+          (fn [raw-items]
+            (->> raw-items
+                 (map ODocument->map)
+                 (remove :sentinel)
 
-        (finally
-          (.close db)))))
+                 (filter #(not (second (clojure.data/diff % request))))))
+          (-find-all-deep-paginated this collection)))))
 
 
   (find-by-id
@@ -477,4 +538,43 @@
   (patch!
     [this collection id item]
     (let [document (find-by-id this collection id)]
-      (update! this collection id (merge document item)))))
+      (update! this collection id (merge document item))))
+
+  LichessDal
+
+  (username->games
+    [this username]
+
+    (flatten
+      (lazy-cat
+        (map
+          (fn [raw-items]
+            (->> raw-items
+                 (map ODocument->map)
+                 (remove :sentinel)))
+          (-username->games-paginated this username)))))
+
+  (username->last-game
+    [this username]
+
+    (let [query     (OSQLSynchQuery. (str "SELECT FROM game"
+                                          " WHERE userId = ?"
+                                          " ORDER BY timestamp DESC"
+                                          " LIMIT " 1)
+                                     1)
+          db        (pool->db config)]
+      (try
+        (-> db
+            .activateOnCurrentThread
+
+            (.query ^OSQLSynchQuery query (to-array [username]))
+            seq
+            first
+            (#(when %
+                (when-not (:sentinel %)
+                  (ODocument->map %)))))
+
+        (catch java.lang.IllegalArgumentException e nil)
+
+        (finally
+          (.close db))))))