Added support for connection retry params. These params are passed to Python driver

- retry_count -> Number of times to retry
- retry_delay -> delay (seconds) in between each retry

Author: aosingh

dbt/adapters/oracle/connections.py

@@ -70,6 +70,10 @@ class OracleAdapterCredentials(Credentials):
     cclass: Optional[str] = None
     purity: Optional[str] = None
 
+    # Connection retry params
+    retry_count: Optional[int] = 1
+    retry_delay: Optional[int] = 3
+
 
     _ALIASES = {
         'dbname': 'database',
@@ -93,7 +97,8 @@ def _connection_keys(self) -> Tuple[str]:
             'protocol', 'host', 'port', 'tns_name',
             'service', 'connection_string',
             'shardingkey', 'supershardingkey',
-            'cclass', 'purity'
+            'cclass', 'purity', 'retry_count',
+            'retry_delay'
         )
 
     @classmethod
@@ -125,7 +130,7 @@ def get_dsn(self) -> str:
             return self.connection_string
 
         # Assume host connection method OracleConnectionMethod.HOST and necessary parameters are defined
-        return f'{self.protocol}://{self.host}:{self.port}/{self.service}'
+        return f'{self.protocol}://{self.host}:{self.port}/{self.service}?retry_count={self.retry_count}&retry_delay={self.retry_delay}'
 
 
 class OracleAdapterConnectionManager(SQLConnectionManager):

dbt_adbs_test_project/profiles.yml

@@ -11,6 +11,8 @@ dbt_test:
          service: "{{ env_var('DBT_ORACLE_SERVICE') }}"
          #database: "{{ env_var('DBT_ORACLE_DATABASE') }}"
          schema: "{{ env_var('DBT_ORACLE_SCHEMA') }}"
+         retry_count: 1
+         retry_delay: 5
          shardingkey:
            - skey
          supershardingkey: