Sync WCI compute status to worker deployment version state

Add DescribeWorkerControllerInstanceStatus activity to fetch the WCI validation status and map it to a ComputeStatus. During handleSyncState, invoke syncVersionDataToComputeStatus to update the version state's ComputeStatus with the provider validation result. The sync is gated behind the "sync-compute-validation-status" workflow version for backward compatibility.

Author: 02strich

service/worker/workerdeployment/version_activities.go

@@ -199,6 +199,15 @@ func (a *VersionActivities) GetVersionDrainageStatus(ctx context.Context, versio
 	}, nil
 }
 
+func (a *VersionActivities) DescribeWorkerControllerInstanceStatus(ctx context.Context, version *deploymentspb.WorkerDeploymentVersion) (*deploymentpb.ComputeStatus, error) {
+	resp, _, err := a.WorkerControllerInstanceClient.DescribeWorkerControllerInstance(ctx, a.namespace, &deploymentpb.WorkerDeploymentVersion{DeploymentName: version.GetDeploymentName(), BuildId: version.GetBuildId()})
+	if err != nil {
+		return nil, err
+	}
+
+	return wciValidationStatusToComputeStatus(resp.ValidationStatus), nil
+}
+
 func (a *VersionActivities) UpdateWorkerControllerInstance(ctx context.Context, input *deploymentspb.UpdateWorkerControllerInstanceInput) (*computepb.ComputeConfigSummary, error) {
 	upserts := scalingGroupUpdatesToWCI(input.GetUpsertScalingGroups())
 	resp, err := a.WorkerControllerInstanceClient.UpdateWorkerControllerInstance(ctx, a.namespace, input.GetVersion(), nil, input.GetIdentity(), upserts, input.GetRemoveScalingGroups())

service/worker/workerdeployment/version_workflow.go

@@ -964,6 +964,10 @@ func (d *VersionWorkflowRunner) handleSyncState(ctx workflow.Context, args *depl
 		state.LastDeactivationTime = nil
 	}
 
+	if err := d.syncVersionDataToComputeStatus(ctx); err != nil {
+		return nil, err
+	}
+
 	return &deploymentspb.SyncVersionStateResponse{
 		Summary: versionStateToSummary(state),
 	}, nil
@@ -1244,6 +1248,27 @@ func (d *VersionWorkflowRunner) syncVersionStatusAfterDrainageStatusChange(ctx w
 	return d.syncVersionDataToTaskQueues(ctx, versionData)
 }
 
+// syncVersionDataToComputeStatus is a helper that syncs the compute status from WCI to the worker deployment version
+func (d *VersionWorkflowRunner) syncVersionDataToComputeStatus(ctx workflow.Context) error {
+	if workflow.GetVersion(ctx, "sync-compute-validation-status", workflow.DefaultVersion, 0) == workflow.DefaultVersion {
+		return nil
+	}
+
+	state := d.GetVersionState()
+
+	var result deploymentpb.ComputeStatus
+	resp := workflow.ExecuteActivity(workflow.WithActivityOptions(ctx, defaultActivityOptions), d.a.DescribeWorkerControllerInstanceStatus, state.GetVersion())
+	if err := resp.Get(ctx, &result); err != nil {
+		return err
+	}
+	if result.ProviderValidation == nil {
+		return nil
+	}
+
+	state.ComputeStatus = &result
+	return nil
+}
+
 // syncVersionDataToTaskQueues is a helper that syncs the provided version data to all task queues.
 // This function does NOT acquire the workflow lock - the caller is responsible for that.
 func (d *VersionWorkflowRunner) syncVersionDataToTaskQueues(ctx workflow.Context, versionData *deploymentspb.DeploymentVersionData) error {