Add support for multiple [Pre/Post/Create/Drop/Table] attributes

Author: mythz

src/ServiceStack.OrmLite/OrmLiteConfigExtensions.cs

@@ -17,6 +17,7 @@
 using System.Threading;
 using ServiceStack.DataAnnotations;
 using ServiceStack.OrmLite.Converters;
+using ServiceStack.Text;
 
 namespace ServiceStack.OrmLite
 {
@@ -51,21 +52,36 @@ internal static ModelDefinition GetModelDefinition(this Type modelType)
             var modelAliasAttr = modelType.FirstAttribute<AliasAttribute>();
             var schemaAttr = modelType.FirstAttribute<SchemaAttribute>();
 
-            var preCreate = modelType.FirstAttribute<PreCreateTableAttribute>();
-            var postCreate = modelType.FirstAttribute<PostCreateTableAttribute>();
-            var preDrop = modelType.FirstAttribute<PreDropTableAttribute>();
-            var postDrop = modelType.FirstAttribute<PostDropTableAttribute>();
+            var preCreates = modelType.AllAttributes<PreCreateTableAttribute>();
+            var postCreates = modelType.AllAttributes<PostCreateTableAttribute>();
+            var preDrops = modelType.AllAttributes<PreDropTableAttribute>();
+            var postDrops = modelType.AllAttributes<PostDropTableAttribute>();
+
+            string JoinSql(List<string> statements)
+            {
+                if (statements.Count == 0)
+                    return null;
+                var sb = StringBuilderCache.Allocate();
+                foreach (var sql in statements)
+                {
+                    if (sb.Length > 0)
+                        sb.AppendLine(";");
+                    sb.Append(sql);
+                }
+                var to = StringBuilderCache.ReturnAndFree(sb);
+                return to;
+            }
 
             modelDef = new ModelDefinition
             {
                 ModelType = modelType,
                 Name = modelType.Name,
                 Alias = modelAliasAttr?.Name,
                 Schema = schemaAttr?.Name,
-                PreCreateTableSql = preCreate?.Sql,
-                PostCreateTableSql = postCreate?.Sql,
-                PreDropTableSql = preDrop?.Sql,
-                PostDropTableSql = postDrop?.Sql,
+                PreCreateTableSql = JoinSql(preCreates.Map(x => x.Sql)),
+                PostCreateTableSql = JoinSql(postCreates.Map(x => x.Sql)),
+                PreDropTableSql = JoinSql(preDrops.Map(x => x.Sql)),
+                PostDropTableSql = JoinSql(postDrops.Map(x => x.Sql)),
             };
 
             modelDef.CompositeIndexes.AddRange(

tests/ServiceStack.OrmLite.Tests/CustomSqlTests.cs

@@ -44,6 +44,16 @@ public class ModelWithSeedDataSql
         public string Name { get; set; }
     }
 
+    [PostCreateTable("INSERT INTO ModelWithSeedDataSqlMulti (Name) VALUES ('Foo')"),
+     PostCreateTable("INSERT INTO ModelWithSeedDataSqlMulti (Name) VALUES ('Bar')")]
+    public class ModelWithSeedDataSqlMulti
+    {
+        [AutoIncrement]
+        public int Id { get; set; }
+
+        public string Name { get; set; }
+    }
+
     public class DynamicAttributeSeedData
     {
         [AutoIncrement]
@@ -126,6 +136,18 @@ public void Does_execute_CustomSql_after_table_created()
             Assert.That(seedDataNames, Is.EquivalentTo(new[] {"Foo", "Bar"}));
         }
 
+        [Test]
+        [IgnoreDialect(Dialect.AnyOracle | Dialect.AnyPostgreSql, "multiple SQL statements need to be wrapped in an anonymous block")]
+        public void Does_execute_multi_CustomSql_after_table_created()
+        {
+            using var db = OpenDbConnection();
+            db.DropAndCreateTable<ModelWithSeedDataSqlMulti>();
+
+            var seedDataNames = db.Select<ModelWithSeedDataSqlMulti>().ConvertAll(x => x.Name);
+
+            Assert.That(seedDataNames, Is.EquivalentTo(new[] {"Foo", "Bar"}));
+        }
+
         [Test]
         [IgnoreDialect(Dialect.AnyOracle | Dialect.AnyPostgreSql, "multiple SQL statements need to be wrapped in an anonymous block")]
         public void Does_execute_CustomSql_after_table_created_using_dynamic_attribute()