Add new InsertUsingDefaults API

Author: mythz

src/ServiceStack.OrmLite/OrmLiteWriteApi.cs

@@ -65,13 +65,13 @@ public static long Insert<T>(this IDbConnection dbConn, T obj, bool selectIdenti
         }
 
         /// <summary>
-        /// Insert 1 or more POCOs in a transaction. E.g:
-        /// <para>db.Insert(new Person { Id = 1, FirstName = "Tupac", LastName = "Shakur", Age = 25 },</para>
-        /// <para>          new Person { Id = 2, FirstName = "Biggie", LastName = "Smalls", Age = 24 })</para>
+        /// Insert 1 or more POCOs in a transaction using Table default values when defined. E.g:
+        /// <para>db.InsertUsingDefaults(new Person { FirstName = "Tupac", LastName = "Shakur" },</para>
+        /// <para>                       new Person { FirstName = "Biggie", LastName = "Smalls" })</para>
         /// </summary>
-        public static void Insert<T>(this IDbConnection dbConn, params T[] objs)
+        public static void InsertUsingDefaults<T>(this IDbConnection dbConn, params T[] objs)
         {
-            dbConn.Exec(dbCmd => dbCmd.Insert(objs));
+            dbConn.Exec(dbCmd => dbCmd.InsertUsingDefaults(objs));
         }
 
         /// <summary>
@@ -83,6 +83,16 @@ public static void InsertAll<T>(this IDbConnection dbConn, IEnumerable<T> objs)
             dbConn.Exec(dbCmd => dbCmd.InsertAll(objs));
         }
 
+        /// <summary>
+        /// Insert 1 or more POCOs in a transaction. E.g:
+        /// <para>db.Insert(new Person { Id = 1, FirstName = "Tupac", LastName = "Shakur", Age = 25 },</para>
+        /// <para>          new Person { Id = 2, FirstName = "Biggie", LastName = "Smalls", Age = 24 })</para>
+        /// </summary>
+        public static void Insert<T>(this IDbConnection dbConn, params T[] objs)
+        {
+            dbConn.Exec(dbCmd => dbCmd.Insert(objs));
+        }
+
         /// <summary>
         /// Updates 1 POCO. All fields are updated except for the PrimaryKey which is used as the identity selector. E.g:
         /// <para>db.Update(new Person { Id = 1, FirstName = "Jimi", LastName = "Hendrix", Age = 27 })</para>

src/ServiceStack.OrmLite/OrmLiteWriteCommandExtensions.cs

@@ -16,6 +16,7 @@
 using System.Linq;
 using ServiceStack.Data;
 using ServiceStack.Logging;
+using ServiceStack.Text;
 
 namespace ServiceStack.OrmLite
 {
@@ -628,7 +629,6 @@ internal static long Insert<T>(this IDbCommand dbCmd, T obj, bool selectIdentity
             return dbCmd.ExecNonQuery();
         }
 
-
         internal static void Insert<T>(this IDbCommand dbCmd, params T[] objs)
         {
             InsertAll(dbCmd, objs);
@@ -671,6 +671,50 @@ internal static void InsertAll<T>(this IDbCommand dbCmd, IEnumerable<T> objs)
             }
         }
 
+        internal static void InsertUsingDefaults<T>(this IDbCommand dbCmd, params T[] objs)
+        {
+            IDbTransaction dbTrans = null;
+
+            try
+            {
+                if (dbCmd.Transaction == null)
+                    dbCmd.Transaction = dbTrans = dbCmd.Connection.BeginTransaction();
+
+                var dialectProvider = dbCmd.GetDialectProvider();
+
+                var modelDef = typeof(T).GetModelDefinition();
+                var fieldsWithoutDefaults = modelDef.FieldDefinitionsArray
+                    .Where(x => x.DefaultValue == null)
+                    .Select(x => x.Name)
+                    .ToHashSet(); 
+
+                dialectProvider.PrepareParameterizedInsertStatement<T>(dbCmd,
+                    insertFields: fieldsWithoutDefaults);
+
+                foreach (var obj in objs)
+                {
+                    OrmLiteConfig.InsertFilter?.Invoke(dbCmd, obj);
+                    dialectProvider.SetParameterValues<T>(dbCmd, obj);
+
+                    try
+                    {
+                        dbCmd.ExecNonQuery();
+                    }
+                    catch (Exception ex)
+                    {
+                        Log.Error("SQL ERROR: {0}".Fmt(dbCmd.GetLastSqlAndParams()), ex);
+                        throw;
+                    }
+                }
+
+                dbTrans?.Commit();
+            }
+            finally
+            {
+                dbTrans?.Dispose();
+            }
+        }
+
         internal static int Save<T>(this IDbCommand dbCmd, params T[] objs)
         {
             return SaveAll(dbCmd, objs);

tests/ServiceStack.OrmLite.Tests/DefaultValueTests.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Data;
+using System.Linq;
 using NUnit.Framework;
 using ServiceStack.OrmLite.SqlServer;
 using ServiceStack.OrmLite.SqlServer.Converters;
@@ -303,5 +304,42 @@ public void Can_filter_updateAdd_sqlexpression_to_insert_date()
                 VerifyUpdateDate(db);
             }
         }
+
+        [Test]
+        public void Does_use_defaults_for_missing_values()
+        {
+            using (var db = OpenDbConnection())
+            {
+                db.DropAndCreateTable<ModelWithDefaults>();
+
+                db.Insert(new ModelWithDefaults { DefaultInt = 10 });
+
+                var row = db.Select<ModelWithDefaults>().FirstOrDefault();
+
+                Assert.That(row.Id, Is.GreaterThan(0));
+                Assert.That(row.DefaultInt, Is.EqualTo(10));
+                Assert.That(row.DefaultString, Is.EqualTo("String"));
+            }
+        }
+
+        [Test]
+        public void Does_only_use_defaults_for_all_default_properties()
+        {
+            using (var db = OpenDbConnection())
+            {
+                db.DropAndCreateTable<ModelWithDefaults>();
+
+                db.InsertUsingDefaults(
+                    new ModelWithDefaults { Name = "foo", DefaultInt = 10 },
+                    new ModelWithDefaults { Name = "bar", DefaultString = "qux" });
+
+                var rows = db.Select<ModelWithDefaults>();
+                rows.PrintDump();
+
+                Assert.That(rows.All(x => x.Id > 0));
+                Assert.That(rows.All(x => x.DefaultInt == 1));
+                Assert.That(rows.All(x => x.DefaultString == "String"));
+            }
+        }
     }
 }

tests/ServiceStack.OrmLite.Tests/Shared/DefaultValues.cs

@@ -33,4 +33,18 @@ public class DefaultValues
         [Default(OrmLiteVariables.SystemUtc)]
         public DateTime UpdatedDateUtc { get; set; }
     }
+
+    public class ModelWithDefaults
+    {
+        [AutoIncrement]
+        public int Id { get; set; }
+
+        public string Name { get; set; }
+
+        [Default(1)]
+        public int DefaultInt { get; set; }
+
+        [Default("'String'")]
+        public string DefaultString { get; set; }
+    }
 }