Implemented NotMapped feature

cdemi · Feb 13, 2019 · b05ee76 · b05ee76
1 parent 5ac5cd3
commit b05ee76
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/Demo/Program.cs b/Demo/Program.cs
@@ -52,6 +52,7 @@ public MyClass(int id, int score, string winner)
             Foo = "Maps to bar!";
         }
 
+        [NotMapped]
         public string Winner { get; private set; }
         public int Score { get; set; }
         public int Id { get; set; }

diff --git a/Logic/Common.cs b/Logic/Common.cs
@@ -1,6 +1,5 @@
 using System.Collections.Generic;
 using System.ComponentModel.DataAnnotations.Schema;
-using System.Linq;
 using System.Reflection;
 
 namespace System.Data.SqlClient
@@ -13,6 +12,8 @@ public static DataTable GetDataTableFromFields<T>(IEnumerable<T> data, SqlBulkCo
             Type listType = typeof (T);
             foreach (PropertyInfo propertyInfo in listType.GetProperties())
             {
+                if(IsColumnNotMapped(propertyInfo)) continue;
+
                 var columnName = GetColumnName(propertyInfo);
                 dt.Columns.Add(columnName, propertyInfo.PropertyType);
                 sqlBulkCopy.ColumnMappings.Add(columnName, columnName);
@@ -23,6 +24,8 @@ public static DataTable GetDataTableFromFields<T>(IEnumerable<T> data, SqlBulkCo
                 DataRow dr = dt.NewRow();
                 foreach (PropertyInfo propertyInfo in listType.GetProperties())
                 {
+                    if (IsColumnNotMapped(propertyInfo)) continue;
+
                     var columnName = GetColumnName(propertyInfo);
                     dr[columnName] = propertyInfo.GetValue(value, null);
                 }
@@ -50,5 +53,17 @@ public static string GetColumnName(PropertyInfo propertyInfo)
             //it doesn't exist so return the property name
             return propertyInfo.Name;
         }
+
+        /// <summary>
+        /// Check if property is should be mapped
+        /// If the System.ComponentModel.DataAnnotations.NotMappedAttribute is used
+        /// it will skip this property
+        /// </summary>
+        /// <param name="propertyInfo"></param>
+        /// <returns></returns>
+        public static bool IsColumnNotMapped(PropertyInfo propertyInfo)
+        {
+            return propertyInfo.GetCustomAttribute<NotMappedAttribute>(false) != null;
+        }
     }
 }