in components/post-setup-config/src/dbconfig_main.py [0:0]
def init_connection_pool(connector: Connector) -> sqlalchemy.engine.Engine:
# function used to generate database connection
def getconn():
conn = connector.connect(
instance_uri=os.environ["ALLOYDB_INSTANCE"],
driver="pg8000",
db=os.environ["ALLOYDB_DATABASE"],
enable_iam_auth=True,
user=os.environ["ALLOYDB_USER_CONFIG"],
ip_type=IPTypes.PSC,
)
return conn
# create connection pool
pool = sqlalchemy.create_engine(
"postgresql+pg8000://",
creator=getconn,
)
return pool