Mercurial > pub > bltoolkit
view Source/Templates/MSSQL.ttinclude @ 4:f757da6161a1
!bug 100 + 2h fixed gregression
author | cin |
---|---|
date | Sun, 24 Aug 2014 17:57:42 +0400 |
parents | f990fcb411a9 |
children |
line wrap: on
line source
<# 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"); }; } } #>