OLTP Connector Synapse integration using Linked Service (Azure#22116)

OLTP Connector Synapse integration using Linked Service

Author: aayush3011

sdk/cosmos/azure-cosmos-spark_3-1_2-12/src/main/scala/com/azure/cosmos/spark/AccountDataResolver.scala

@@ -0,0 +1,8 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.cosmos.spark
+
+trait AccountDataResolver {
+    def getAccountDataConfig(configs : Map[String, String]): Map[String, String]
+}

sdk/cosmos/azure-cosmos-spark_3-1_2-12/src/main/scala/com/azure/cosmos/spark/CosmosConfig.scala

@@ -20,7 +20,7 @@ import org.apache.spark.sql.connector.read.streaming.ReadLimit
 import java.net.{URI, URISyntaxException, URL}
 import java.time.format.DateTimeFormatter
 import java.time.{Duration, Instant}
-import java.util.Locale
+import java.util.{Locale, ServiceLoader}
 import scala.collection.immutable.{HashSet, Map}
 
 // scalastyle:off underscore.import
@@ -124,6 +124,8 @@ private object CosmosConfigNames {
   }
 }
 
+
+
 private object CosmosConfig {
   def getEffectiveConfig
   (
@@ -133,8 +135,18 @@ private object CosmosConfig {
     // spark application configteams
     userProvidedOptions: Map[String, String] // user provided config
   ) : Map[String, String] = {
+    var accountDataResolverCls = None : Option[AccountDataResolver]
+    val serviceLoader = ServiceLoader.load(classOf[AccountDataResolver])
+    val iterator = serviceLoader.iterator()
+    if (iterator.hasNext()) {
+        accountDataResolverCls = Some(iterator.next())
+    }
 
     var effectiveUserConfig = CaseInsensitiveMap(userProvidedOptions)
+    if (accountDataResolverCls.isDefined) {
+        val accountDataConfig = accountDataResolverCls.get.getAccountDataConfig(effectiveUserConfig)
+        effectiveUserConfig = CaseInsensitiveMap(accountDataConfig)
+    }
 
     if (databaseName.isDefined) {
       effectiveUserConfig += (CosmosContainerConfig.DATABASE_NAME_KEY -> databaseName.get)