Get client id from env var

snowplow · Dec 14, 2023 · 0bb7de4 · 0bb7de4
1 parent 039d62b
commit 0bb7de4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...wanalytics.snowplow.collectors.scalastream/sinks/AzureAuthenticationCallbackHandler.scala b/...wanalytics.snowplow.collectors.scalastream/sinks/AzureAuthenticationCallbackHandler.scala
@@ -27,7 +27,11 @@ import com.nimbusds.jwt.JWTParser
 
 class AzureAuthenticationCallbackHandler extends AuthenticateCallbackHandler {
 
-  val credentials: MSICredentials = new MSICredentials()
+  val credentials: MSICredentials = {
+    val clientId = sys.env.get("AZURE_CLIENT_ID_FOR_EVENT_HUB").orElse(sys.env.get("AZURE_CLIENT_ID"))
+    val creds    = new MSICredentials()
+    clientId.map(creds.withClientId).getOrElse(creds)
+  }
 
   var sbUri: String = ""