fix: add index to speed up download with >10k notes DB

fix #14
vzhd1701 · Feb 14, 2022 · 2dbb599 · 2dbb599
1 parent 0430bfd
commit 2dbb599
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/evernote_backup/config.py b/evernote_backup/config.py
@@ -1,4 +1,4 @@
 API_DATA = b"Q0sJVEtPDWt7bmg2YnhtY344enZlMSRsPjxtaTI7P2Q0PTRhNWc="
 API_DATA_OAUTH = b"eW91c3V3dn9mYjF2az48bzM7Pm4wZzdlZzpk"
 
-CURRENT_DB_VERSION = 4
+CURRENT_DB_VERSION = 5
diff --git a/evernote_backup/note_storage.py b/evernote_backup/note_storage.py
@@ -48,6 +48,8 @@ class NoteForSync(NamedTuple):
                      ON notes(title COLLATE NOCASE);
                     CREATE INDEX IF NOT EXISTS idx_notebooks_linked
                      ON notebooks_linked(guid, notebook_guid);
+                    CREATE INDEX IF NOT EXISTS idx_notes_raw
+                     ON notes(guid) WHERE raw_note IS NULL;
 """
 
 
@@ -131,6 +133,13 @@ def upgrade_db(self, db_version: int) -> None:
                     " ON notebooks_linked(guid, notebook_guid);"
                 )
 
+        if db_version < 5:
+            with self.db as con4:
+                con4.execute(
+                    "CREATE INDEX IF NOT EXISTS idx_notes_raw_null"
+                    " ON notes(guid) WHERE raw_note IS NULL;"
+                )
+
         self.config.set_config_value("DB_VERSION", str(CURRENT_DB_VERSION))
 
         if need_resync: