[moc] Add AvailabilitySet Operation for Add/Remove VM

.pipelines/build.yaml

@@ -79,7 +79,7 @@ jobs:
   steps:
   - task: GoTool@0
     inputs:
-      version: '1.22.5'
+      version: '1.24.3'
 
   - task: InstallSSHKey@0
     inputs:

Makefile

@@ -25,7 +25,7 @@ format:
 	gofmt -s -w rpc/ pkg/ 
 
 bootstrap:
-	GOOS="linux" go get -u google.golang.org/grpc@v1.59.0
+	GOOS="linux" go get -u google.golang.org/grpc@v1.72.0
 	GOOS="linux" go install github.com/golang/protobuf/protoc-gen-go@v1.3.2
 
 test: unittest
@@ -55,4 +55,4 @@ mocks:
 
 golangci-lint:
 	$(GOCMD) install github.com/golangci/golangci-lint/cmd/golangci-lint@latest
-	$(GOPATH_BIN)/golangci-lint run --config .golangci.yml
+	$(GOPATH_BIN)/golangci-lint run --config .golangci.yml

go.mod

@@ -1,6 +1,6 @@
 module github.com/microsoft/moc
 
-go 1.23.0
+go 1.24.3
 
 require (
 	github.com/go-logr/logr v1.4.2
@@ -19,10 +19,10 @@ require (
 require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
-	golang.org/x/net v0.39.0 // indirect
-	golang.org/x/sys v0.32.0 // indirect
-	golang.org/x/text v0.24.0 // indirect
-	google.golang.org/genproto/googleapis/rpc v0.0.0-20250428153025-10db94c68c34 // indirect
+	golang.org/x/net v0.41.0 // indirect
+	golang.org/x/sys v0.33.0 // indirect
+	golang.org/x/text v0.26.0 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822 // indirect
 	google.golang.org/protobuf v1.36.6 // indirect
 )
 

go.sum

@@ -69,8 +69,8 @@ golang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
 golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
-golang.org/x/text v0.24.0 h1:dd5Bzh4yt5KYA8f9CJHCP4FB4D51c2c6JvN37xJJkJ0=
-golang.org/x/text v0.24.0/go.mod h1:L8rBsPeo2pSS+xqN0d5u2ikmjtmoJbDBT1b7nHvFCdU=
+golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
+golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
 golang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=
@@ -81,8 +81,8 @@ golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxb
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20250428153025-10db94c68c34 h1:h6p3mQqrmT1XkHVTfzLdNz1u7IhINeZkz67/xTbOuWs=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20250428153025-10db94c68c34/go.mod h1:qQ0YXyHHx3XkvlzUtpXDkS29lDSafHMZBAZDc03LQ3A=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822 h1:fc6jSaCT0vBduLYZHYrBBNY4dsWuvgyff9noRNDdBeE=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20250603155806-513f23925822/go.mod h1:qQ0YXyHHx3XkvlzUtpXDkS29lDSafHMZBAZDc03LQ3A=
 google.golang.org/grpc v1.72.0 h1:S7UkcVa60b5AAQTaO6ZKamFp1zMZSU0fGDK2WZLbBnM=
 google.golang.org/grpc v1.72.0/go.mod h1:wH5Aktxcg25y1I3w7H69nHfXdOG3UiadoBtjh3izSDM=
 google.golang.org/protobuf v1.36.6 h1:z1NpPI8ku2WgiWnf+t9wTPsn6eP1L7ksHUlkfLvd9xY=

rpc/common/moc_common_computecommon.proto

@@ -258,3 +258,8 @@ message ProxyConfiguration {
 	repeated string noProxy = 3;
 	string trustedCa = 4 [(sensitivecompute) = true];
 }
+
+enum AvailabilitySetOperation {
+	ADD_VM = 0;
+	REMOVE_VM = 1;
+}

rpc/nodeagent/compute/availabilityset/moc_nodeagent_availabilityset.proto

@@ -7,6 +7,8 @@ package moc.nodeagent.compute;
 
 import "google/protobuf/wrappers.proto";
 import "moc_common_common.proto";
+import "moc_common_computecommon.proto";
+
 
 message AvailabilitySetRequest {
 	repeated AvailabilitySet AvailabilitySets = 1;
@@ -32,9 +34,22 @@ message AvailabilitySet {
     Tags tags = 4;
     Entity entity = 5;
     int32 platformFaultDomainCount = 6;
-    repeated VirtualMachineReference virtualMachines = 7;
+    repeated VirtualMachineReference virtualMachines = 7; // TODO(pgayam): figure how to mark this field obselete
+}
+
+message AvailabilitySetOperationRequest {
+	string AvailabilitySet = 1;
+    string VirtualMachine = 2;
+	AvailabilitySetOperation OperationType = 3;
+}
+
+message AvailabilitySetOperationResponse {
+	google.protobuf.BoolValue Result = 2;
+	string  Error = 3;
 }
 
+
 service AvailabilitySetAgent {
 	rpc Invoke(AvailabilitySetRequest) returns (AvailabilitySetResponse) {}
+    rpc Operate(AvailabilitySetOperationRequest) returns (AvailabilitySetOperationResponse) {}
 }