FT-770 Added logic to prevent updating of connection name.

samples/packages/enrichment-migrator/src/main/kotlin/EnrichmentMigrator.kt

@@ -3,14 +3,17 @@
 import co.elastic.clients.elasticsearch._types.SortOrder
 import com.atlan.exception.ErrorCode
 import com.atlan.exception.InvalidRequestException
+import com.atlan.exception.NotFoundException
 import com.atlan.model.assets.Asset
+import com.atlan.model.assets.Connection
 import com.atlan.model.assets.Database
 import com.atlan.model.fields.CustomMetadataField
 import com.atlan.pkg.Utils
 import com.atlan.pkg.aim.Importer
 import com.atlan.pkg.serde.RowSerde
 import mu.KotlinLogging
 import java.io.File
+import kotlin.jvm.optionals.getOrElse
 
 /**
  * Actually run the migrator, taking all settings from environment variables.
@@ -103,6 +106,7 @@ object EnrichmentMigrator {
                     MigratorContext(
                         sourceConnectionQN = sourceConnectionQN,
                         targetConnectionQN = targetConnectionQN,
+                        targetConnectionName = getConnectionName(targetConnectionQN),
                         includeArchived = includeArchived,
                         sourceDatabaseName = sourceDatabaseName,
                         targetDatabaseName = targetDatabaseName,
@@ -157,6 +161,19 @@ object EnrichmentMigrator {
         return databaseNames
     }
 
+    @JvmStatic
+    fun getConnectionName(connectionQN: String): String {
+        val connection =
+            Connection.select()
+                .where(Asset.QUALIFIED_NAME.eq(connectionQN))
+                .stream()
+                .findFirst()
+                .getOrElse {
+                    throw NotFoundException(ErrorCode.ASSET_NOT_FOUND_BY_QN, connectionQN, "Connection")
+                }
+        return connection.name
+    }
+
     @JvmStatic
     fun getTargetDatabaseName(
         targetConnectionQN: String,
@@ -166,7 +183,7 @@ object EnrichmentMigrator {
             return listOf("")
         }
         val databaseNames = getDatabaseNames(targetConnectionQN, targetDatabasePattern)
-        if (databaseNames.size < 1) {
+        if (databaseNames.isEmpty()) {
             throw InvalidRequestException(
                 ErrorCode.UNEXPECTED_NUMBER_OF_DATABASES_FOUND,
                 "at least one",
@@ -201,6 +218,7 @@ object EnrichmentMigrator {
     data class MigratorContext(
         val sourceConnectionQN: String,
         val targetConnectionQN: String,
+        val targetConnectionName: String,
         val includeArchived: Boolean,
         val sourceDatabaseName: String,
         val targetDatabaseName: String,

samples/packages/enrichment-migrator/src/main/kotlin/Transformer.kt

@@ -21,12 +21,16 @@ class Transformer(
     override fun mapRow(inputRow: Map<String, String>): List<List<String>> {
         // Pick the fields to include in the output based on the header,
         // and replace the source connection with the target connection
+        val rowIsConnection = (inputRow[Asset.TYPE_NAME.atlanFieldName] == "Connection")
         val values =
             header.map {
                 val raw =
                     if (it == Asset.STATUS.atlanFieldName) {
                         // Add the status column as explicitly ACTIVE (to convert any included archived assets)
                         AtlanStatus.ACTIVE.value
+                    } else if (rowIsConnection && it == Asset.NAME.atlanFieldName) {
+                        // Don't change the name of the connection
+                        ctx.targetConnectionName
                     } else {
                         inputRow[it] ?: ""
                     }