Implemented transient error handling for system errors in MSSQL (#48484)

airbyte-integrations/connectors/source-mssql/build.gradle

@@ -35,3 +35,7 @@ dependencies {
     testImplementation 'org.hamcrest:hamcrest-all:1.3'
     testImplementation 'org.testcontainers:mssqlserver:1.19.0'
 }
+
+compileKotlin {
+
+}

airbyte-integrations/connectors/source-mssql/metadata.yaml

@@ -9,7 +9,7 @@ data:
   connectorSubtype: database
   connectorType: source
   definitionId: b5ea17b1-f170-46dc-bc31-cc744ca984c1
-  dockerImageTag: 4.1.15
+  dockerImageTag: 4.1.16
   dockerRepository: airbyte/source-mssql
   documentationUrl: https://docs.airbyte.com/integrations/sources/mssql
   githubIssueLabel: source-mssql

airbyte-integrations/connectors/source-mssql/src/main/java/io/airbyte/integrations/source/mssql/MssqlSource.java

@@ -572,8 +572,9 @@ public JdbcDatabase createDatabase(final JsonNode sourceConfig) throws SQLExcept
 
   public static void main(final String[] args) throws Exception {
     final Source source = MssqlSource.sshWrappedSource(new MssqlSource());
+    final MSSqlSourceExceptionHandler exceptionHandler = new MSSqlSourceExceptionHandler();
     LOGGER.info("starting source: {}", MssqlSource.class);
-    new IntegrationRunner(source).run(args);
+    new IntegrationRunner(source).run(args, exceptionHandler);
     LOGGER.info("completed source: {}", MssqlSource.class);
   }
 

airbyte-integrations/connectors/source-mssql/src/main/kotlin/io/airbyte/integrations/source/mssql/MSSqlSourceExceptionHandler.kt

@@ -0,0 +1,33 @@
+/*
+ * Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.integrations.source.mssql
+
+import io.airbyte.cdk.integrations.util.ConnectorErrorProfile
+import io.airbyte.cdk.integrations.util.ConnectorExceptionHandler
+import io.airbyte.cdk.integrations.util.FailureType
+
+class MSSqlSourceExceptionHandler : ConnectorExceptionHandler() {
+    override fun initializeErrorDictionary() {
+
+        val DATABASE_READ_ERROR = "Encountered an error while reading the database, will retry"
+
+        // include common error profiles
+        super.initializeErrorDictionary()
+
+        // adding connector specific error profiles
+        add(
+            ConnectorErrorProfile(
+                errorClass = "MS SQL Exception", // which should we use?
+                regexMatchingPattern =
+                    ".*returned an incomplete response. The connection has been closed.*",
+                failureType = FailureType.TRANSIENT,
+                externalMessage = DATABASE_READ_ERROR,
+                sampleInternalMessage =
+                    "com.microsoft.sqlserver.jdbc.SQLServerException: SQL Server returned an incomplete response. The connection has been closed.",
+                referenceLinks = listOf("https://github.com/airbytehq/oncall/issues/6623")
+            )
+        )
+    }
+}