diff Source/Templates/MSSQL.ttinclude @ 0:f990fcb411a9

Копия текущей версии из github
author cin
date Thu, 27 Mar 2014 21:46:09 +0400
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Source/Templates/MSSQL.ttinclude	Thu Mar 27 21:46:09 2014 +0400
@@ -0,0 +1,424 @@
+<#
+	ConnectionType = typeof(System.Data.SqlClient.SqlConnection).AssemblyQualifiedName;
+#><#+
+private void LoadServerMetadata()
+{
+	var tables  = CreateList(new { ID = "", Table  = new Table() });
+	var columns = CreateList(new { ID = "", Column = new Column() });
+
+	using (var conn = GetConnection())
+	using (var cmd  = conn.CreateCommand())
+	{
+		// Load tables & views.
+		//
+		string s = @"
+			SELECT
+				TABLE_CATALOG + '.' + TABLE_SCHEMA + '.' + TABLE_NAME AS TABLE_FULLNAME,
+				TABLE_SCHEMA,
+				TABLE_NAME,
+				TABLE_TYPE,
+				ISNULL(CONVERT(varchar(8000), x.Value), '') AS TABLE_DESC
+			FROM
+				INFORMATION_SCHEMA.TABLES s
+				LEFT JOIN 
+					sys.tables t 
+				ON 
+					OBJECT_ID(TABLE_CATALOG + '.' + TABLE_SCHEMA + '.' + TABLE_NAME) = t.object_id
+				LEFT JOIN 
+					sys.extended_properties x 
+				ON 
+					OBJECT_ID(TABLE_CATALOG + '.' + TABLE_SCHEMA + '.' + TABLE_NAME) = x.major_id AND 
+					x.minor_id = 0 AND 
+					x.name = 'MS_Description'
+			WHERE {0}
+			(
+				t.object_id IS NULL OR
+				t.is_ms_shipped <> 1 AND
+				(
+					SELECT 
+						major_id 
+					FROM 
+						sys.extended_properties 
+					WHERE
+						major_id = t.object_id and 
+						minor_id = 0           and 
+						class    = 1           and 
+						name     = N'microsoft_database_tools_support'
+					) IS NULL
+			)";
+
+		if (string.IsNullOrWhiteSpace(OwnerToInclude))
+			cmd.CommandText = string.Format(s, "");
+		else
+			cmd.CommandText = string.Format(s, " TABLE_SCHEMA = '" + OwnerToInclude + "' AND ");
+
+		using (var rd = cmd.ExecuteReader())
+		{
+			while (rd.Read())
+			{
+				var t = new
+				{
+					ID    = Convert.ToString(rd[0]),
+					Table = new Table
+					{
+						Owner         = rd[1].ToString(),
+						TableName     = rd[2].ToString(),
+						ClassName     = rd[2].ToString(),
+						IsView        = rd[3].ToString() == "VIEW",
+						BaseClassName = BaseEntityClass,
+						Description   = Convert.ToString(rd[4]),
+					}
+				};
+
+				tables.Add(t);
+			}
+		}
+
+		// Load columns.
+		//
+		s = @"
+			SELECT
+				(TABLE_CATALOG + '.' + TABLE_SCHEMA + '.' + TABLE_NAME) as id,
+				(CASE WHEN IS_NULLABLE = 'YES' THEN 1 ELSE 0 END)         as isNullable,
+				ORDINAL_POSITION         as colid,
+				COLUMN_NAME              as name,
+				c.DATA_TYPE              as dataType,
+				CHARACTER_MAXIMUM_LENGTH as length, 
+				ISNULL(NUMERIC_PRECISION, DATETIME_PRECISION) AS prec,
+				NUMERIC_SCALE            as scale,
+				COLUMNPROPERTY(object_id('[' + TABLE_SCHEMA + '].[' + TABLE_NAME + ']'), COLUMN_NAME, 'IsIdentity') as isIdentity,
+				ISNULL(CONVERT(varchar(8000), x.Value), '') AS COLUMN_DESC
+			FROM
+				INFORMATION_SCHEMA.COLUMNS c
+			  LEFT JOIN 
+				sys.extended_properties x 
+			  ON 
+				OBJECT_ID(TABLE_CATALOG + '.' + TABLE_SCHEMA + '.' + TABLE_NAME) = x.major_id AND 
+				ORDINAL_POSITION = x.minor_id AND 
+				x.name = 'MS_Description'
+			{0}";
+
+		if(string.IsNullOrWhiteSpace(OwnerToInclude))
+			cmd.CommandText = string.Format(s, "");
+		else
+			cmd.CommandText = string.Format(s, "WHERE TABLE_SCHEMA = '" + OwnerToInclude + "' ");
+
+		using (var rd = cmd.ExecuteReader())
+		{
+			while (rd.Read())
+			{
+				var col = new
+				{
+					ID     = Convert.ToString(rd["id"]),
+					Column = new Column
+					{
+						ID         = Convert.ToInt16  (rd["colid"]),
+						ColumnName = Convert.ToString (rd["name"]),
+						MemberName = Convert.ToString (rd["name"]),
+						ColumnType = Convert.ToString (rd["dataType"]),
+						IsNullable = Convert.ToBoolean(rd["isNullable"]),
+						IsIdentity = Convert.ToBoolean(rd["isIdentity"]),
+						Length     = rd.IsDBNull(rd.GetOrdinal("length")) ? 0 : Convert.ToInt64(rd["length"]),
+						Precision  = rd.IsDBNull(rd.GetOrdinal("prec"))   ? 0 : Convert.ToInt32(rd["prec"]),
+						Scale      = rd.IsDBNull(rd.GetOrdinal("scale"))  ? 0 : Convert.ToInt32(rd["scale"]),
+						Description = Convert.ToString(rd["COLUMN_DESC"]),
+					}
+				};
+
+				var c = col.Column;
+
+				switch (c.ColumnType)
+				{
+					case "image"            : c.Type = "byte[]";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.Image;            break;
+					case "text"             : c.Type = "string";         c.DbType = DbType.String;         c.SqlDbType = SqlDbType.Text;             break;
+					case "binary"           : c.Type = "byte[]";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.Binary;           break;
+					case "tinyint"          : c.Type = "byte";           c.DbType = DbType.Byte;           c.SqlDbType = SqlDbType.TinyInt;          break;
+					case "date"             : c.Type = "DateTime";       c.DbType = DbType.Date;           c.SqlDbType = SqlDbType.Date;             break;
+					case "time"             : c.Type = "DateTime";       c.DbType = DbType.Time;           c.SqlDbType = SqlDbType.Time;             break;
+					case "bit"              : c.Type = "bool";           c.DbType = DbType.Boolean;        c.SqlDbType = SqlDbType.Bit;              break;
+					case "smallint"         : c.Type = "short";          c.DbType = DbType.Int16;          c.SqlDbType = SqlDbType.SmallInt;         break;
+					case "decimal"          : c.Type = "decimal";        c.DbType = DbType.Decimal;        c.SqlDbType = SqlDbType.Decimal;          break;
+					case "int"              : c.Type = "int";            c.DbType = DbType.Int32;          c.SqlDbType = SqlDbType.Int;              break;
+					case "smalldatetime"    : c.Type = "DateTime";       c.DbType = DbType.DateTime;       c.SqlDbType = SqlDbType.SmallDateTime;    break;
+					case "real"             : c.Type = "float";          c.DbType = DbType.Single;         c.SqlDbType = SqlDbType.Real;             break;
+					case "money"            : c.Type = "decimal";        c.DbType = DbType.Currency;       c.SqlDbType = SqlDbType.Money;            break;
+					case "datetime"         : c.Type = "DateTime";       c.DbType = DbType.DateTime;       c.SqlDbType = SqlDbType.DateTime;         break;
+					case "float"            : c.Type = "double";         c.DbType = DbType.Double;         c.SqlDbType = SqlDbType.Float;            break;
+					case "numeric"          : c.Type = "decimal";        c.DbType = DbType.Decimal;        c.SqlDbType = SqlDbType.Decimal;          break;
+					case "smallmoney"       : c.Type = "decimal";        c.DbType = DbType.Currency;       c.SqlDbType = SqlDbType.SmallMoney;       break;
+					case "datetime2"        : c.Type = "DateTime";       c.DbType = DbType.DateTime2;      c.SqlDbType = SqlDbType.DateTime2;        break;
+					case "bigint"           : c.Type = "long";           c.DbType = DbType.Int64;          c.SqlDbType = SqlDbType.BigInt;           break;
+					case "varbinary"        : c.Type = "byte[]";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.VarBinary;        break;
+					case "timestamp"        : c.Type = "byte[]";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.Timestamp;        break;
+					case "sysname"          : c.Type = "string";         c.DbType = DbType.String;         c.SqlDbType = SqlDbType.NVarChar;         break;
+					case "nvarchar"         : c.Type = "string";         c.DbType = DbType.String;         c.SqlDbType = SqlDbType.NVarChar;         break;
+					case "varchar"          : c.Type = "string";         c.DbType = DbType.AnsiString;     c.SqlDbType = SqlDbType.VarChar;          break;
+					case "ntext"            : c.Type = "string";         c.DbType = DbType.String;         c.SqlDbType = SqlDbType.NText;            break;
+					case "uniqueidentifier" : c.Type = "Guid";           c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.UniqueIdentifier; break;
+					case "datetimeoffset"   : c.Type = "DateTimeOffset"; c.DbType = DbType.DateTimeOffset; c.SqlDbType = SqlDbType.DateTimeOffset;   break;
+					case "sql_variant"      : c.Type = "object";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.Variant;          break;
+					case "xml"              : c.Type = "string";         c.DbType = DbType.Xml;            c.SqlDbType = SqlDbType.Xml;              break;
+
+					case "char" :
+						c.Type      = Convert.ToInt32 (rd["length"]) == 1 ? "char" : "string";
+						c.DbType    = DbType.AnsiStringFixedLength;
+						c.SqlDbType = SqlDbType.Char;
+						break;
+
+					case "nchar" :
+						c.Type      = Convert.ToInt32 (rd["length"]) == 1 ? "char" : "string";
+						c.DbType    = DbType.StringFixedLength;
+						c.SqlDbType = SqlDbType.NChar;
+						break;
+
+					//hierarchyid
+					//geometry
+					//geography
+					default                 : c.Type = "byte[]";         c.DbType = DbType.Binary;         c.SqlDbType = SqlDbType.Binary;           break;
+				}
+
+				switch (c.Type)
+				{
+					case "string" :
+					case "object" :
+					case "byte[]" : c.IsClass = true; break;
+				}
+
+				if (c.IsNullable && !c.IsClass)
+					c.Type += "?";
+
+				columns.Add(col);
+			}
+		}
+
+		// Load PKs.
+		//
+		s = @"
+			SELECT
+				(k.TABLE_CATALOG + '.' + k.TABLE_SCHEMA + '.' + k.TABLE_NAME) as id,
+				k.CONSTRAINT_NAME                                             as name,
+				k.COLUMN_NAME                                                 as colname,
+				k.ORDINAL_POSITION                                            as colid
+			FROM
+				INFORMATION_SCHEMA.KEY_COLUMN_USAGE k
+			  JOIN 
+				INFORMATION_SCHEMA.TABLE_CONSTRAINTS c 
+			  ON 
+				k.CONSTRAINT_CATALOG = c.CONSTRAINT_CATALOG AND 
+				k.CONSTRAINT_SCHEMA = c.CONSTRAINT_SCHEMA AND 
+				k.CONSTRAINT_NAME = c.CONSTRAINT_NAME
+			WHERE
+				c.CONSTRAINT_TYPE='PRIMARY KEY' 
+				{0}";
+
+		if (string.IsNullOrWhiteSpace(OwnerToInclude))
+			cmd.CommandText = string.Format(s, "");
+		else
+			cmd.CommandText = string.Format(s, "AND k.TABLE_SCHEMA = '" + OwnerToInclude + "' ");
+
+		using (var rd = cmd.ExecuteReader())
+		{
+			while (rd.Read())
+			{
+				var id      = Convert.ToString(rd["id"]);
+				var colid   = Convert.ToInt32 (rd["colid"]);
+				var colname = Convert.ToString(rd["colname"]);
+
+				columns.Single(_ => _.ID == id && _.Column.ColumnName == colname).Column.PKIndex = colid;
+			}
+		}
+
+		// Load FKs.
+		//
+		s = @"
+			SELECT
+				rc.CONSTRAINT_NAME  as Name, 
+				fk.TABLE_CATALOG + '.' + fk.TABLE_SCHEMA + '.' + fk.TABLE_NAME as ThisTable,
+				fk.COLUMN_NAME      as ThisColumn,
+				pk.TABLE_CATALOG + '.' + pk.TABLE_SCHEMA + '.' + pk.TABLE_NAME as OtherTable,
+				pk.COLUMN_NAME      as OtherColumn,
+				cu.ORDINAL_POSITION as Ordinal
+			FROM
+				INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS rc
+			  JOIN 
+				INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE fk
+					ON
+						rc.CONSTRAINT_CATALOG        = fk.CONSTRAINT_CATALOG AND
+				rc.CONSTRAINT_SCHEMA		 = fk.CONSTRAINT_SCHEMA AND
+						rc.CONSTRAINT_NAME           = fk.CONSTRAINT_NAME
+			  JOIN 
+				INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE pk
+					ON
+						rc.UNIQUE_CONSTRAINT_CATALOG = pk.CONSTRAINT_CATALOG AND
+				rc.UNIQUE_CONSTRAINT_SCHEMA	 = pk.CONSTRAINT_SCHEMA AND
+						rc.UNIQUE_CONSTRAINT_NAME    = pk.CONSTRAINT_NAME
+			  JOIN 
+				INFORMATION_SCHEMA.KEY_COLUMN_USAGE cu
+					ON
+				rc.CONSTRAINT_CATALOG = cu.CONSTRAINT_CATALOG AND
+				rc.CONSTRAINT_SCHEMA = cu.CONSTRAINT_SCHEMA AND
+						rc.CONSTRAINT_NAME = cu.CONSTRAINT_NAME
+			{0}
+			ORDER BY
+				ThisTable,
+				Ordinal";
+
+		if (string.IsNullOrWhiteSpace(OwnerToInclude))
+			cmd.CommandText = string.Format(s, "");
+		else
+			cmd.CommandText = string.Format(s, "WHERE fk.TABLE_SCHEMA = '" + OwnerToInclude + "' ");
+
+		using (var rd = cmd.ExecuteReader())
+		{
+			while (rd.Read())
+			{
+				var name            = Convert.ToString(rd["Name"]);
+				var thisTableID     = Convert.ToString(rd["ThisTable"]);
+				var otherTableID    = Convert.ToString(rd["OtherTable"]);
+				var thisColumnName  = Convert.ToString(rd["ThisColumn"]);
+				var otherColumnName = Convert.ToString(rd["OtherColumn"]);
+
+				var thisTable   = (from t in tables  where t.ID == thisTableID  select t.Table).Single();
+				var otherTable  = (from t in tables  where t.ID == otherTableID select t.Table).Single();
+				var thisColumn  = (from c in columns where c.ID == thisTableID  && c.Column.ColumnName == thisColumnName  select c.Column).Single();
+				var otherColumn = (from c in columns where c.ID == otherTableID && c.Column.ColumnName == otherColumnName select c.Column).Single();
+
+				if (thisTable.ForeignKeys.ContainsKey(name) == false)
+					thisTable.ForeignKeys.Add(name, new ForeignKey { KeyName = name, MemberName = name, OtherTable = otherTable });
+
+				var key = thisTable.ForeignKeys[name];
+
+				key.ThisColumns. Add(thisColumn);
+				key.OtherColumns.Add(otherColumn);
+			}
+		}
+	}
+
+	var qc =
+		from c in columns
+		group c by c.ID into gr
+		join t in tables on gr.Key equals t.ID
+		select new { t.Table, gr };
+
+	foreach (var c in qc)
+	{
+		foreach (var col in from col in c.gr orderby col.Column.ID select col.Column)
+			c.Table.Columns.Add(col.ColumnName, col);
+
+		if (c.Table.Owner == "dbo")
+		{
+			c.Table.Owner = null;
+			Tables.Add(c.Table.TableName, c.Table);
+		}
+		else
+		{
+			Tables.Add(c.Table.Owner + "." + c.Table.TableName, c.Table);
+		}
+	}
+
+	{
+		Usings.Add("System.Collections.Generic");
+		Usings.Add("System.Linq");
+		Usings.Add("System.Linq.Expressions");
+		Usings.Add("System.Reflection");
+		Usings.Add("System.Text");
+		Usings.Add("BLToolkit.Data.DataProvider");
+		Usings.Add("BLToolkit.Data.Sql");
+		Usings.Add("BLToolkit.Data.Sql.SqlProvider");
+
+		var mssqlAfterWriteTableProperty = AfterWriteTableProperty;
+
+		AfterWriteTableProperty = tt =>
+		{
+			mssqlAfterWriteTableProperty(tt);
+
+			tt.WriteLine(@"
+#region FreeTextTable
+
+public class FreeTextKey<T>
+{
+	public T   Key;
+	public int Rank;
+}
+
+class FreeTextTableExpressionAttribute : TableExpressionAttribute
+{
+	public FreeTextTableExpressionAttribute()
+		: base("""")
+	{
+	}
+
+	public override void SetTable(SqlTable table, MemberInfo member, IEnumerable<Expression> expArgs, IEnumerable<ISqlExpression> sqlArgs)
+	{
+		var aargs  = sqlArgs.ToArray();
+		var arr    = ConvertArgs(member, aargs).ToList();
+		var method = (MethodInfo)member;
+		var sp     = new MsSql2008SqlProvider();
+
+		{
+			var ttype  = method.GetGenericArguments()[0];
+			var tbl    = new SqlTable(ttype);
+
+			var database     = tbl.Database     == null ? null : sp.Convert(tbl.Database,     ConvertType.NameToDatabase).  ToString();
+			var owner        = tbl.Owner        == null ? null : sp.Convert(tbl.Owner,        ConvertType.NameToOwner).     ToString();
+			var physicalName = tbl.PhysicalName == null ? null : sp.Convert(tbl.PhysicalName, ConvertType.NameToQueryTable).ToString();
+
+			var name   = sp.BuildTableName(new StringBuilder(), database, owner, physicalName);
+
+			arr.Add(new SqlExpression(name.ToString(), Precedence.Primary));
+		}
+
+		{
+			var field = ((ConstantExpression)expArgs.First()).Value;
+
+			if (field is string)
+			{
+				arr[0] = new SqlExpression(field.ToString(), Precedence.Primary);
+			}
+			else if (field is LambdaExpression)
+			{
+				var body = ((LambdaExpression)field).Body;
+
+				if (body is MemberExpression)
+				{
+					var name = ((MemberExpression)body).Member.Name;
+
+					name = sp.Convert(name, ConvertType.NameToQueryField).ToString();
+
+					arr[0] = new SqlExpression(name, Precedence.Primary);
+				}
+			}
+		}
+
+		table.SqlTableType   = SqlTableType.Expression;
+		table.Name           = ""FREETEXTTABLE({6}, {2}, {3}) {1}"";
+		table.TableArguments = arr.ToArray();
+	}
+}
+
+[FreeTextTableExpressionAttribute]
+public Table<FreeTextKey<TKey>> FreeTextTable<TTable,TKey>(string field, string text)
+{
+	return this.GetTable<FreeTextKey<TKey>>(
+		this,
+		((MethodInfo)(MethodBase.GetCurrentMethod())).MakeGenericMethod(typeof(TTable), typeof(TKey)),
+		field,
+		text);
+}
+
+[FreeTextTableExpressionAttribute]
+public Table<FreeTextKey<TKey>> FreeTextTable<TTable,TKey>(Expression<Func<TTable,string>> fieldSelector, string text)
+{
+	return this.GetTable<FreeTextKey<TKey>>(
+		this,
+		((MethodInfo)(MethodBase.GetCurrentMethod())).MakeGenericMethod(typeof(TTable), typeof(TKey)),
+		fieldSelector,
+		text);
+}
+
+#endregion");
+
+		};
+	}
+}
+#>
\ No newline at end of file