From 8783fbedc599a8a977f345f07a1b3763dd5b15b0 Mon Sep 17 00:00:00 2001
From: Mathieu Tortuyaux <mtortuyaux@microsoft.com>
Date: Wed, 24 Jan 2024 11:51:39 +0100
Subject: [PATCH] api/azure: inject availability set

Signed-off-by: Mathieu Tortuyaux <mtortuyaux@microsoft.com>
---
 platform/api/azure/instance.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/platform/api/azure/instance.go b/platform/api/azure/instance.go
index cfbfa4c7f..e0ee2a903 100644
--- a/platform/api/azure/instance.go
+++ b/platform/api/azure/instance.go
@@ -103,6 +103,9 @@ func (a *API) getVMParameters(name, userdata, sshkey, storageAccountURI string,
 		},
 		Plan: plan,
 		VirtualMachineProperties: &compute.VirtualMachineProperties{
+			AvailabilitySet: &compute.SubResource{
+				ID: &a.Opts.AvailabilitySetID,
+			},
 			HardwareProfile: &compute.HardwareProfile{
 				VMSize: compute.VirtualMachineSizeTypes(a.Opts.Size),
 			},