Address Ian's comments

snowplow · Dec 14, 2023 · 039d62b · 039d62b
1 parent 7cc0c5b
commit 039d62b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/...wanalytics.snowplow.collectors.scalastream/sinks/AzureAuthenticationCallbackHandler.scala b/...wanalytics.snowplow.collectors.scalastream/sinks/AzureAuthenticationCallbackHandler.scala
@@ -36,8 +36,18 @@ class AzureAuthenticationCallbackHandler extends AuthenticateCallbackHandler {
     saslMechanism: String,
     jaasConfigEntries: util.List[AppConfigurationEntry]
   ): Unit = {
-    val bootstrapServer = configs.get(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG).toString.replaceAll("\\[|\\]", "")
-    val uri             = URI.create("https://" + bootstrapServer)
+    val bootstrapServer =
+      configs
+        .get(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG)
+        .toString
+        .replaceAll("\\[|\\]", "")
+        .split(",")
+        .toList
+        .headOption match {
+        case Some(s) => s
+        case None    => throw new Exception("Empty bootstrap servers list")
+      }
+    val uri = URI.create("https://" + bootstrapServer)
     this.sbUri = uri.getScheme + "://" + uri.getHost
   }