Add new AliasNamingStrategy and convenient SqlTable, SqlColumn and SqlValue methods for constructing custom sql

Author: mythz

src/ServiceStack.OrmLite/AliasNamingStrategy.cs

@@ -0,0 +1,27 @@
+using System.Collections.Generic;
+
+namespace ServiceStack.OrmLite
+{
+    public class AliasNamingStrategy : OrmLiteNamingStrategyBase
+    {
+        public Dictionary<string, string> TableAliases = new Dictionary<string, string>();
+        public Dictionary<string, string> ColumnAliases = new Dictionary<string, string>();
+        public INamingStrategy UseNamingStrategy { get; set; }
+
+        public override string GetTableName(string name)
+        {
+            string alias;
+            return UseNamingStrategy != null 
+                ? UseNamingStrategy.GetTableName(TableAliases.TryGetValue(name, out alias) ? alias : name)
+                : base.GetTableName(TableAliases.TryGetValue(name, out alias) ? alias : name);
+        }
+
+        public override string GetColumnName(string name)
+        {
+            string alias;
+            return UseNamingStrategy != null
+                ? UseNamingStrategy.GetColumnName(ColumnAliases.TryGetValue(name, out alias) ? alias : name)
+                : base.GetColumnName(ColumnAliases.TryGetValue(name, out alias) ? alias : name);
+        }
+    }
+}

src/ServiceStack.OrmLite/OrmLiteUtilExtensions.cs

@@ -153,7 +153,22 @@ public static string SqlFmt(this string sqlText, params object[] sqlParams)
 			return string.Format(sqlText, escapedParams.ToArray());
 		}
 
-	    public static string[] IllegalSqlFragmentTokens = { 
+        public static string SqlColumn(this string columnName)
+        {
+            return OrmLiteConfig.DialectProvider.GetQuotedColumnName(columnName);
+        }
+
+        public static string SqlTable(this string tableName)
+        {
+            return OrmLiteConfig.DialectProvider.GetQuotedTableName(tableName);
+        }
+
+        public static string SqlValue(this object value)
+        {
+            return "{0}".SqlFmt(value);
+        }
+
+        public static string[] IllegalSqlFragmentTokens = { 
             "--", ";--", ";", "%", "/*", "*/", "@@", "@", 
             "char", "nchar", "varchar", "nvarchar",
             "alter", "begin", "cast", "create", "cursor", "declare", "delete",

src/ServiceStack.OrmLite/ServiceStack.OrmLite.Signed.csproj

@@ -110,6 +110,7 @@
     </Reference>
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="AliasNamingStrategy.cs" />
     <Compile Include="Expressions\SqlExpressionVisitor.cs" />
     <Compile Include="Expressions\ParameterRebinder.cs" />
     <Compile Include="Expressions\PredicateBuilder.cs" />

src/ServiceStack.OrmLite/ServiceStack.OrmLite.csproj

@@ -104,6 +104,7 @@
     </Reference>
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="AliasNamingStrategy.cs" />
     <Compile Include="Expressions\SqlExpressionVisitor.cs" />
     <Compile Include="Expressions\ParameterRebinder.cs" />
     <Compile Include="Expressions\PredicateBuilder.cs" />

tests/ServiceStack.OrmLite.Tests/OrmLiteCreateTableWithNamingStrategyTest.cs

@@ -1,6 +1,8 @@
 using System;
 using NUnit.Framework;
 using ServiceStack.Common.Tests.Models;
+using ServiceStack.OrmLite.PostgreSQL;
+using ServiceStack.Text;
 
 namespace ServiceStack.OrmLite.Tests
 {
@@ -54,6 +56,45 @@ public void Can_create_TableWithNamigStrategy_table_nameUnderscoreCoumpound()
 			OrmLiteConfig.DialectProvider.NamingStrategy= new OrmLiteNamingStrategyBase();
 		}
 
+        [Test]
+        public void Can_create_TableWithNamigStrategy_table_aliases()
+        {
+	        var aliasNamingStrategy = new AliasNamingStrategy
+	        {
+                TableAliases = { { "ModelWithOnlyStringFields", "TableAlias" } },
+                ColumnAliases = { { "Name", "ColumnAlias" } },
+            };
+	        OrmLiteConfig.DialectProvider.NamingStrategy = aliasNamingStrategy;
+
+            using (var db = OpenDbConnection())
+            {
+                db.CreateTable<ModelWithOnlyStringFields>(true);
+
+                var sql = db.GetLastSql();
+                Assert.That(sql, Is.StringContaining("CREATE TABLE \"TableAlias\""));
+                Assert.That(sql, Is.StringContaining("\"ColumnAlias\""));
+
+                var result = db.SqlList<ModelWithIdAndName>(
+                    "SELECT * FROM {0} WHERE {1} = {2}"
+                        .Fmt("ModelWithOnlyStringFields".SqlTable(),
+                             "Name".SqlColumn(),
+                             "foo".SqlValue()));
+
+                Assert.That(db.GetLastSql(), Is.EqualTo("SELECT * FROM \"TableAlias\" WHERE \"ColumnAlias\" = 'foo'"));
+
+                db.DropTable<ModelWithOnlyStringFields>();
+
+                aliasNamingStrategy.UseNamingStrategy = new PostgreSqlNamingStrategy();
+
+                db.CreateTable<ModelWithOnlyStringFields>(true);
+                sql = db.GetLastSql();
+                Assert.That(sql, Is.StringContaining("CREATE TABLE \"table_alias\""));
+                Assert.That(sql, Is.StringContaining("\"column_alias\""));
+            }
+
+            OrmLiteConfig.DialectProvider.NamingStrategy = new OrmLiteNamingStrategyBase();
+        }
+
 		[Test]
 		public void Can_get_data_from_TableWithNamigStrategy_with_GetById()
 		{

tests/ServiceStack.OrmLite.Tests/SqlFormatTests.cs

@@ -72,5 +72,6 @@ public void SqlParam_sanitizes_param_values()
         {
             Assert.That("' or Field LIKE '%".SqlParam(), Is.EqualTo("'' or Field LIKE ''%"));
         }
+        
     }
 }