[Placement Group] Add support for soft anti affinity

pkg/cluster/compute/affinityrule/affinityrule.go

@@ -4,6 +4,7 @@
 package affinityrule
 
 import (
+	"strings"
 	"time"
 
 	"github.com/microsoft/wmi/pkg/base/host"
@@ -97,3 +98,45 @@ func GetAffinityRule(whost *host.WmiHost, affinityRuleName string) (caffinityRul
 	caffinityRule = &AffinityRule{wmiafRule}
 	return
 }
+
+// SetAffinityRule updates an existing affinity rule
+func SetAffinityRule(whost *host.WmiHost, name string, ruleType int, softAntiAffinity bool) (err error) {
+	affinityRule, err := GetAffinityRule(whost, name)
+	if err != nil {
+		return
+	}
+	defer affinityRule.Close()
+
+	propeties := affinityRule.GetClass().GetPropertiesNames()
+	if len(propeties) == 0 {
+		err = errors.Wrapf(errors.Unknown, "Failed to get the properties of MSCluster_AffinityRule")
+		return
+	}
+
+	softAntiAffinitySupported := false
+	for _, property := range propeties {
+		if strings.EqualFold(property, "SoftAntiAffinity") {
+			softAntiAffinitySupported = true
+			break
+		}
+	}
+
+	// SoftAntiAffinity is supported on platforms 24H2 or higher
+	if softAntiAffinitySupported {
+		softAntiAffinityValue := 0
+		if softAntiAffinity {
+			softAntiAffinityValue = 1
+		}
+		_, err = affinityRule.InvokeMethod("SetAffinityRule", int(ruleType), 1 /* Enabled */, softAntiAffinityValue /* SoftAntiAffinity */)
+		if err != nil {
+			return
+		}
+	} else {
+		_, err = affinityRule.SetAffinityRule(uint32(ruleType), 1 /* Enabled */)
+		if err != nil {
+			return
+		}
+	}
+
+	return
+}

pkg/cluster/storage/clustersharedvolume/clustersharedvolume.go

@@ -8,13 +8,14 @@ import (
 	"path/filepath"
 	"strings"
 
+	"reflect"
+
 	"github.com/microsoft/wmi/pkg/base/host"
 	"github.com/microsoft/wmi/pkg/base/instance"
 	"github.com/microsoft/wmi/pkg/base/query"
 	fcconstant "github.com/microsoft/wmi/pkg/cluster/constant"
 	"github.com/microsoft/wmi/pkg/constant"
 	"github.com/microsoft/wmi/pkg/errors"
-	"reflect"
 
 	"github.com/microsoft/wmi/pkg/cluster/compute/resource"
 	wmi "github.com/microsoft/wmi/pkg/wmiinstance"
@@ -96,7 +97,9 @@ func GetClusterSharedVolumebyName(whost *host.WmiHost, name string) (cvolume *Cl
 			return
 		}
 		matchingPath := strings.ToLower(filepath.Clean(instanceName))
-		if strings.Contains(inPath, matchingPath) {
+		// Append "\\" to eliminate false positive due to partial match
+		// e.g. "C:\ClusterStorage\Volume10\image" and "C:\ClusterStorage\Volume1"
+		if strings.HasPrefix(inPath+"\\", matchingPath+"\\") {
 			tmpInstance, err1 := instance.Clone()
 			if err1 != nil {
 				err = err1