Added support for VirtualDictionary

Author: dei79

CoreHelpers.WindowsAzure.Storage.Table.Tests/ITS22PartialUpates.cs

@@ -16,7 +16,7 @@ public ITS22PartialUpates(IStorageContext context)
         }
 
         [Fact]
-        public async Task CreateAndVerifyBackup()
+        public async Task VerifyPartialUpdate()
         {
             using (var scp = _rootContext.CreateChildContext())
             {

CoreHelpers.WindowsAzure.Storage.Table.Tests/ITS23PartialDirectoryUpates.cs

@@ -0,0 +1,84 @@
+using System;
+using CoreHelpers.WindowsAzure.Storage.Table.Backup;
+using CoreHelpers.WindowsAzure.Storage.Table.Tests.Contracts;
+using CoreHelpers.WindowsAzure.Storage.Table.Tests.Extensions;
+using CoreHelpers.WindowsAzure.Storage.Table.Tests.Models;
+
+namespace CoreHelpers.WindowsAzure.Storage.Table.Tests
+{
+    public class ITS23PartialDirectoryUpates
+    {
+        private readonly IStorageContext _rootContext;
+
+        public ITS23PartialDirectoryUpates(IStorageContext context)
+        {
+            _rootContext = context;
+        }
+
+        [Fact]
+        public async Task VerifyDirectoryUpdate()
+        {
+            using (var scp = _rootContext.CreateChildContext())
+            {
+                // set the tablename context
+                scp.SetTableContext();
+
+                // configure the entity mapper
+                scp.AddAttributeMapper<PartialDirectoryUpdateModel>();
+
+                // verify we don't have any value
+                Assert.Empty(await scp.EnableAutoCreateTable().Query<PartialDirectoryUpdateModel>().InAllPartitions().Now());
+
+
+                // insert the model with no dictionary
+                await scp.MergeOrInsertAsync<PartialDirectoryUpdateModel>(new PartialDirectoryUpdateModel() { CustomerId = "C01", MeterId = "M01" });
+
+                var result = (await scp.EnableAutoCreateTable().Query<PartialDirectoryUpdateModel>().Now()).First();
+                Assert.Empty(result.Costs.Values);
+
+                // insert the model with a 2 value dictionary
+                await scp.MergeOrInsertAsync<PartialDirectoryUpdateModel>(new PartialDirectoryUpdateModel() { CustomerId = "C01", MeterId = "M01", Costs = new Dictionary<int, double>() {
+                    { 1, 11.1 },
+                    { 5, 55.5 }
+                } });
+
+                result = (await scp.EnableAutoCreateTable().Query<PartialDirectoryUpdateModel>().Now()).First();
+                Assert.Equal(2, result.Costs.Values.Count);
+                Assert.Equal(11.1, result.Costs[1]);
+                Assert.Equal(55.5, result.Costs[5]);
+                Assert.False(result.Costs.ContainsKey(0));
+                Assert.False(result.Costs.ContainsKey(2));
+                Assert.False(result.Costs.ContainsKey(3));
+                Assert.False(result.Costs.ContainsKey(4));
+
+                // add another value 
+                await scp.MergeOrInsertAsync<PartialDirectoryUpdateModel>(new PartialDirectoryUpdateModel()
+                {
+                    CustomerId = "C01",
+                    MeterId = "M01",
+                    Costs = new Dictionary<int, double>() {
+                    { 2, 22.2 }                    
+                }
+                });
+
+                result = (await scp.EnableAutoCreateTable().Query<PartialDirectoryUpdateModel>().Now()).First();
+                Assert.Equal(3, result.Costs.Values.Count);
+                Assert.Equal(11.1, result.Costs[1]);
+                Assert.Equal(22.2, result.Costs[2]);
+                Assert.Equal(55.5, result.Costs[5]);
+                Assert.False(result.Costs.ContainsKey(0));                
+                Assert.False(result.Costs.ContainsKey(3));
+                Assert.False(result.Costs.ContainsKey(4));
+
+                // override
+                await scp.InsertOrReplaceAsync<PartialDirectoryUpdateModel>(new PartialDirectoryUpdateModel() { CustomerId = "C01", MeterId = "M01" });
+                result = (await scp.EnableAutoCreateTable().Query<PartialDirectoryUpdateModel>().Now()).First();
+                Assert.Empty(result.Costs.Values);
+
+                // clean up
+                scp.DropTable<PartialDirectoryUpdateModel>();
+            }
+        }
+    }
+}
+

CoreHelpers.WindowsAzure.Storage.Table.Tests/Models/PartialDirectoryUpdateModel.cs

@@ -0,0 +1,20 @@
+using System;
+using CoreHelpers.WindowsAzure.Storage.Table.Attributes;
+
+namespace CoreHelpers.WindowsAzure.Storage.Table.Tests.Models
+{   
+    [Storable(Tablename = "PartialDirectoryUpdate")]
+    public class PartialDirectoryUpdateModel
+    {
+
+        [PartitionKey]
+        public string CustomerId { get; set; } = String.Empty;
+
+        [RowKey]
+        public string MeterId { get; set; } = String.Empty;
+
+        [VirtualDictionary("DC")]
+        public Dictionary<int, double> Costs { get; set; } = new Dictionary<int, double>();
+    }
+}
+

CoreHelpers.WindowsAzure.Storage.Table/Attributes/VirtualDictionaryAttribute.cs

@@ -0,0 +1,85 @@
+using System;
+using System.Collections;
+using System.Linq;
+using System.Reflection;
+using System.Xml.Linq;
+using Azure.Data.Tables;
+using CoreHelpers.WindowsAzure.Storage.Table.Extensions;
+using CoreHelpers.WindowsAzure.Storage.Table.Serialization;
+using HandlebarsDotNet;
+
+namespace CoreHelpers.WindowsAzure.Storage.Table.Attributes
+{
+    public class VirtualDictionaryAttribute : Attribute, IVirtualTypeAttribute
+    {
+        private string _propertyPrefix;
+
+        public VirtualDictionaryAttribute(string PropertyPrefix)
+        {
+            _propertyPrefix = PropertyPrefix;
+        }
+
+        public void ReadProperty<T>(TableEntity dataObject, PropertyInfo propertyInfo, T obj)
+        {
+            // search all properties with the prefix
+            var keys = dataObject.Keys.Where(k => k.StartsWith(_propertyPrefix));
+            
+            // visit every key
+            foreach(var key in keys)
+            {
+                // check if we have the name 
+                if (!dataObject.ContainsKey(key))
+                    continue;
+
+                // prepare the key value
+                var keyStringValue = key.Remove(0, _propertyPrefix.Length);
+
+                // read the value
+                var dataObjectPropertyValue = dataObject[key];
+
+                // get the dictionary 
+                var dictionary = propertyInfo.GetValue(obj) as IDictionary;
+
+                // work with the type
+                var arguments = dictionary.GetType().GetGenericArguments();
+                var keyType = arguments[0];
+                var valueType = arguments[1];
+
+                var keyValue = default(object);
+
+                switch(keyType.ToString())
+                {
+                    case "System.Int32":
+                        keyValue = Convert.ToInt32(keyStringValue);
+                        break;
+                    default:
+                        keyValue = keyStringValue;
+                        break;
+                }
+
+                dictionary.Add(keyValue, dataObjectPropertyValue);
+            }            
+        }
+
+        public void WriteProperty<T>(PropertyInfo propertyInfo, T obj, TableEntityBuilder builder)
+        {
+            // get the value
+            var dictionaryValue = propertyInfo.GetValue(obj);
+
+            // check if enumerable 
+            if ((dictionaryValue as IDictionary) == null)
+                return;
+
+            // visit every element
+            foreach(DictionaryEntry kvp in (dictionaryValue as IDictionary))
+            {
+                // generate the property name
+                var propertyName = $"{_propertyPrefix}{kvp.Key}";
+
+                // write the property
+                builder.AddProperty(propertyName, kvp.Value);                
+            }           
+        }
+    }
+}
+