Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-server-tools-postgres_vacuum/postgres_vacuum/models/ir_cron.py

@@ -0,0 +1,67 @@
+# Copyright 2025 Therp BV
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl-3.0)
+
+import logging
+from datetime import datetime, timedelta
+
+from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
+
+from odoo import models
+from odoo.sql_db import db_connect
+
+
+class IrCron(models.Model):
+    _inherit = "ir.cron"
+
+    def _postgres_vacuum(self, max_minutes, full_vacuum):
+        """
+        Vacuum or analyze tables, don't continue after max_minutes have elapsed
+
+        Do either a VACUUM FULL (full_vacuum=True) or ANALYZE
+        """
+
+        _logger = logging.getLogger("odoo.addons.postgres_vacuum")
+
+        query = "ANALYZE (SKIP_LOCKED)"
+        if full_vacuum:
+            query = "VACUUM (FULL, SKIP_LOCKED)"
+
+        current_time = datetime.now()
+        exit_time_window = current_time + timedelta(minutes=max_minutes)
+
+        order_by_clause = "last_analyze"
+        if full_vacuum:
+            order_by_clause = "last_vacuum"
+
+        connection = db_connect(self.env.cr.dbname)
+
+        with connection.cursor() as cr:
+            cr._cnx.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
+            # pylint: disable=sql-injection
+            cr.execute(f"SET statement_timeout = '{int(max_minutes)}min'")
+
+            # pylint: disable=sql-injection
+            cr.execute(
+                f"""
+                SELECT relname
+                FROM pg_stat_user_tables
+                ORDER BY {order_by_clause} ASC NULLS FIRST
+                """
+            )
+            for (table,) in cr.fetchall():
+                action = "vacuum" if full_vacuum else "analyze"
+
+                if datetime.now() >= exit_time_window:
+                    _logger.info(
+                        "{max_minutes} minutes elapsed, not continuing to {action}"
+                    )
+                    return
+
+                _logger.debug(f"{action} of table {table}")
+                try:
+                    # pylint: disable=sql-injection
+                    cr.execute(f'{query} "{table}"')
+                except Exception:
+                    _logger.exception(f"{action} of table {table} failed")
+                else:
+                    _logger.debug(f"{action} of table {table} done")