feat: enable lazy refresh for postgres, mysql and sql server connectors

Author: kgala2

cloud-sql/mysql/sqlalchemy/connect_connector.py

@@ -15,7 +15,7 @@
 # [START cloud_sql_mysql_sqlalchemy_connect_connector]
 import os
 
-from google.cloud.sql.connector import Connector, IPTypes
+from google.cloud.sql.connector import Connector, IPTypes, RefreshStrategy
 import pymysql
 
 import sqlalchemy
@@ -40,8 +40,9 @@ def connect_with_connector() -> sqlalchemy.engine.base.Engine:
     db_name = os.environ["DB_NAME"]  # e.g. 'my-database'
 
     ip_type = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
+    refresh_strategy = RefreshStrategy.LAZY
 
-    connector = Connector(ip_type)
+    connector = Connector(ip_type = ip_type, refresh_strategy = refresh_strategy)
 
     def getconn() -> pymysql.connections.Connection:
         conn: pymysql.connections.Connection = connector.connect(

cloud-sql/mysql/sqlalchemy/connect_connector_auto_iam_authn.py

@@ -15,7 +15,7 @@
 # [START cloud_sql_mysql_sqlalchemy_auto_iam_authn]
 import os
 
-from google.cloud.sql.connector import Connector, IPTypes
+from google.cloud.sql.connector import Connector, IPTypes, RefreshStrategy
 import pymysql
 
 import sqlalchemy
@@ -38,9 +38,10 @@ def connect_with_connector_auto_iam_authn() -> sqlalchemy.engine.base.Engine:
     db_name = os.environ["DB_NAME"]  # e.g. 'my-database'
 
     ip_type = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
+    refresh_strategy = RefreshStrategy.LAZY
 
     # initialize Cloud SQL Python Connector object
-    connector = Connector()
+    connector = Connector(refresh_strategy = refresh_strategy)
 
     def getconn() -> pymysql.connections.Connection:
         conn: pymysql.connections.Connection = connector.connect(

cloud-sql/postgres/sqlalchemy/connect_connector.py

@@ -15,7 +15,7 @@
 # [START cloud_sql_postgres_sqlalchemy_connect_connector]
 import os
 
-from google.cloud.sql.connector import Connector, IPTypes
+from google.cloud.sql.connector import Connector, IPTypes, RefreshStrategy
 import pg8000
 
 import sqlalchemy
@@ -40,9 +40,10 @@ def connect_with_connector() -> sqlalchemy.engine.base.Engine:
     db_name = os.environ["DB_NAME"]  # e.g. 'my-database'
 
     ip_type = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
+    refresh_strategy = RefreshStrategy.LAZY
 
     # initialize Cloud SQL Python Connector object
-    connector = Connector()
+    connector = Connector(refresh_strategy = refresh_strategy)
 
     def getconn() -> pg8000.dbapi.Connection:
         conn: pg8000.dbapi.Connection = connector.connect(

cloud-sql/postgres/sqlalchemy/connect_connector_auto_iam_authn.py

@@ -15,7 +15,7 @@
 # [START cloud_sql_postgres_sqlalchemy_auto_iam_authn]
 import os
 
-from google.cloud.sql.connector import Connector, IPTypes
+from google.cloud.sql.connector import Connector, IPTypes, RefreshStrategy
 import pg8000
 
 import sqlalchemy
@@ -38,9 +38,10 @@ def connect_with_connector_auto_iam_authn() -> sqlalchemy.engine.base.Engine:
     db_name = os.environ["DB_NAME"]  # e.g. 'my-database'
 
     ip_type = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
+    refresh_strategy = RefreshStrategy.LAZY
 
     # initialize Cloud SQL Python Connector object
-    connector = Connector()
+    connector = Connector(refresh_strategy = refresh_strategy)
 
     def getconn() -> pg8000.dbapi.Connection:
         conn: pg8000.dbapi.Connection = connector.connect(

cloud-sql/sql-server/sqlalchemy/connect_connector.py

@@ -15,7 +15,7 @@
 # [START cloud_sql_sqlserver_sqlalchemy_connect_connector]
 import os
 
-from google.cloud.sql.connector import Connector, IPTypes
+from google.cloud.sql.connector import Connector, IPTypes, RefreshStrategy
 import pytds
 
 import sqlalchemy
@@ -40,8 +40,9 @@ def connect_with_connector() -> sqlalchemy.engine.base.Engine:
     db_name = os.environ["DB_NAME"]  # e.g. 'my-database'
 
     ip_type = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
+    refresh_strategy = RefreshStrategy.LAZY
 
-    connector = Connector(ip_type)
+    connector = Connector(ip_type = ip_type, refresh_strategy = refresh_strategy)
 
     connect_args = {}
     # If your SQL Server instance requires SSL, you need to download the CA