Skip to content

feat: Support Choosing Fabric WSPL FQDN #2376

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

feat: Support Choosing Fabric WSPL FQDN #2376

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3822ba9
support use fabric wspl fqdn
mslhrotk May 20, 2025
51862bf
use scala read
mslhrotk May 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
88 changes: 67 additions & 21 deletions core/src/main/scala/com/microsoft/azure/synapse/ml/fabric/FabricClient.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,44 +3,32 @@

package com.microsoft.azure.synapse.ml.fabric

import spray.json.DefaultJsonProtocol.StringJsonFormat
import spray.json.JsValue
import spray.json.DefaultJsonProtocol.{StringJsonFormat, mapFormat}
import spray.json._

import java.net.{MalformedURLException, URL}
import java.util.UUID
import scala.io.Source

object FabricClient extends RESTUtils {
private val PbiGlobalServiceEndpoints = Map(
"public" -> "https://api.powerbi.com/",
"fairfax" -> "https://api.powerbigov.us",
"mooncake" -> "https://api.powerbi.cn",
"blackforest" -> "https://app.powerbi.de",
"msit" -> "https://api.powerbi.com/",
"prod" -> "https://api.powerbi.com/",
"int3" -> "https://biazure-int-edog-redirect.analysis-df.windows.net/",
"dxt" -> "https://powerbistagingapi.analysis.windows.net/",
"edog" -> "https://biazure-int-edog-redirect.analysis-df.windows.net/",
"dev" -> "https://onebox-redirect.analysis.windows-int.net/",
"console" -> "http://localhost:5001/",
"daily" -> "https://dailyapi.powerbi.com/")

private val WorkloadEndpointTypeML = "ML";
private val WorkloadEndpointTypeLLMPlugin = "LlmPlugin"
private val WorkloadEndpointTypeAutomatic = "Automatic"
private val WorkloadEndpointTypeRegistry = "Registry"
private val WorkloadEndpointTypeAdmin = "MLAdmin"
private val ContextFilePath = "/home/trusted-service-user/.trident-context";
private val SparkConfPath = "/opt/spark/conf/spark-defaults.conf";
private val ClusterInfoPath = "/opt/health-agent/conf/cluster-info.json";

lazy val CapacityID: Option[String] = getCapacityID;
lazy val WorkspaceID: Option[String] = getWorkspaceID;
lazy val ArtifactID: Option[String] = getArtifactID;
lazy val PbiEnv: String = getPbiEnv;
lazy val FabricContext: Map[String, String] = getFabricContextFile;
lazy val MLWorkloadHost: Option[String] = getMLWorkloadHost;
lazy val WorkspacePeEnabled: Boolean = getWorkspacePeEnabled;

lazy val PbiSharedHost: String = getPbiSharedHost;
lazy val PbiSharedHost: Option[String] = getPbiSharedHost;
lazy val MLWorkloadEndpointML: String = getMLWorkloadEndpoint(WorkloadEndpointTypeML);
lazy val MLWorkloadEndpointLLMPlugin: String = getMLWorkloadEndpoint(WorkloadEndpointTypeLLMPlugin);
lazy val MLWorkloadEndpointAutomatic: String = getMLWorkloadEndpoint(WorkloadEndpointTypeAutomatic);
Expand Down Expand Up @@ -81,9 +69,23 @@ object FabricClient extends RESTUtils {
}

private def getMLWorkloadHost: Option[String] = {
if (WorkspacePeEnabled) {
getMLWorkloadPEHost
}
extractSchemeAndHost(FabricContext.get("trident.lakehouse.tokenservice.endpoint"))
}

private def getMLWorkloadPEHost: Option[String] = {
WorkspaceID.map { wsId =>
val cleanedWsId = wsId.toLowerCase.replace("-", "")
val envMark = PbiEnv match {
case "daily" | "dxt" | "msit" => s"$PbiEnv-"
case _ => ""
}
s"https://${cleanedWsId}.z${cleanedWsId.take(2)}.${envMark}c.fabric.microsoft.com"
}
}

private def readFabricContextFile(): Map[String, String] = {
val source = Source.fromFile(ContextFilePath)
try {
Expand Down Expand Up @@ -120,6 +122,25 @@ object FabricClient extends RESTUtils {
}
}

private def readClusterMetadata(): Map[String, String] = {
val source = Source.fromFile(ClusterInfoPath)
try {
val jsonString = try source.mkString finally source.close()
val jsValue = jsonString.parseJson
val clusterMetadataJson = jsValue.asJsObject.fields("cluster_metadata")
clusterMetadataJson.convertTo[Map[String, String]]
} catch {
case _: Exception => Map.empty[String, String]
} finally {
source.close()
}
}

private def getWorkspacePeEnabled: Boolean = {
val metadata = readClusterMetadata()
metadata.get("workspace-pe-enabled").exists(_.equalsIgnoreCase("true"))
}

private def getHeaders: Map[String, String] = {
Map(
"Authorization" -> s"${getMLWorkloadAADAuthHeader}",
Expand All @@ -129,10 +150,35 @@ object FabricClient extends RESTUtils {
)
}

private def getPbiSharedHost: String = {
val clusterDetailUrl = s"${PbiGlobalServiceEndpoints(PbiEnv)}powerbi/globalservice/v201606/clusterDetails";
val headers = getHeaders;
usageGet(clusterDetailUrl, headers).asJsObject.fields("clusterUrl").convertTo[String];
private def getPbiSharedHost: Option[String] = {
if (WorkspacePeEnabled) {
getPEPbiSharedHost
}
val endpoint = FabricContext.get("spark.trident.pbiHost") match {
case Some(value) if value.nonEmpty =>
value.replace("https://", "").replace("http://", "")
case _ =>
PbiEnv match {
case "edog" => "powerbiapi.analysis-df.windows.net"
case "daily" => "dailyapi.fabric.microsoft.com"
case "dxt" => "dxtapi.fabric.microsoft.com"
case "msit" => "msitapi.fabric.microsoft.com"
case _ => "api.fabric.microsoft.com"
}
}

Some("https://" + endpoint)
}

private def getPEPbiSharedHost: Option[String] = {
WorkspaceID.map { wsId =>
val cleanedWsId = wsId.toLowerCase.replace("-", "")
val envMark = PbiEnv match {
case "daily" | "dxt" | "msit" => PbiEnv
case _ => ""
}
s"https://${cleanedWsId}.z${cleanedWsId.take(2)}.w.${envMark}api.fabric.microsoft.com"
}
}

private def getMLWorkloadEndpoint(endpointType: String): String = {
Expand Down
Loading