[ADD] pg.query_ids: query large numbers of ids memory-safely

This is mainly the code that has been recently added to `orm.recompute_fields`,
here we're making it re-usasble.

Author: cawo-odoo

src/util/pg.py

@@ -1932,3 +1932,41 @@ def bulk_update_table(cr, table, columns, mapping, key_col="id"):
         key_col=key_col,
     )
     cr.execute(query, [Json(mapping)])
+
+
+class query_ids(named_cursor):
+    def __init__(self, cr, query, itersize=None):
+        self._cr = cr
+        self._tmp_tbl = "_upgrade_query_ids_{}".format(uuid.uuid4().hex)
+        cr.execute(
+            format_query(
+                cr,
+                "CREATE UNLOGGED TABLE {}(id) AS (WITH query AS ({}) SELECT * FROM query)",
+                self._tmp_tbl,
+                SQLStr(query),
+            )
+        )
+        self._len = cr.rowcount
+        cr.execute(
+            format_query(
+                cr, "ALTER TABLE {} ADD CONSTRAINT {} PRIMARY KEY (id)", self._tmp_tbl, "pk_{}_id".format(self._tmp_tbl)
+            )
+        )
+        super(query_ids, self).__init__(cr, itersize)
+        self.execute(format_query(cr, "SELECT id FROM {} ORDER BY id", self._tmp_tbl))
+
+    def __len__(self):
+        return self._len
+
+    def __iter__(self):
+        return (id_ for (id_,) in super(query_ids, self).__iter__())
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self._cr.execute(format_query(self._cr, "DROP TABLE IF EXISTS {}", self._tmp_tbl))
+        self._cr = None
+        return super(query_ids, self).__exit__(exc_type, exc_value, traceback)
+
+    def __del__(self):
+        if self._cr:
+            self._cr.execute(format_query(self._cr, "DROP TABLE IF EXISTS {}", self._tmp_tbl))
+        return super(query_ids, self).__del__()