为PetaPoco添加实体模板

Brad为我们提供了T4模板，因为公司一直在使用CodeSmith，故为其写了一个CodeSmith的模板，代码如下：

<%--
Name:EntityTemplates
Author:
Description:Generate a entity file in C#
--%>

<%@ CodeTemplate Language="C#" TargetLanguage="Text" Src="" Inherits="" Debug="False" Description="" ResponseEncoding="UTF-8" %>
<%@ Property Name="Namespace" Type="System.String" Default="TianChenMeiKuang.Entity" Optional="False" Category="Strings" Description="实体类命名空间" %>
<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="源表" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Assembly Name="System.Data" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Import Namespace="System.Data" %>
/**********************************************************
Name：<%= GetClassName(SourceTable) %>
Author：
Date：<%=DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss") %>
Description：
Modify Remark:
**********************************************************/
using System;
using WebApp.Matrix.Data;

namespace <%=Namespace%>
{
/// <summary>
/// This Entity is Mapping To [<%=SourceTable.Name%>] Table
/// Remark Ignore Attribute for the field when it is not need mapping
/// </summary>
[Serializable]
[TableName("[<%=SourceTable.Name%>]")]
<%
ColumnSchema primaryKeyColumn = GetPrimaryKeyColumn();
if(primaryKeyColumn != null)
{
if(Convert.ToBoolean(primaryKeyColumn.ExtendedProperties["CS_isIdentity"].Value)==true){ %>
[PrimaryKey("<%=primaryKeyColumn.Name%>", autoIncrement=true)]
<% }
else {
%>
[PrimaryKey("<%=primaryKeyColumn.Name%>", autoIncrement=false)]
<% }
}%>
public class <%= GetClassName(SourceTable) %>
{
<% for (int i = 0; i < SourceTable.Columns.Count; i++) { %>
/// <summary>
/// <%= SourceTable.Columns[i].Name %>
/// </summary>
public <%= GetCSharpVariableType(SourceTable.Columns[i]) %> <%= GetPropertyName(SourceTable.Columns[i]) %>
{
get; set;
}
<% if (i < SourceTable.Columns.Count - 1) Response.Write("\r\n"); %>
<% } %>

/// <summary>
/// Equals
/// </summary>
public override bool Equals(object obj)
{
<%= GetClassName(SourceTable) %> other = obj as <%= GetClassName(SourceTable) %>;
if (<%=GetFirstKeyCondition()%>)
{
return false;
}
if (<%=GetTwoKeyCondition()%>)
{
return false;
}
return true;
}
/// <summary>
/// GetHashCode
/// </summary>
public override int GetHashCode()
{
return base.GetHashCode();
}
}
}

<script runat="template">
public string GetClassName(TableSchema table)
{
string className = table.Name;
return className;
}
public string GetPropertyName(ColumnSchema column)
{
string propertyName = column.Name;

return propertyName;
}
public string GetFieldName(ColumnSchema column)
{
string propertyName = column.Name;
return propertyName;
}
public ColumnSchema GetPrimaryKeyColumn()
{
for (int i = 0; i < SourceTable.Columns.Count; i++)
{
if(SourceTable.Columns[i].IsPrimaryKeyMember)
{
return     SourceTable.Columns[i];
}
}
return null;
}
public string GetKey()
{
for (int i = 0; i < SourceTable.Columns.Count; i++)
{
if(SourceTable.Columns[i].IsPrimaryKeyMember)
{
return     SourceTable.Columns[i].Name;
}
}
return "GetHashCode().ToString()";
}
public string GetCSharpVariableType(ColumnSchema column)
{
switch (column.DataType)
{
case DbType.AnsiString: return "string";
case DbType.AnsiStringFixedLength: return "string";
case DbType.Binary: return "Nullable<int>";
case DbType.Boolean: if(column.AllowDBNull){return "Nullable<bool>";}else{return "bool";};
case DbType.Byte: return "int";
case DbType.Currency: if(column.AllowDBNull){return "Nullable<decimal>";}else{return "decimal";};
case DbType.Date: if(column.AllowDBNull){return "Nullable<DateTime>";}else{return "DateTime";};
case DbType.DateTime: if(column.AllowDBNull){return "Nullable<DateTime>";}else{return "DateTime";};
case DbType.Decimal: if(column.AllowDBNull){return "Nullable<decimal>";}else{return "decimal";};
case DbType.Double: if(column.AllowDBNull){return "Nullable<double>";}else{return "double";};
case DbType.Guid: return "Guid";
case DbType.Int16: if(column.AllowDBNull){return "Nullable<short>";}else{return "short";};
case DbType.Int32: if(column.AllowDBNull){return "Nullable<int>";}else{return "int";};
case DbType.Int64: if(column.AllowDBNull){return "Nullable<long>";}else{return "long";};
case DbType.Object: return "object";
case DbType.SByte: if(column.AllowDBNull){return "Nullable<sbyte>";}else{return "sbyte";};
case DbType.Single: if(column.AllowDBNull){return "Nullable<float>";}else{return "float";};
case DbType.String: return "string";
case DbType.StringFixedLength: return "string";
case DbType.Time: if(column.AllowDBNull){return "Nullable<TimeSpan>";}else{return "TimeSpan";};
case DbType.UInt16: if(column.AllowDBNull){return "Nullable<ushort>";}else{return "ushort";};
case DbType.UInt32: if(column.AllowDBNull){return "Nullable<uint>";}else{return "uint";};
case DbType.UInt64: if(column.AllowDBNull){return "Nullable<ulong>";}else{return "ulong";};
case DbType.VarNumeric: if(column.AllowDBNull){return "Nullable<decimal>";}else{return "decimal";};
default:return "string";
}
}
public string GetFirstKeyCondition()
{
string condition = " other==null";
for (int i = 0; i < SourceTable.Columns.Count; i++)
{
var tempColumn = SourceTable.Columns[i];
if(tempColumn.IsPrimaryKeyMember)
{
if(((bool)tempColumn.ExtendedProperties["CS_IsIdentity"].Value))
{
condition += " || this." + SourceTable.Columns[i].Name + " == 0";
condition += " || other."+ SourceTable.Columns[i].Name + " == 0";
}
else
{
condition += " || string.IsNullOrEmpty(this." + SourceTable.Columns[i].Name + ")";
condition += " || string.IsNullOrEmpty(other."+ SourceTable.Columns[i].Name + ")";
}
}
}
return condition;
}
public string GetTwoKeyCondition()
{
string condition = " ";
for (int i = 0; i < SourceTable.Columns.Count; i++)
{
if(SourceTable.Columns[i].IsPrimaryKeyMember)
{
condition += " || this."+SourceTable.Columns[i].Name +" !=other."+SourceTable.Columns[i].Name;
}
}
return condition.Substring(4,condition.Length-4).ToString();
}
public string GetHashCodeStr()
{
string hashCode = " ";
for (int i = 0; i < SourceTable.Columns.Count; i++)
{
if(SourceTable.Columns[i].IsPrimaryKeyMember)
{
hashCode += "+\"|\" +this."+SourceTable.Columns[i].Name +".ToLower()";
}
}
return hashCode.Substring(7,hashCode.Length-7).ToString();
}
public string GetDefaultValue(ColumnSchema column)
{
string DefaultValue = "";
switch(column.DataType)
{
case DbType.Int16:
case DbType.Int32:
case DbType.Int64:
DefaultValue = "0";
break;
case DbType.Decimal:
DefaultValue = "0.000000M";
break;
case DbType.AnsiString:
case DbType.String:
case DbType.StringFixedLength:
DefaultValue = "\"\"";
break;
case DbType.Date:
case DbType.DateTime:
case DbType.DateTime2:
DefaultValue = "DateTime.Parse(\"1999-01-01 00:00:00\")";
break;
case DbType.Binary:
DefaultValue = "new byte[] { }";
break;
case DbType.Boolean:
DefaultValue = "False";
break;
case DbType.Byte:
DefaultValue = "1";
break;
default:
break;
}
return DefaultValue;
}
</script>

该模板只适用于但主键的环境，且主键必须为字符串类型，或者为自增长列。

来源：http://www.cnblogs.com/youring2