学步园

using System;<br /> using System.Text;<br /> using System.ComponentModel;<br /> using CodeSmith.Engine;<br /> using System.Data;<br /> using SchemaExplorer;</p> <p>namespace CodeTemplateRule<br /> {<br /> /**//// <summary><br /> /// CodeSmith自动生成代码所用到的函数<br /> /// </summary><br /> public class CodeTemplateRule<br /> {<br /> /**//// <summary><br /> /// 获取所有列的集合<br /> /// </summary><br /> /// <param name="dataBase">数据库Schema</param><br /> /// <param name="tableName">表名</param><br /> /// <returns>Columns Schema</returns><br /> public ColumnSchemaCollection GetColumnCollection(DatabaseSchema dataBase,string tableName)<br /> {<br /> TableSchemaCollection tables = new TableSchemaCollection(dataBase.Tables);<br /> ColumnSchemaCollection columns=null;<br /> for(int i=0;i<tables.Count;i++)<br /> {<br /> if(tables[i].Name.ToUpper()==tableName.ToUpper())<br /> {<br /> TableSchema ts=tables[i];<br /> columns=new ColumnSchemaCollection(ts.Columns);<br /> }<br /> }<br /> return columns;<br /> }</p> <p> /**//// <summary><br /> /// 获取处理后的表名 例如：ORG_Person,返回Person<br /> /// </summary><br /> /// <param name="tableName">表名</param><br /> /// <returns>返回处理后的表名</returns><br /> public string GetTableName(string tableName)<br /> {<br /> int i=tableName.IndexOf("_");<br /> return tableName.Substring(i+1,tableName.Length-(i+1));<br /> }</p> <p> /**//// <summary><br /> /// 获取唯一号的变量名，第一个字母小写<br /> /// </summary><br /> /// <param name="guidColumn">唯一号列名</param><br /> /// <returns>唯一号的变量名</returns><br /> public string GetGuidColumnVar(string guidColumn)<br /> {<br /> string guid;<br /> guid=guidColumn.Substring(0,1).ToLower()+guidColumn.Substring(1,guidColumn.Length-1);<br /> return guid;<br /> } </p> <p> /**//// <summary><br /> /// 获取c#变量类型<br /> /// </summary><br /> /// <param name="column">列</param><br /> /// <returns>c#变量类型</returns><br /> public string GetType(ColumnSchema column)<br /> {<br /> if (column.Name.EndsWith("TypeCode")) return column.Name;</p> <p> switch (column.DataType)<br /> {<br /> case DbType.AnsiString: return "string";<br /> case DbType.AnsiStringFixedLength: return "string";<br /> case DbType.Binary: return "byte[]";<br /> case DbType.Boolean: return "bool";<br /> case DbType.Byte: return "int";<br /> case DbType.Currency: return "decimal";<br /> case DbType.Date: return "DateTime";<br /> case DbType.DateTime: return "DateTime";<br /> case DbType.Decimal: return "decimal";<br /> case DbType.Double: return "double";<br /> case DbType.Guid: return "Guid";<br /> case DbType.Int16: return "short";<br /> case DbType.Int32: return "int";<br /> case DbType.Int64: return "long";<br /> case DbType.Object: return "object";<br /> case DbType.SByte: return "sbyte";<br /> case DbType.Single: return "float";<br /> case DbType.String: return "string";<br /> case DbType.StringFixedLength: return "string";<br /> case DbType.Time: return "TimeSpan";<br /> case DbType.UInt16: return "ushort";<br /> case DbType.UInt32: return "uint";<br /> case DbType.UInt64: return "ulong";<br /> case DbType.VarNumeric: return "decimal";<br /> default:<br /> {<br /> return "__UNKNOWN__" + column.NativeType;<br /> }<br /> }<br /> }</p> <p> /**//// <summary><br /> /// 获取c#变量类型默认值<br /> /// </summary><br /> /// <param name="column">列</param><br /> /// <returns>c#变量类型默认值</returns><br /> public string GetDefaultValue(ColumnSchema column)<br /> {<br /> if (column.Name.EndsWith("TypeCode")) return column.Name;</p> <p> switch (column.DataType)<br /> {<br /> case DbType.AnsiString: return "/"/"";<br /> case DbType.AnsiStringFixedLength: return "/"/"";<br /> case DbType.Binary: return "null";<br /> case DbType.Boolean: return "false";<br /> case DbType.Byte: return "0";<br /> case DbType.Currency: return "0";<br /> case DbType.Date: return "DateTime.Parse(/"1900-1-1/")";<br /> case DbType.DateTime: return "DateTime.Parse(/"1900-1-1/")";<br /> case DbType.Decimal: return "0";<br /> case DbType.Double: return "0";<br /> case DbType.Guid: return "Guid.NewGuid().ToString()";<br /> case DbType.Int16: return "0";<br /> case DbType.Int32: return "0";<br /> case DbType.Int64: return "0";<br /> case DbType.Object: return "/"/"";<br /> case DbType.SByte: return "0";<br /> case DbType.Single: return "0";<br /> case DbType.String: return "/"/"";<br /> case DbType.StringFixedLength: return "";<br /> case DbType.Time: return "DateTime.Parse(/"1900-1-1/")";<br /> case DbType.UInt16: return "0";<br /> case DbType.UInt32: return "0";<br /> case DbType.UInt64: return "0";<br /> case DbType.VarNumeric: return "0";<br /> default:<br /> {<br /> return "__UNKNOWN__" + column.NativeType;<br /> }<br /> }<br /> }</p> <p> /**//// <summary><br /> /// 获取c#变量类型默认值<br /> /// </summary><br /> /// <param name="column">列</param><br /> /// <returns>c#变量类型默认值</returns><br /> public string GetConvert(ColumnSchema column)<br /> {<br /> if (column.Name.EndsWith("TypeCode")) return column.Name;</p> <p> switch (column.DataType)<br /> {<br /> case DbType.AnsiString: return "Convert.ToString";<br /> case DbType.AnsiStringFixedLength: return "Convert.ToString";<br /> case DbType.Binary: return "Convert.ToByte";<br /> case DbType.Boolean: return "Convert.ToBoolean";<br /> case DbType.Byte: return "Convert.ToInt32";<br /> case DbType.Currency: return "Convert.ToDecimal";<br /> case DbType.Date: return "Convert.DateTime";<br /> case DbType.DateTime: return "Convert.DateTime";<br /> case DbType.Decimal: return "Convert.ToDecimal";<br /> case DbType.Double: return "Convert.ToDouble";<br /> case DbType.Guid: return "Convert.ToString";<br /> case DbType.Int16: return "Convert.ToInt16";<br /> case DbType.Int32: return "Convert.ToInt32";<br /> case DbType.Int64: return "Convert.ToInt64";<br /> case DbType.Object: return "Convert.ToString";<br /> case DbType.SByte: return "Convert.ToByte";<br /> case DbType.Single: return "Convert.ToInt32";<br /> case DbType.String: return "Convert.ToString";<br /> case DbType.StringFixedLength: return "Convert.ToString";<br /> case DbType.Time: return "Convert.DateTime";<br /> case DbType.UInt16: return "Convert.ToUInt16";<br /> case DbType.UInt32: return "Convert.ToUInt32";<br /> case DbType.UInt64: return "Convert.ToUInt64";<br /> case DbType.VarNumeric: return "Convert.ToDecimal";<br /> default:<br /> {<br /> return "__UNKNOWN__" + column.NativeType;<br /> }<br /> }<br /> }<br /> }<br /> }