Add script

Author: Xiaoyu Wang

.gitignore

@@ -0,0 +1 @@
+SHIR/IntegrationRuntime*.msi

Dockerfile

@@ -0,0 +1,12 @@
+FROM mcr.microsoft.com/windows/servercore:ltsc2019
+
+# Download the latest self-hosted integration runtime installer into the SHIR folder
+COPY SHIR C:/SHIR/
+
+RUN ["powershell", "C:/SHIR/build.ps1"]
+
+CMD ["powershell", "C:/SHIR/setup.ps1"]
+
+ENV SHIR_WINDOWS_CONTAINER_ENV True
+
+HEALTHCHECK --start-period=120s CMD ["powershell", "C:/SHIR/health-check.ps1"]

README.md

@@ -1,3 +1,9 @@
+Azure Data Factory Integration Runtime in Windows Container Sample
+=======
+This repo contains the sample for running the Azure Data Factory Integration Runtime in Windows Container
+
+For more information about Azure Data Factory, see [https://docs.microsoft.com/en-us/azure/data-factory/concepts-integration-runtime](https://docs.microsoft.com/en-us/azure/data-factory/concepts-integration-runtime)
+
 
 # Contributing
 

SHIR/build.ps1

@@ -0,0 +1,28 @@
+$DmgcmdPath = "C:\Program Files\Microsoft Integration Runtime\4.0\Shared\dmgcmd.exe"
+
+function Write-Log($Message) {
+    function TS { Get-Date -Format 'MM/dd/yyyy hh:mm:ss' }
+    Write-Host "[$(TS)] $Message"
+}
+
+function Install-SHIR() {
+    Write-Log "Install the Self-hosted Integration Runtime in the Windows container"
+
+    $MsiFileName = (Get-ChildItem -Path C:\SHIR | Where-Object { $_.Name -match [regex] "IntegrationRuntime_.*.msi" })[0].Name
+    Start-Process msiexec.exe -Wait -ArgumentList "/i C:\SHIR\$MsiFileName /qn"
+    if (!$?) {
+        Write-Log "SHIR MSI Install Failed"
+    }
+
+    Write-Log "SHIR MSI Install Successfully"
+}
+
+function SetupEnv() {
+    Write-Log "Begin to Setup the SHIR Environment"
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-Stop -StopUpgradeService -TurnOffAutoUpdate"
+    Write-Log "SHIR Environment Setup Successfully"
+}
+
+Install-SHIR
+
+exit 0

SHIR/health-check.ps1

@@ -0,0 +1,18 @@
+$DmgcmdPath = "C:\Program Files\Microsoft Integration Runtime\4.0\Shared\dmgcmd.exe"
+
+function Check-Node-Connection() {
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-cgc" -RedirectStandardOutput "C:\SHIR\status-check.txt"
+    $ConnectionResult = Get-Content "C:\SHIR\status-check.txt"
+    Remove-Item -Force "C:\SHIR\status-check.txt"
+
+    if ($ConnectionResult -like "Connected") {
+        return $TRUE
+    }
+    else {
+        exit 1
+    }
+}
+
+if (Check-Node-Connection) {   
+    exit 0
+}

SHIR/setup.ps1

@@ -0,0 +1,115 @@
+$DmgcmdPath = "C:\Program Files\Microsoft Integration Runtime\4.0\Shared\dmgcmd.exe"
+
+function Write-Log($Message) {
+    function TS { Get-Date -Format 'MM/dd/yyyy hh:mm:ss' }
+    Write-Host "[$(TS)] $Message"
+}
+
+function Check-Is-Registered() {
+    $result = Get-ItemProperty -Path 'HKLM:\SOFTWARE\Microsoft\DataTransfer\DataManagementGateway\ConfigurationManager' -Name HaveRun -ErrorAction SilentlyContinue
+    if (($result -ne $null) -and ($result.HaveRun -eq 'Mdw')) {
+        return $TRUE
+    }
+    return $FALSE
+}
+
+function Check-Main-Process() {
+    $ProcessResult = Get-WmiObject Win32_Process -Filter "name = 'diahost.exe'"
+    
+    if ($ProcessResult) {
+        return $TRUE
+    }
+    else {
+        throw "Main Process not found"   
+    }
+}
+
+function Check-Node-Connection() {
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-cgc" -RedirectStandardOutput "C:\SHIR\status-check.txt"
+    $ConnectionResult = Get-Content "C:\SHIR\status-check.txt"
+    Remove-Item -Force "C:\SHIR\status-check.txt"
+
+    if ($ConnectionResult -like "Connected") {
+        return $TRUE
+    }
+    else {
+        throw "Node is offline"    
+    }
+}
+
+
+function RegisterNewNode {
+    Param(
+        $AUTH_KEY,
+        $NODE_NAME,
+        $ENABLE_HA,
+        $HA_PORT
+    )
+
+    Write-Log "Start registering the new SHIR node"
+
+    if (!$NODE_NAME) {
+        Start-Process $DmgcmdPath -Wait -ArgumentList "-RegisterNewNode", "$($AUTH_KEY)" -RedirectStandardOutput "C:\SHIR\register-out.txt" -RedirectStandardError "C:\SHIR\register-error.txt"
+    } else {
+        Start-Process $DmgcmdPath -Wait -ArgumentList "-RegisterNewNode", "$($AUTH_KEY)", "$($NODE_NAME)" -RedirectStandardOutput "C:\SHIR\register-out.txt" -RedirectStandardError "C:\SHIR\register-error.txt"
+    }
+
+    if ($ENABLE_HA -eq "true") {
+        Write-Log "Enable High Availability"
+        $PORT = $HA_PORT -or "8060"
+        Start-Process $DmgcmdPath -Wait -ArgumentList "-EnableRemoteAccess", "$($PORT)"
+    }
+
+    $StdOutResult = Get-Content "C:\SHIR\register-out.txt"
+    $StdErrResult = Get-Content "C:\SHIR\register-error.txt"
+
+
+    if ($StdOutResult)
+    {
+        Write-Log "Registration output:"
+        $StdOutResult | ForEach-Object { Write-Log $_ }
+    }
+
+    if ($StdErrResult)
+    {
+        Write-Log "Registration errors:"
+        $StdErrResult | ForEach-Object { Write-Log $_ }
+    }
+}
+
+# Register SHIR with key from Env Variable: AUTH_KEY
+if (Check-Is-Registered) {
+    Write-Log "Restart the existing node"
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-Start"
+} elseif (Test-Path Env:AUTH_KEY) {
+    Write-Log "Registering SHIR with the node key: $((Get-Item Env:AUTH_KEY).Value)"
+    Write-Log "Registering SHIR with the node name: $((Get-Item Env:NODE_NAME).Value)"
+    Write-Log "Registering SHIR with the enable high availability flag: $((Get-Item Env:ENABLE_HA).Value)"
+    Write-Log "Registering SHIR with the tcp port: $((Get-Item Env:HA_PORT).Value)"
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-Start"
+    RegisterNewNode (Get-Item Env:AUTH_KEY).Value (Get-Item Env:NODE_NAME).Value (Get-Item Env:ENABLE_HA).Value (Get-Item Env:HA_PORT).Value
+} else {
+    Write-Log "Invalid AUTH_KEY Value"
+    exit 1
+}
+
+Write-Log "Waiting 30 seconds waiting for connecting"
+Start-Sleep -Seconds 30
+
+try {
+    while ($TRUE) {
+        if ((Check-Main-Process) -and (Check-Node-Connection)) {   
+            Write-Log "Node Health Check Pass"
+            Start-Sleep -Seconds 60
+            continue
+        }
+    }
+}
+finally {
+    Write-Log "Stop the node connection"
+    Start-Process $DmgcmdPath -Wait -ArgumentList "-Stop"
+    Write-Log "Stop the node connection successfully"
+    exit 0
+}
+
+exit 1