diff --git a/docker-compose.yml b/docker-compose.yml index 1f83a7320b..a34730965d 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -916,6 +916,7 @@ services: TASK_API_ENFORCE_HTTPS: ${TASK_API_ENFORCE_HTTPS:-true} LOW_NUMBER_SUPPRESSION_THRESHOLD: ${LOW_NUMBER_SUPPRESSION_THRESHOLD:-10} ROUNDING_TARGET: ${ROUNDING_TARGET:-10} + D2E_BIGQUERY_JDBC_URL: ${D2E_BIGQUERY_JDBC_URL:-} # Needed only for future alterations of schema when version upgrades # alp-logto-db-post-alterations-init: diff --git a/plugins/flows/_shared_flow_utils/dao/daobase.py b/plugins/flows/_shared_flow_utils/dao/daobase.py index 185106ed46..31071302b3 100644 --- a/plugins/flows/_shared_flow_utils/dao/daobase.py +++ b/plugins/flows/_shared_flow_utils/dao/daobase.py @@ -33,7 +33,8 @@ def build_bigquery_r_connection_string( "jdbc:bigquery://https://www.googleapis.com/bigquery/v2:443;" f"ProjectId={project};OAuthType=0;" f"OAuthServiceAcctEmail={client_email};" - f"OAuthPvtKeyPath={key_path}" + f"OAuthPvtKeyPath={key_path};" + "EnableSession=1" ) return ( "connectionDetails <- DatabaseConnector::createConnectionDetails(" diff --git a/plugins/flows/_shared_flow_utils/tests/test_r_connection_string.py b/plugins/flows/_shared_flow_utils/tests/test_r_connection_string.py index f54f68c1b0..fe2a5d8b3e 100644 --- a/plugins/flows/_shared_flow_utils/tests/test_r_connection_string.py +++ b/plugins/flows/_shared_flow_utils/tests/test_r_connection_string.py @@ -13,6 +13,7 @@ def test_bigquery_r_connection_string_shape(): "connectionString = 'jdbc:bigquery://https://www.googleapis.com/bigquery/v2:443;" "ProjectId=my-gcp-project;OAuthType=0;" "OAuthServiceAcctEmail=svc@my-gcp-project.iam.gserviceaccount.com;" - "OAuthPvtKeyPath=/tmp/google-sa.json'" in result + "OAuthPvtKeyPath=/tmp/google-sa.json;" + "EnableSession=1'" in result ) assert "user = ''" in result and "password = ''" in result