Added Azure SQL Standard estimation

TheCloudTheory · Jul 24, 2022 · deed68a · deed68a
1 parent 0279311
commit deed68a
Show file tree

Hide file tree

Showing 14 changed files with 347 additions and 17 deletions.
diff --git a/arm-estimator-generator/Program.cs b/arm-estimator-generator/Program.cs
@@ -4,16 +4,21 @@ internal class Program
 {
     private static async Task<int> Main(string[] args)
     {
-        File.AppendAllText("compute.csv", "type;serviceId;serviceName;skuId;skuName;productId;productName;meterId;meterName" + Environment.NewLine);
+        if(File.Exists("sql.csv"))
+        {
+            File.Delete("sql.csv");
+        }
+
+        File.AppendAllText("sql.csv", "type;serviceId;serviceName;skuId;skuName;productId;productName;meterId;meterName" + Environment.NewLine);
 
         var client = new HttpClient();
-        var request = new HttpRequestMessage(HttpMethod.Get, $"https://prices.azure.com/api/retail/prices?$filter=serviceFamily eq 'Compute' and armRegionName eq 'westeurope'");
+        var request = new HttpRequestMessage(HttpMethod.Get, $"https://prices.azure.com/api/retail/prices?$filter=serviceName eq 'SQL Database' and armRegionName eq 'westeurope'");
         var response = await client.SendAsync(request);
         var data = JsonSerializer.Deserialize<RetailAPIResponse>(await response.Content.ReadAsStreamAsync());
 
         foreach(var item in data.Items)
         {
-            File.AppendAllText("compute.csv", $"{item.type};{item.serviceId};{item.serviceName};{item.skuId};{item.skuName};{item.productId};{item.productName};{item.meterId};{item.meterName}" + Environment.NewLine);
+            File.AppendAllText("sql.csv", $"{item.type};{item.serviceId};{item.serviceName};{item.skuId};{item.skuName};{item.productId};{item.productName};{item.meterId};{item.meterName}" + Environment.NewLine);
         }
 
         var nextLink = data.NextPageLink;
@@ -33,7 +38,7 @@ private static async Task<int> Main(string[] args)
 
         foreach (var item in data.Items)
         {
-            File.AppendAllText("compute.csv", $"{item.type};{item.serviceId};{item.serviceName};{item.skuId};{item.skuName};{item.productId};{item.productName};{item.meterId};{item.meterName}" + Environment.NewLine);
+            File.AppendAllText("sql.csv", $"{item.type};{item.serviceId};{item.serviceName};{item.skuId};{item.skuName};{item.productId};{item.productName};{item.meterId};{item.meterName}" + Environment.NewLine);
         }
 
         return data.NextPageLink;

diff --git a/arm-estimator/Calculation/BaseEstimation.cs b/arm-estimator/Calculation/BaseEstimation.cs
@@ -2,9 +2,11 @@
 {
     internal static readonly int HoursInMonth = 720;
     internal readonly RetailItem[] items;
+    internal readonly WhatIfAfterChange change;
 
-    public BaseEstimation(RetailItem[] items)
+    public BaseEstimation(RetailItem[] items, WhatIfAfterChange change)
     {
         this.items = items;
+        this.change = change;
     }
 }
diff --git a/arm-estimator/Products/AKS/AKSEstimationCalculation.cs b/arm-estimator/Products/AKS/AKSEstimationCalculation.cs
@@ -1,7 +1,7 @@
 internal class AKSEstimationCalculation : BaseEstimation, IEstimationCalculation
 {
-    public AKSEstimationCalculation(RetailItem[] items)
-        : base(items)
+    public AKSEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
     {
     }
 

diff --git a/arm-estimator/Products/AppService/AppServicePlanEstimationCalculation.cs b/arm-estimator/Products/AppService/AppServicePlanEstimationCalculation.cs
@@ -1,7 +1,7 @@
 internal class AppServicePlanEstimationCalculation : BaseEstimation, IEstimationCalculation
 {
-    public AppServicePlanEstimationCalculation(RetailItem[] items)
-        : base(items)
+    public AppServicePlanEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
     {
     }
 

diff --git a/arm-estimator/Products/ContainerApps/ContainerAppsEstimationCalculation.cs b/arm-estimator/Products/ContainerApps/ContainerAppsEstimationCalculation.cs
@@ -1,7 +1,7 @@
 internal class ContainerAppsEstimationCalculation : BaseEstimation, IEstimationCalculation
 {
-    public ContainerAppsEstimationCalculation(RetailItem[] items)
-        : base(items)
+    public ContainerAppsEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
     {
     }
 

diff --git a/arm-estimator/Products/ContainerRegistry/ContainerRegistryEstimationCalculation.cs b/arm-estimator/Products/ContainerRegistry/ContainerRegistryEstimationCalculation.cs
@@ -1,7 +1,7 @@
 internal class ContainerRegistryEstimationCalculation : BaseEstimation, IEstimationCalculation
 {
-    public ContainerRegistryEstimationCalculation(RetailItem[] items)
-        : base(items)
+    public ContainerRegistryEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
     {
     }
 

diff --git a/arm-estimator/Products/SQL/SQLEstimationCalculation.cs b/arm-estimator/Products/SQL/SQLEstimationCalculation.cs
@@ -0,0 +1,60 @@
+internal class SQLEstimationCalculation : BaseEstimation, IEstimationCalculation
+{
+    public SQLEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
+    {
+    }
+
+    public IOrderedEnumerable<RetailItem> GetItems()
+    {
+        var consumptionMetrics = this.items.Where(_ => _.type != "Reservation");
+
+        return this.items.Where(_ => _.type != "Reservation").OrderByDescending(_ => _.retailPrice);
+    }
+
+    public double GetTotalCost()
+    {
+        double? estimatedCost = 0;
+        var items = GetItems();
+
+        foreach(var item in items)
+        {
+            // B DTUs
+            if(item.meterId == "cae64797-9ecf-4906-b517-6238c80c045f")
+            {
+                estimatedCost += item.retailPrice * 30;
+            }
+            // S0 DTUs
+            else if (item.meterId == "a149966f-73b4-4e1d-b335-d2a572b1e6bd")
+            {
+                estimatedCost += item.retailPrice * 30;
+            }
+            // S1 DTUs
+            else if (item.meterId == "54c6bbf3-322b-406d-8411-101bc4b9443a")
+            {
+                estimatedCost += item.retailPrice * 30;
+            }
+            // S2 DTUs
+            else if (item.meterId == "f303a21a-05f1-4563-b2ad-7523eebc4058")
+            {
+                estimatedCost += item.retailPrice * 30;
+            }
+            // S3 DTUs
+            else if (item.meterId == "6bafdc11-b964-4895-9d4e-a0e548db1b2b")
+            {
+                estimatedCost += item.retailPrice * 30;
+            }
+            // 10 DTU - everything above S3
+            else if (item.meterId == "94aaed62-bf2f-4461-8f69-078933cb2f72")
+            {
+                estimatedCost +=  item.retailPrice * 30;
+            }
+            else
+            {
+                estimatedCost += item.retailPrice;
+            }
+        }
+
+        return estimatedCost == null ? 0 : (double)estimatedCost;
+    }
+}
diff --git a/arm-estimator/Products/SQL/SQLQueryFilter.cs b/arm-estimator/Products/SQL/SQLQueryFilter.cs
@@ -0,0 +1,31 @@
+using Microsoft.Extensions.Logging;
+
+internal class SQLQueryFilter : IQueryFilter
+{
+    private const string ServiceId = "DZH3180HX10K";
+
+    private readonly WhatIfAfterChange afterState;
+    private readonly ILogger logger;
+
+    public SQLQueryFilter(WhatIfAfterChange afterState, ILogger logger)
+    {
+        this.afterState = afterState;
+        this.logger = logger;
+    }
+
+    public string? GetFiltersBasedOnDesiredState()
+    {
+        var location = this.afterState.location;
+        var sku = this.afterState.sku?.name;
+        if (sku == null)
+        {
+            this.logger.LogError("Can't create a filter for Azure SQL when SKU is unavailable.");
+            return null;
+        }
+
+        var skuIds = SQLSupportedData.SkuToSkuIdMap[sku];
+        var skuIdsFilter = string.Join(" or ", skuIds.Select(_ => $"skuId eq '{_}'"));
+
+        return $"$filter=serviceId eq '{ServiceId}' and armRegionName eq '{location}' and ({skuIdsFilter})";
+    }
+}
diff --git a/arm-estimator/Products/SQL/SQLRetailQuery.cs b/arm-estimator/Products/SQL/SQLRetailQuery.cs
@@ -0,0 +1,21 @@
+using Microsoft.Extensions.Logging;
+
+internal class SQLRetailQuery : BaseRetailQuery, IRetailQuery
+{
+    public SQLRetailQuery(WhatIfChange change, ILogger logger)
+        : base(change, logger)
+    {
+    }
+
+    public string? GetQueryUrl()
+    {
+        if (this.change.after == null)
+        {
+            this.logger.LogError("Can't generate Retail API query if desired state is unavailable.");
+            return null;
+        }
+
+        var filter = new SQLQueryFilter(this.change.after, this.logger).GetFiltersBasedOnDesiredState();
+        return $"https://prices.azure.com/api/retail/prices?{filter}";
+    }
+}
diff --git a/arm-estimator/Products/SQL/SQLSupportedData.cs b/arm-estimator/Products/SQL/SQLSupportedData.cs
@@ -0,0 +1,59 @@
+internal class SQLSupportedData
+{
+    public static readonly IReadOnlyDictionary<string, string[]> SkuToSkuIdMap = new Dictionary<string, string[]>()
+    {
+        { "Free", new[] {
+            "DZH318Z0BQH9/0013"
+            }
+        },
+        { "Basic", new[] {
+            "DZH318Z0BQGZ/004K"
+            }
+        },
+        { "S0", new[] {
+            "DZH318Z0BQHF/00LB"
+            }
+        },
+        { "S1", new[] {
+            "DZH318Z0BQHF/00LT"
+            }
+        },
+        { "S2", new[] {
+            "DZH318Z0BQHF/017X"
+            }
+        },
+        { "S3", new[] {
+            "DZH318Z0BQHF/00TX"
+            }
+        },
+        { "S4", new[] {
+            "DZH318Z0BQHF/014D"
+            }
+        },
+        { "S6", new[] {
+            "DZH318Z0BQHF/00TZ"
+            }
+        },
+        { "S7", new[] {
+            "DZH318Z0BQHF/012N"
+            }
+        },
+        { "S9", new[] {
+            "DZH318Z0BQHF/00SQ"
+            }
+        },
+        { "S12", new[] {
+            "DZH318Z0BQHF/00LJ"
+            }
+        }
+    };
+
+    public static readonly IReadOnlyDictionary<string, int> SkuToDTUsMap = new Dictionary<string, int>()
+    {
+        { "S4", 200 },
+        { "S6", 400 },
+        { "S7", 800 },
+        { "S9", 1600 },
+        { "S12", 3000 }
+    };
+}
diff --git a/arm-estimator/Products/StorageAccount/StorageAccountEstimationCalculation.cs b/arm-estimator/Products/StorageAccount/StorageAccountEstimationCalculation.cs
@@ -1,7 +1,7 @@
 internal class StorageAccountEstimationCalculation : BaseEstimation, IEstimationCalculation
 {
-    public StorageAccountEstimationCalculation(RetailItem[] items)
-        : base(items)
+    public StorageAccountEstimationCalculation(RetailItem[] items, WhatIfAfterChange change)
+        : base(items, change)
     {
     }
 

diff --git a/arm-estimator/WhatIf/WhatIfProcessor.cs b/arm-estimator/WhatIf/WhatIfProcessor.cs
@@ -51,6 +51,12 @@ public async Task Process(WhatIfChange[] changes)
                 case "Microsoft.App/containerApps":
                     totalCost += await Calculate<ContainerAppsRetailQuery, ContainerAppsEstimationCalculation>(change, id);
                     break;
+                case "Microsoft.Sql/servers":
+                    totalCost += 0;
+                    break;
+                case "Microsoft.Sql/servers/databases":
+                    totalCost += await Calculate<SQLRetailQuery, SQLEstimationCalculation>(change, id);
+                    break;
                 default:
                     logger.LogWarning("{resourceType} is not yet supported.", id.ResourceType);
                     break;
@@ -71,7 +77,13 @@ private async Task<double> Calculate<TQuery, TCalculation>(WhatIfChange change,
             return 0;
         }
 
-        if (Activator.CreateInstance(typeof(TCalculation), new object[] { data.Items }) is not TCalculation estimation)
+        if(change.after == null)
+        {
+            this.logger.LogError("No data available for WhatIf operation.");
+            return 0;
+        }
+
+        if (Activator.CreateInstance(typeof(TCalculation), new object[] { data.Items, change.after }) is not TCalculation estimation)
         {
             this.logger.LogError("Couldn't create an instance of {type}.", typeof(TCalculation));
             return 0;

diff --git a/arm-estimator/templates/complex.bicep b/arm-estimator/templates/complex.bicep
@@ -19,6 +19,67 @@ resource containerapp 'Microsoft.App/containerApps@2022-03-01' = {
   name: 'containerapp'
   location: 'westeurope'
   properties: {
+  }
+}
+
+resource dbserver 'Microsoft.Sql/servers@2021-11-01-preview' = {
+  name: 'sqlserver'
+  location: 'westeurope'
+  properties: {
 
   }
 }
+
+resource dbbasic 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbbasic'
+  location: 'westeurope'
+  sku: {
+    name: 'Basic'
+  }
+}
+
+resource dbstandard0 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbstandard0'
+  location: 'westeurope'
+  sku: {
+    name: 'S0'
+  }
+}
+
+resource dbstandard1 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbstandard1'
+  location: 'westeurope'
+  sku: {
+    name: 'S1'
+  }
+}
+
+resource dbstandard2 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbstandard2'
+  location: 'westeurope'
+  sku: {
+    name: 'S2'
+  }
+}
+
+resource dbstandard3 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbstandard3'
+  location: 'westeurope'
+  sku: {
+    name: 'S3'
+  }
+}
+
+resource dbstandard4 'Microsoft.Sql/servers/databases@2021-11-01-preview' = {
+  parent: dbserver
+  name: 'dbstandard4'
+  location: 'westeurope'
+  sku: {
+    name: 'S4'
+  }
+}