GetAsync

huangpf · Oct 26, 2017 · 3eaa6c0 · 3eaa6c0
1 parent 4a41002
commit 3eaa6c0
Show file tree

Hide file tree

Showing 8 changed files with 55 additions and 6 deletions.
diff --git a/experiments/Azure.Experiments/Azure.Experiments/Compute/VirtualMachineParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Compute/VirtualMachineParameters.cs
@@ -18,8 +18,9 @@ public VirtualMachineParameters(
             Ni = ni;
         }
 
-        public override Task<VirtualMachine> GetAsync(Context context)
+        public override Task<VirtualMachine> GetAsync(GetContext context)
             => context
+                .Context
                 .CreateCompute()
                 .VirtualMachines
                 .GetAsync(ResourceGroup.Name, Name);

diff --git a/experiments/Azure.Experiments/Azure.Experiments/GetContext.cs b/experiments/Azure.Experiments/Azure.Experiments/GetContext.cs
@@ -0,0 +1,37 @@
+using System.Collections.Concurrent;
+using System.Threading.Tasks;
+
+namespace Microsoft.Azure.Experiments
+{
+    public sealed class GetContext
+    {
+        public Context Context { get; }
+
+        public GetContext(Context context)
+        {
+            Context = context;
+        }
+
+        public async Task<T> GetOrNullAsync<T>(Parameters<T> parameters)
+        {
+            var result = await Map.GetOrAdd(
+                parameters, _ => GetObjectOrNullAsync(parameters));
+            return (T)result;
+        }
+
+        private async Task<object> GetObjectOrNullAsync<T>(Parameters<T> parameters)
+        {
+            try
+            {
+                return await parameters.GetAsync(this); 
+            }
+            catch
+            {
+                return null;
+            }
+        }
+
+        private ConcurrentDictionary<Parameters, Task<object>> Map { get; }
+            = new ConcurrentDictionary<Parameters, Task<object>>();
+    }
+}
diff --git a/experiments/Azure.Experiments/Azure.Experiments/Network/NetworkInterfaceParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Network/NetworkInterfaceParameters.cs
@@ -27,8 +27,9 @@ public NetworkInterfaceParameters(
             Pia = pia;
         }
 
-        public override Task<NetworkInterface> GetAsync(Context context)
+        public override Task<NetworkInterface> GetAsync(GetContext context)
             => context
+                .Context
                 .CreateNetwork()
                 .NetworkInterfaces
                 .GetAsync(ResourceGroup.Name, Name);

diff --git a/experiments/Azure.Experiments/Azure.Experiments/Network/NetworkSecurityGroupParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Network/NetworkSecurityGroupParameters.cs
@@ -13,8 +13,9 @@ public NetworkSecurityGroupParameters(
         {
         }
 
-        public override Task<NetworkSecurityGroup> GetAsync(Context context)
+        public override Task<NetworkSecurityGroup> GetAsync(GetContext context)
             => context
+                .Context
                 .CreateNetwork()
                 .NetworkSecurityGroups
                 .GetAsync(ResourceGroup.Name, Name);

diff --git a/experiments/Azure.Experiments/Azure.Experiments/Network/PublicIpAddressParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Network/PublicIpAddressParameters.cs
@@ -14,8 +14,9 @@ public PublicIpAddressParameters(
         {
         }
 
-        public override Task<PublicIPAddress> GetAsync(Context context)
+        public override Task<PublicIPAddress> GetAsync(GetContext context)
             => context
+                .Context
                 .CreateNetwork()
                 .PublicIPAddresses
                 .GetAsync(ResourceGroup.Name, Name);

diff --git a/experiments/Azure.Experiments/Azure.Experiments/Network/SubnetParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Network/SubnetParameters.cs
@@ -1,5 +1,6 @@
 using System.Threading.Tasks;
 using Microsoft.Azure.Management.Network.Models;
+using System.Linq;
 
 namespace Microsoft.Azure.Experiments.Network
 {
@@ -13,5 +14,11 @@ public SubnetParameters(
         {
             VirtualNetwork = virtualNetwork;
         }
+
+        public override async Task<Subnet> GetAsync(GetContext context)
+        {
+            var virtualNetwork = await context.GetOrNullAsync(VirtualNetwork);
+            return virtualNetwork?.Subnets.FirstOrDefault(s => s.Name == Name);
+        }
     }
 }
diff --git a/experiments/Azure.Experiments/Azure.Experiments/Network/VirtualNetworkParameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Network/VirtualNetworkParameters.cs
@@ -13,8 +13,9 @@ public VirtualNetworkParameters(
         {
         }
 
-        public override Task<VirtualNetwork> GetAsync(Context context)
+        public override Task<VirtualNetwork> GetAsync(GetContext context)
             => context
+                .Context
                 .CreateNetwork()
                 .VirtualNetworks
                 .GetAsync(ResourceGroup.Name, Name);

diff --git a/experiments/Azure.Experiments/Azure.Experiments/Parameters.cs b/experiments/Azure.Experiments/Azure.Experiments/Parameters.cs
@@ -27,6 +27,6 @@ protected Parameters(string name, IEnumerable<Parameters> parameters)
         {
         }
 
-        public abstract Task<T> GetAsync(Context context);
+        public abstract Task<T> GetAsync(GetContext context);
     }
 }