如何从 powershell 脚本重新启动 azure vm 并继续其余步骤

Question

I have an azure VM . Am writing a powershell script which executes two .bat files after that it has to reboot the azure VM after that it has to continue to execute remaining .bat files.

Here's the below script

Step1: set-location -path C:/temp

Step2: start-process cmd.exe /c "abc.bat" -wait

Step3: start-process cmd.exe /c "123.bat" -wait

Step4: restart-computer -Force

Step5: start-process cmd.exe /c "prod.bat" -wait

Step6: start-process cmd.exe /c "dev.bat" -wait

Am executing this script remotely by giving below command

PS C:\> Invoke-AzVMRunCommand -ResourceGroupName 'rgname' -VMName 'vmname' -CommandId 'RunPowerShellScript' -ScriptPath '/Users/username/sample.ps1'

After running the above command its executing fine up to "step 4" .After that its not continuing with the remaining steps

Can you please help me on this ?

Answer 1

There are two approaches that I can think of and detailing them below :

Approach 1 :

Using the registry key RUNONCE - which can be used to specify commands that the system will execute one time and then delete.

More information available here .

There will be slight modification in when running it remotely.

When you use the Invoke-AzVMRunCommand , the initial command will have to copy the 2 script files to a temporary location in the remote VM and kick start the first script (this contains steps 1-3)

And before step 4 - you will create a registry key entry under RUNONCE :

HKEY_LOCAL_MACHINE\Software\Microsoft\Windows\CurrentVersion\RunOnce

set-location HKLM:\Software\Microsoft\Windows\CurrentVersion\RunOnce
new-itemproperty . MyKey -propertytype String -value "Powershell c:\temp\script2.ps1"

Or Alternatively you can directly add the .bat files under the RunOnce keys.

The file gets under RUNONCE keys gets executed after login and gets deleted after the execution.

Approach 2 :

You could refer to the below article which creates workflow and also creates ScheduledTask to resume the suspended workflow.

https://devblogs.microsoft.com/scripting/powershell-workflows-restarting-the-computer/#automatic-restart-of-workflow

workflow test-restart {

 #STEPS 1-3

 Restart-Computer -Wait

 #STEPS 5 - 6
} 

For scheduling :

$act = New-ScheduledTaskAction -Execute $pstart -Argument $actionscript
$trig = New-ScheduledTaskTrigger -AtLogOn

NOTE : It is not mandatory to use workflows like in the example. You can write your own logic to resume at the point where you had left once it has been re-triggered - by persisting the step data.

The key to this approach is to create a scheduled task which gets triggered to continue the script automatically after a restart.

Even in this case, it may be necessary to copy the scripts locally to the machine.