retry once if inner is null for Disk (#1390)

Azure · Apr 16, 2021 · 4731448 · 4731448
1 parent cc03437
commit 4731448
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/...compute/src/main/java/com/microsoft/azure/management/compute/implementation/DiskImpl.java b/...compute/src/main/java/com/microsoft/azure/management/compute/implementation/DiskImpl.java
@@ -376,6 +376,28 @@ public DiskImpl withAvailabilityZone(AvailabilityZoneId zoneId) {
 
     @Override
     public Observable<Disk> createResourceAsync() {
+        final String resourceGroupName = resourceGroupName();
+        final String resourceName = name();
+        final DiskInner createParameters = this.inner();
+        final DiskImpl self = this;
+        return manager().inner().disks().createOrUpdateAsync(resourceGroupName, resourceName, createParameters)
+                .flatMap(new Func1<DiskInner, Observable<Disk>>() {
+                    @Override
+                    public Observable<Disk> call(DiskInner diskInner) {
+                        if (diskInner == null || diskInner.id() == null) {
+                            // retry once
+                            return manager().inner().disks()
+                                    .createOrUpdateAsync(resourceGroupName, resourceName, createParameters)
+                                    .map(innerToFluentMap(self));
+                        } else {
+                            return Observable.just(innerToFluentMap(self).call(diskInner));
+                        }
+                    }
+                });
+    }
+
+    @Override
+    public Observable<Disk> updateResourceAsync() {
         return manager().inner().disks().createOrUpdateAsync(resourceGroupName(), name(), this.inner())
                 .map(innerToFluentMap(this));
     }