Add support for Check Constraints

Author: mythz

lib/ServiceStack.Interfaces.dll

@@ -54,6 +54,8 @@ public class FieldDefinition
 
         public string DefaultValue { get; set; }
 
+        public string CheckConstraint { get; set; }
+
         public ForeignKeyConstraint ForeignKey { get; set; }
 
         public PropertyGetterDelegate GetValueFn { get; set; }

src/ServiceStack.OrmLite/FieldDefinition.cs

@@ -126,6 +126,7 @@ internal static ModelDefinition GetModelDefinition(this Type modelType)
                 var referenceAttr = propertyInfo.FirstAttribute<ReferenceAttribute>();
                 var fkAttr = propertyInfo.FirstAttribute<ForeignKeyAttribute>();
                 var customFieldAttr = propertyInfo.FirstAttribute<CustomFieldAttribute>();
+                var chkConstraintAttr = propertyInfo.FirstAttribute<CheckConstraintAttribute>();
 
                 var fieldDefinition = new FieldDefinition
                 {
@@ -147,6 +148,7 @@ internal static ModelDefinition GetModelDefinition(this Type modelType)
                     IsRowVersion = isRowVersion,
                     FieldLength = stringLengthAttr?.MaximumLength,
                     DefaultValue = defaultValueAttr?.DefaultValue,
+                    CheckConstraint = chkConstraintAttr?.Constraint,
                     ForeignKey = fkAttr == null
                         ? referencesAttr != null ? new ForeignKeyConstraint(referencesAttr.Type) : null
                         : new ForeignKeyConstraint(fkAttr.Type, fkAttr.OnDelete, fkAttr.OnUpdate, fkAttr.ForeignKeyName),

src/ServiceStack.OrmLite/OrmLiteConfigExtensions.cs

@@ -1228,6 +1228,12 @@ public virtual string ToCreateTableStatement(Type tableType)
 
                 sbColumns.Append(columnDefinition);
 
+                var sqlConstraint = GetCheckConstraint(fieldDef);
+                if (sqlConstraint != null)
+                {
+                    sbConstraints.Append(",\n" + sqlConstraint);
+                }
+
                 if (fieldDef.ForeignKey == null || OrmLiteConfig.SkipForeignKeys)
                     continue;
 
@@ -1246,6 +1252,14 @@ public virtual string ToCreateTableStatement(Type tableType)
             return sql;
         }
 
+        public virtual string GetCheckConstraint(FieldDefinition fieldDef)
+        {
+            if (fieldDef.CheckConstraint == null)
+                return null;
+
+            return $"CONSTRAINT CHK_{fieldDef.FieldName} CHECK ({fieldDef.CheckConstraint})";
+        }
+
         public virtual string ToPostCreateTableStatement(ModelDefinition modelDef)
         {
             return null;

src/ServiceStack.OrmLite/OrmLiteDialectProviderBase.cs

@@ -0,0 +1,69 @@
+using System;
+using NUnit.Framework;
+using ServiceStack.DataAnnotations;
+using ServiceStack.Logging;
+using ServiceStack.Text;
+
+namespace ServiceStack.OrmLite.Tests
+{
+    public class CheckConstraintTest
+    {
+        [AutoIncrement]
+        public int Id { get; set; }
+
+        [Required]
+        [CheckConstraint("Age > 1")]
+        public int Age { get; set; }
+
+        [CheckConstraint("Name IS NOT NULL")]
+        public string Name { get; set; }
+    }
+
+    public class CheckConstraintTests : OrmLiteTestBase
+    {
+        [Test]
+        public void Does_create_table_with_CheckConstraints()
+        {
+            if (Dialect == Dialect.MySql) return; //parsed but not supported http://stackoverflow.com/a/2115641/85785
+
+            using (var db = OpenDbConnection())
+            {
+                db.DropAndCreateTable<CheckConstraintTest>();
+
+                try
+                {
+                    db.Insert(new CheckConstraintTest { Age = 1 });
+                    Assert.Fail("Should fail");
+                }
+                catch (Exception ex)
+                {
+                    Assert.That(ex.Message.ToLower(), Does.Contain("age"));
+                    Assert.That(ex.Message.ToLower(), Does.Contain("constraint"));
+                }
+
+                try
+                {
+                    db.Insert(new CheckConstraintTest { Age = 2 });
+                    Assert.Fail("Should fail");
+                }
+                catch (Exception ex)
+                {
+                    ex.Message.Print();
+                    Assert.That(ex.Message.ToLower(), Does.Contain("name"));
+                    Assert.That(ex.Message.ToLower(), Does.Contain("constraint"));
+                }
+            }
+        }
+
+        [Test]
+        public void Can_insert_record_passing_check_constraints()
+        {
+            using (var db = OpenDbConnection())
+            {
+                db.DropAndCreateTable<CheckConstraintTest>();
+
+                db.Insert(new CheckConstraintTest { Age = 2, Name = "foo" });
+            }
+        }
+    }
+}

tests/ServiceStack.OrmLite.Tests/CheckConstraintTests.cs

@@ -136,6 +136,7 @@
     <Compile Include="Async\SqlExpressionTests.cs" />
     <Compile Include="Async\SqlServerProviderTestsAsync.cs" />
     <Compile Include="Async\UpdateAsyncTests.cs" />
+    <Compile Include="CheckConstraintTests.cs" />
     <Compile Include="CustomSqlExpressionTests.cs" />
     <Compile Include="Expression\JoinAliasTests.cs" />
     <Compile Include="Expression\RestrictionTests.cs" />