Spark-työmääritelmän luominen ja päivittäminen Microsoft Fabric Rest -ohjelmointirajapinnan avulla
Microsoft Fabric Rest -ohjelmointirajapinta tarjoaa palvelun päätepisteen Fabric-kohteiden CRUD-toiminnoille. Tässä opetusohjelmassa käydään läpi, miten Spark-työn määritelmän artefakti luodaan ja päivitetään. Siihen liittyy kolme korkean tason vaihetta:
- luo Spark Job Definition -kohde, jolla on alkuperäinen tila
- ladata päämääritelmätiedoston ja muut lib-tiedostot
- päivitä Spark-työn määrityskohde päämääritystiedoston OneLake-URL-osoitteella ja muilla lib-tiedostoilla
Edellytykset
- Fabric REST -ohjelmointirajapinnan käyttämiseen vaaditaan Microsoft Entra -tunnus. Tunnuksen noutamista suositellaan MSAL-kirjastolle. Lisätietoja on kohdassa Todennustyönkulun tuki MSAL:ssä.
- OneLake-ohjelmointirajapinnan käyttämiseen vaaditaan tallennustunnus. Lisätietoja on kohdassa MSAL for Python.
Spark-työn määritelmäkohteen luominen alkuperäisen tilan mukaan
Microsoft Fabric Rest -ohjelmointirajapinta määrittää yhdistetyn päätepisteen Fabric-kohteiden CRUD-toiminnoille. Päätepiste on https://api.fabric.microsoft.com/v1/workspaces/{workspaceId}/items
.
Kohteen tiedot on määritetty pyynnön leipätekstissä. Tässä on esimerkki pyynnön leipätekstistä Spark-työn määrityskohteen luomista varten:
{
"displayName": "SJDHelloWorld",
"type": "SparkJobDefinition",
"definition": {
"format": "SparkJobDefinitionV1",
"parts": [
{
"path": "SparkJobDefinitionV1.json",
"payload":"eyJleGVjdXRhYmxlRmlsZSI6bnVsbCwiZGVmYXVsdExha2Vob3VzZUFydGlmYWN0SWQiOiIiLCJtYWluQ2xhc3MiOiIiLCJhZGRpdGlvbmFsTGFrZWhvdXNlSWRzIjpbXSwicmV0cnlQb2xpY3kiOm51bGwsImNvbW1hbmRMaW5lQXJndW1lbnRzIjoiIiwiYWRkaXRpb25hbExpYnJhcnlVcmlzIjpbXSwibGFuZ3VhZ2UiOiIiLCJlbnZpcm9ubWVudEFydGlmYWN0SWQiOm51bGx9",
"payloadType": "InlineBase64"
}
]
}
}
Tässä esimerkissä Spark-työn määritelmä -kohteen nimeksi SJDHelloWorld
annetaan . Kenttä payload
on tietojen määrityksen Base64-koodattu sisältö, jonka sisältö on poistettu koodauksen jälkeen:
{
"executableFile":null,
"defaultLakehouseArtifactId":"",
"mainClass":"",
"additionalLakehouseIds":[],
"retryPolicy":null,
"commandLineArguments":"",
"additionalLibraryUris":[],
"language":"",
"environmentArtifactId":null
}
Seuraavassa on kaksi apufunktiota, joiden avulla voit koodata ja purkaa yksityiskohtaisen määrityksen:
import base64
def json_to_base64(json_data):
# Serialize the JSON data to a string
json_string = json.dumps(json_data)
# Encode the JSON string as bytes
json_bytes = json_string.encode('utf-8')
# Encode the bytes as Base64
base64_encoded = base64.b64encode(json_bytes).decode('utf-8')
return base64_encoded
def base64_to_json(base64_data):
# Decode the Base64-encoded string to bytes
base64_bytes = base64_data.encode('utf-8')
# Decode the bytes to a JSON string
json_string = base64.b64decode(base64_bytes).decode('utf-8')
# Deserialize the JSON string to a Python dictionary
json_data = json.loads(json_string)
return json_data
Tässä on koodikatkelman, jolla voit luoda Spark-työn määrityskohteen:
import requests
bearerToken = "breadcrumb"; # replace this token with the real AAD token
headers = {
"Authorization": f"Bearer {bearerToken}",
"Content-Type": "application/json" # Set the content type based on your request
}
payload = "eyJleGVjdXRhYmxlRmlsZSI6bnVsbCwiZGVmYXVsdExha2Vob3VzZUFydGlmYWN0SWQiOiIiLCJtYWluQ2xhc3MiOiIiLCJhZGRpdGlvbmFsTGFrZWhvdXNlSWRzIjpbXSwicmV0cnlQb2xpY3kiOm51bGwsImNvbW1hbmRMaW5lQXJndW1lbnRzIjoiIiwiYWRkaXRpb25hbExpYnJhcnlVcmlzIjpbXSwibGFuZ3VhZ2UiOiIiLCJlbnZpcm9ubWVudEFydGlmYWN0SWQiOm51bGx9"
# Define the payload data for the POST request
payload_data = {
"displayName": "SJDHelloWorld",
"Type": "SparkJobDefinition",
"definition": {
"format": "SparkJobDefinitionV1",
"parts": [
{
"path": "SparkJobDefinitionV1.json",
"payload": payload,
"payloadType": "InlineBase64"
}
]
}
}
# Make the POST request with Bearer authentication
sjdCreateUrl = f"https://api.fabric.microsoft.com//v1/workspaces/{workspaceId}/items"
response = requests.post(sjdCreateUrl, json=payload_data, headers=headers)
Päämääritystiedoston ja muiden lib-tiedostojen lataaminen palvelimeen
Tallennustunnus vaaditaan tiedoston lataamiseksi OneLakeen. Seuraavassa on apufunktio tallennustunnuksen hakemiseen:
import msal
def getOnelakeStorageToken():
app = msal.PublicClientApplication(
"{client id}", # this filed should be the client id
authority="https://login.microsoftonline.com/microsoft.com")
result = app.acquire_token_interactive(scopes=["https://storage.azure.com/.default"])
print(f"Successfully acquired AAD token with storage audience:{result['access_token']}")
return result['access_token']
Nyt olemme luoneet Spark Job Definition -kohteen, jotta se voidaan suorittaa. Meidän on määritettävä päämääritystiedosto ja tarvittavat ominaisuudet. Tämän SJD-kohteen tiedoston lataamisen päätepiste on https://onelake.dfs.fabric.microsoft.com/{workspaceId}/{sjdartifactid}
. Samaa edellisessä vaiheessa käytettyä "workspaceId"-arvoa "sjdartifactid" voidaan löytää edellisen vaiheen vastausten rungosta. Tässä on koodikatkelman, jolla määritetään päämääritystiedosto:
import requests
# three steps are required: create file, append file, flush file
onelakeEndPoint = "https://onelake.dfs.fabric.microsoft.com/workspaceId/sjdartifactid"; # replace the id of workspace and artifact with the right one
mainExecutableFile = "main.py"; # the name of the main executable file
mainSubFolder = "Main"; # the sub folder name of the main executable file. Don't change this value
onelakeRequestMainFileCreateUrl = f"{onelakeEndPoint}/{mainSubFolder}/{mainExecutableFile}?resource=file" # the url for creating the main executable file via the 'file' resource type
onelakePutRequestHeaders = {
"Authorization": f"Bearer {onelakeStorageToken}", # the storage token can be achieved from the helper function above
}
onelakeCreateMainFileResponse = requests.put(onelakeRequestMainFileCreateUrl, headers=onelakePutRequestHeaders)
if onelakeCreateMainFileResponse.status_code == 201:
# Request was successful
print(f"Main File '{mainExecutableFile}' was successfully created in onelake.")
# with previous step, the main executable file is created in OneLake, now we need to append the content of the main executable file
appendPosition = 0;
appendAction = "append";
### Main File Append.
mainExecutableFileSizeInBytes = 83; # the size of the main executable file in bytes
onelakeRequestMainFileAppendUrl = f"{onelakeEndPoint}/{mainSubFolder}/{mainExecutableFile}?position={appendPosition}&action={appendAction}";
mainFileContents = "filename = 'Files/' + Constant.filename; tablename = 'Tables/' + Constant.tablename"; # the content of the main executable file, please replace this with the real content of the main executable file
mainExecutableFileSizeInBytes = 83; # the size of the main executable file in bytes, this value should match the size of the mainFileContents
onelakePatchRequestHeaders = {
"Authorization": f"Bearer {onelakeStorageToken}",
"Content-Type" : "text/plain"
}
onelakeAppendMainFileResponse = requests.patch(onelakeRequestMainFileAppendUrl, data = mainFileContents, headers=onelakePatchRequestHeaders)
if onelakeAppendMainFileResponse.status_code == 202:
# Request was successful
print(f"Successfully Accepted Main File '{mainExecutableFile}' append data.")
# with previous step, the content of the main executable file is appended to the file in OneLake, now we need to flush the file
flushAction = "flush";
### Main File flush
onelakeRequestMainFileFlushUrl = f"{onelakeEndPoint}/{mainSubFolder}/{mainExecutableFile}?position={mainExecutableFileSizeInBytes}&action={flushAction}"
print(onelakeRequestMainFileFlushUrl)
onelakeFlushMainFileResponse = requests.patch(onelakeRequestMainFileFlushUrl, headers=onelakePatchRequestHeaders)
if onelakeFlushMainFileResponse.status_code == 200:
print(f"Successfully Flushed Main File '{mainExecutableFile}' contents.")
else:
print(onelakeFlushMainFileResponse.json())
Seuraa samaa prosessia ladataksesi muut lib-tiedostot tarvittaessa.
Päivitä Spark-työn määrityskohde päämääritystiedoston OneLake-URL-osoitteella ja muilla lib-tiedostoilla
Tähän asti olemme luoneet Spark Job Definition -kohteen, jolla on alkuperäinen tila, ladannut päämääritystiedoston ja muut lib-tiedostot. Viimeinen vaihe on päivittää Spark-työn määrityskohde päämääritystiedoston ja muiden lib-tiedostojen URL-ominaisuuksien määrittämiseksi. Spark-työn määrityskohteen päivittämisen päätepiste on https://api.fabric.microsoft.com/v1/workspaces/{workspaceId}/items/{sjdartifactid}
. Edellisten vaiheiden yhteydessä tulee käyttää samaa "workspaceId"- ja "sjdartifactid"-arvoa. Tässä on koodikatkelman, joka päivittää Spark-työn määritelmän kohteen:
mainAbfssPath = f"abfss://{workspaceId}@onelake.dfs.fabric.microsoft.com/{sjdartifactid}/Main/{mainExecutableFile}" # the workspaceId and sjdartifactid are the same as previous steps, the mainExecutableFile is the name of the main executable file
libsAbfssPath = f"abfss://{workspaceId}@onelake.dfs.fabric.microsoft.com/{sjdartifactid}/Libs/{libsFile}" # the workspaceId and sjdartifactid are the same as previous steps, the libsFile is the name of the libs file
defaultLakehouseId = 'defaultLakehouseid'; # replace this with the real default lakehouse id
updateRequestBodyJson = {
"executableFile":mainAbfssPath,
"defaultLakehouseArtifactId":defaultLakehouseId,
"mainClass":"",
"additionalLakehouseIds":[],
"retryPolicy":None,
"commandLineArguments":"",
"additionalLibraryUris":[libsAbfssPath],
"language":"Python",
"environmentArtifactId":None}
# Encode the bytes as a Base64-encoded string
base64EncodedUpdateSJDPayload = json_to_base64(updateRequestBodyJson)
# Print the Base64-encoded string
print("Base64-encoded JSON payload for SJD Update:")
print(base64EncodedUpdateSJDPayload)
# Define the API URL
updateSjdUrl = f"https://api.fabric.microsoft.com//v1/workspaces/{workspaceId}/items/{sjdartifactid}/updateDefinition"
updatePayload = base64EncodedUpdateSJDPayload
payloadType = "InlineBase64"
path = "SparkJobDefinitionV1.json"
format = "SparkJobDefinitionV1"
Type = "SparkJobDefinition"
# Define the headers with Bearer authentication
bearerToken = "breadcrumb"; # replace this token with the real AAD token
headers = {
"Authorization": f"Bearer {bearerToken}",
"Content-Type": "application/json" # Set the content type based on your request
}
# Define the payload data for the POST request
payload_data = {
"displayName": "sjdCreateTest11",
"Type": Type,
"definition": {
"format": format,
"parts": [
{
"path": path,
"payload": updatePayload,
"payloadType": payloadType
}
]
}
}
# Make the POST request with Bearer authentication
response = requests.post(updateSjdUrl, json=payload_data, headers=headers)
if response.status_code == 200:
print("Successfully updated SJD.")
else:
print(response.json())
print(response.status_code)
Spark Job Definition -kohteen luomiseen ja päivittämiseen tarvitaan sekä Fabric REST -ohjelmointirajapinta että OneLake-ohjelmointirajapinta, jotta voit tiivistää koko prosessin. Fabric REST -ohjelmointirajapinnan avulla luodaan ja päivitetään Spark Job Definition -kohde, OneLake-ohjelmointirajapintaa käytetään päämääritelmätiedoston ja muiden lib-tiedostojen lataamiseen. Päämääritelmätiedosto ja muut lib-tiedostot ladataan ensin OneLakeen. Tämän jälkeen päämääritystiedoston ja muiden lib-tiedostojen URL-ominaisuudet määritetään Spark Job Definition -kohteessa.