[DRAFT] - fix: add pg8000 graceful shutdown on Cloud Run scale-down

api/namex/__init__.py

@@ -18,7 +18,7 @@
 
 jwt = JwtManager()
 
-from cloud_sql_connector import DBConfig, setup_search_path_event_listener
+from cloud_sql_connector import DBConfig, setup_pg8000_close_event_listener, setup_search_path_event_listener
 from flask_cors import CORS
 from flask_migrate import Migrate, upgrade
 
@@ -78,6 +78,7 @@ def create_app(run_mode=os.getenv('FLASK_ENV', 'production')):  # noqa: B008
         with app.app_context():
             engine = db.engine
             setup_search_path_event_listener(engine, schema)
+            setup_pg8000_close_event_listener(engine)
 
     if run_mode == 'migration':
         Migrate(app, db)

services/namex-pay/src/namex_pay/__init__.py

@@ -40,7 +40,7 @@
 
 from flask import Flask
 from flask_restx import Api
-from namex import DBConfig, setup_search_path_event_listener
+from namex import DBConfig, setup_pg8000_close_event_listener, setup_search_path_event_listener
 from namex.models import db
 from namex.resources.ops import api as nr_ops
 from namex.services import queue
@@ -75,6 +75,7 @@ def create_app(environment: Config = ProdConfig, **kwargs) -> Flask:
         with app.app_context():
             engine = db.engine
             setup_search_path_event_listener(engine, schema)
+            setup_pg8000_close_event_listener(engine)
 
     queue.init_app(app)
     register_endpoints(app)

services/solr-names-updater/src/solr_names_updater/__init__.py

@@ -40,7 +40,7 @@
 
 from flask import Flask
 from flask_restx import Api
-from namex import DBConfig, setup_search_path_event_listener
+from namex import DBConfig, setup_pg8000_close_event_listener, setup_search_path_event_listener
 from namex.models import db
 from namex.resources.ops import api as nr_ops
 from namex.services import queue
@@ -73,6 +73,7 @@ def create_app(environment: Config = ProdConfig, **kwargs) -> Flask:
         with app.app_context():
             engine = db.engine
             setup_search_path_event_listener(engine, schema)
+            setup_pg8000_close_event_listener(engine)
 
     queue.init_app(app)
     register_endpoints(app)

solr-admin-app/solr_admin/__init__.py

@@ -2,7 +2,7 @@
 
 import flask
 import flask_admin
-from cloud_sql_connector import DBConfig, setup_search_path_event_listener
+from cloud_sql_connector import DBConfig, setup_pg8000_close_event_listener, setup_search_path_event_listener
 from sqlalchemy.orm import scoped_session, sessionmaker
 from structured_logging import StructuredLogging
 
@@ -60,6 +60,7 @@ def create_application(run_mode=os.getenv('FLASK_ENV', 'production')):
     with application.app_context():
         engine = models.db.engine
         setup_search_path_event_listener(engine, schema)
+        setup_pg8000_close_event_listener(engine)
 
     # The root page - point the users to the admin interface.
     @application.route('/')

solr-synonyms-api/synonyms/__init__.py

@@ -7,7 +7,7 @@
 import flask
 from flask import Flask
 from flask_jwt_oidc import JwtManager
-from namex import DBConfig, setup_search_path_event_listener
+from namex import DBConfig, setup_pg8000_close_event_listener, setup_search_path_event_listener
 from structured_logging import StructuredLogging
 
 from config import CONFIGURATION
@@ -56,6 +56,7 @@ def create_app(run_mode=None):
         with app.app_context():
             engine = db.engine
             setup_search_path_event_listener(engine, schema)
+            setup_pg8000_close_event_listener(engine)
 
     # Initialize Marshmallow with handling for SQLAlchemy auto-detection issues
     try: