Add id index to the work queue table

Needed to efficiently delete ids during processing.
timescale · Oct 27, 2023 · ace0150 · ace0150
1 parent 2ae8fa6
commit ace0150
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/nbs/01_pgvectorizer.ipynb b/nbs/01_pgvectorizer.ipynb
@@ -87,6 +87,8 @@
     "                        id int\n",
     "                    );\n",
     "\n",
+    "                    CREATE INDEX ON {self.schema_name}.{self.work_queue_table_name}(id);\n",
+    "\n",
     "                    CREATE OR REPLACE FUNCTION {self.schema_name}.{self.trigger_name_fn}() RETURNS TRIGGER LANGUAGE PLPGSQL AS $$ \n",
     "                    BEGIN \n",
     "                        IF (TG_OP = 'DELETE') THEN\n",

diff --git a/timescale_vector/pgvectorizer.py b/timescale_vector/pgvectorizer.py
@@ -60,6 +60,8 @@ def register(self):
                         id int
                     );
 
+                    CREATE INDEX ON {self.schema_name}.{self.work_queue_table_name}(id);
+
                     CREATE OR REPLACE FUNCTION {self.schema_name}.{self.trigger_name_fn}() RETURNS TRIGGER LANGUAGE PLPGSQL AS $$ 
                     BEGIN 
                         IF (TG_OP = 'DELETE') THEN
-Original file line number
+Diff line change
@@ Expand Up / @@ -60,6 +60,8 @@ def register(self): @@
                             id int
                         );
+                        CREATE INDEX ON {self.schema_name}.{self.work_queue_table_name}(id);
                         CREATE OR REPLACE FUNCTION {self.schema_name}.{self.trigger_name_fn}() RETURNS TRIGGER LANGUAGE PLPGSQL AS $$
                         BEGIN
                             IF (TG_OP = 'DELETE') THEN
@@ Expand Down @@