From 781384cb86d93ad587f6bbc48c384013d97ac8a9 Mon Sep 17 00:00:00 2001
From: teidesu <86301490+teidesu@users.noreply.github.com>
Date: Wed, 4 Aug 2021 12:58:17 +0300
Subject: [PATCH] fix(sqlite): removed unique idx on phone, optimized queries

---
 packages/sqlite/index.ts | 30 +++++++++++++++++-------------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/packages/sqlite/index.ts b/packages/sqlite/index.ts
index f6263dd7..59018373 100644
--- a/packages/sqlite/index.ts
+++ b/packages/sqlite/index.ts
@@ -91,7 +91,7 @@ const SCHEMA = `
         "full" blob
     );
     create index idx_entities_username on entities (username);
-    create unique index idx_entities_phone on entities (phone);
+    create index idx_entities_phone on entities (phone);
 `
 
 const RESET = `
@@ -147,10 +147,10 @@ const STATEMENTS = {
     upsertEnt:
         'insert or replace into entities (id, hash, type, username, phone, updated, "full") values (?, ?, ?, ?, ?, ?, ?)',
     getEntById: 'select * from entities where id = ?',
-    getEntByPhone: 'select * from entities where phone = ?',
-    getEntByUser: 'select * from entities where username = ?',
+    getEntByPhone: 'select * from entities where phone = ? limit 1',
+    getEntByUser: 'select * from entities where username = ? limit 1',
 
-    delStaleState: 'delete from state where expires < ?'
+    delStaleState: 'delete from state where expires < ?',
 } as const
 
 const EMPTY_BUFFER = Buffer.alloc(0)
@@ -297,7 +297,6 @@ export class SqliteStorage implements ITelegramStorage /*, IStateStorage */ {
             this._pendingUnimportant = {}
         }, params?.unimportantSavesDelay ?? 30000)
 
-
         this._vacuumInterval = params?.vacuumInterval ?? 300_000
 
         // todo: add support for workers (idk if really needed, but still)
@@ -465,21 +464,26 @@ export class SqliteStorage implements ITelegramStorage /*, IStateStorage */ {
         return this._setToKv('self', self)
     }
 
-    getUpdatesState(): [number, number, number] | null {
+    getUpdatesState(): [number, number, number, number] | null {
         const pts = this._getFromKv('pts')
         if (pts == null) return null
 
-        return [pts, this._getFromKv('date')!, this._getFromKv('seq')!]
-    }
-
-    setCommonPts(val: [number, number] | null): void {
-        return this._setToKv('cpts', val)
+        return [
+            pts,
+            this._getFromKv('qts')!,
+            this._getFromKv('date')!,
+            this._getFromKv('seq')!,
+        ]
     }
 
     setUpdatesPts(val: number): void {
         return this._setToKv('pts', val)
     }
 
+    setUpdatesQts(val: number): void {
+        return this._setToKv('qts', val)
+    }
+
     setUpdatesDate(val: number): void {
         return this._setToKv('date', val)
     }
@@ -607,7 +611,7 @@ export class SqliteStorage implements ITelegramStorage /*, IStateStorage */ {
             const full = this._readFullPeer(row.full)
             this._addToCache(id, {
                 peer: getInputPeer(row),
-                full
+                full,
             })
             return full
         }
@@ -689,7 +693,7 @@ export class SqliteStorage implements ITelegramStorage /*, IStateStorage */ {
             // expired or does not exist
             const item: FsmItem = {
                 expires: now + window * 1000,
-                value: limit
+                value: limit,
             }
 
             this._statements.setState.run(