include postCreateOrUpdateHook

azure/services/natgateways/natgateways.go

@@ -17,6 +17,8 @@ limitations under the License.
 package natgateways
 
 import (
+	"context"
+
 	asonetworkv1 "github.com/Azure/azure-service-operator/v2/api/network/v1api20220701"
 	infrav1 "sigs.k8s.io/cluster-api-provider-azure/api/v1beta1"
 	"sigs.k8s.io/cluster-api-provider-azure/azure"
@@ -42,11 +44,22 @@ type Service struct {
 
 // New creates a new service.
 func New(scope NatGatewayScope) *Service {
-	svc := aso.NewService[*asonetworkv1.NatGateway](serviceName, scope)
+	svc := aso.NewService[*asonetworkv1.NatGateway, NatGatewayScope](serviceName, scope)
 	svc.Specs = scope.NatGatewaySpecs()
 	svc.ConditionType = infrav1.NATGatewaysReadyCondition
+	svc.PostCreateOrUpdateResourceHook = func(scope NatGatewayScope, result *asonetworkv1.NatGateway, err error) {
+		if err == nil {
+			scope.SetNatGatewayIDInSubnets(result.Name, *result.Status.Id)
+		}
+	}
 	return &Service{
 		Scope:   scope,
 		Service: svc,
 	}
 }
+
+// IsManaged returns true if the ASO NatGateway was created by CAPZ,
+// meaning that the NATGateway's lifecycle is managed.
+func (s *Service) IsManaged(ctx context.Context, spec azure.ASOResourceSpecGetter[*asonetworkv1.NatGateway]) (bool, error) {
+	return aso.IsManaged(ctx, s.Scope.GetClient(), spec, s.Scope.ClusterName())
+}