CodeSmith生成三层代码之接口层(2)

<%-- 

Name:

Author: 

Description: 

--%>

<%@ CodeTemplate Language="C#" ResponseEncoding="UTF-8" TargetLanguage="Text" Src="" Debug="False" Description="Template description here." %>

<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context"  Description="请选择数据表！" %>

<%@ Property Name="NameSpace" Type="System.String" Default="" Optional="False" Category="Description" Description="请输入命名空间" %>

<%@ Assembly Name="SchemaExplorer" %>

<%@ Import Namespace="SchemaExplorer" %>

/*************************************************************/

/*                   CodeSmith模板生成数据接口                */

/*************************************************************/

using System;

using System.Data;

using System.Collections.Generic;

namespace <%=NameSpace%>.IDAL

{

public interface I<%=SourceTable.Name%>

{

void Add(<%=NameSpace + ".Model." + SourceTable.Name%> model);

void Update(<%=NameSpace + ".Model." + SourceTable.Name%> model);

void Delete(<%=GetPrimaryKeyType(SourceTable) + " " + GetPrimaryKeyName(SourceTable)%>);

List<<%=NameSpace + ".Model." + SourceTable.Name%>> GetList(string strWhere);

DataSet GetData(string strWhere);

<%=NameSpace + ".Model." + SourceTable.Name%> GetModel(<%=GetPrimaryKeyType(SourceTable) + " " + GetPrimaryKeyName(SourceTable)%>);

}

}

<script runat="template">

public string GetPrimaryKeyName(TableSchema table)

{

string primaryKeyName = "";

for(int i=0; i<table.Columns.Count; i++)

{

if(table.Columns[i].IsPrimaryKeyMember)

primaryKeyName = table.Columns[i].Name;

}

return primaryKeyName.ToLower();

}

public string GetPrimaryKeyType(TableSchema table)

{

int temp = 0;

for(int i=0; i<table.Columns.Count; i++)

{

if(table.Columns[i].IsPrimaryKeyMember)

temp = i;

}

string para = "";

if (table.Columns[temp].Name.EndsWith("TypeCode")) 

return table.Columns[temp].Name;

switch (table.Columns[temp].DataType)

{

case DbType.AnsiString: 

para = "string";

break;

case DbType.AnsiStringFixedLength: 

para = "string";

break;

case DbType.Binary: 

para = "byte[]";

break;

case DbType.Boolean: 

para = "bool";

break;

case DbType.Byte: 

para = "int";

break;

case DbType.Currency: 

para = "decimal";

break;

case DbType.Date: 

para = "DateTime";

break;

case DbType.DateTime: 

para = "DateTime";

break;

case DbType.Decimal: 

para = "decimal";

break;

case DbType.Double: 

para = "double";

break;

case DbType.Guid: 

para = "Guid";

break;

case DbType.Int16: 

para = "short";

break;

case DbType.Int32: 

para = "int";

break;

case DbType.Int64:

para = "long";

break;

case DbType.Object: 

para = "object";

break;

case DbType.SByte: 

para = "sbyte";

break;

case DbType.Single: 

para = "float";

break;

case DbType.String: 

para = "string";

break;

case DbType.StringFixedLength: 

para = "string";

break;

case DbType.Time: 

para = "TimeSpan";

break;

case DbType.UInt16: 

para = "ushort";

break;

case DbType.UInt32: 

para = "uint";

break;

case DbType.UInt64: 

para = "ulong";

break;

case DbType.VarNumeric: 

para = "decimal";

break;

default: 

para = "__UNKNOWN__" + table.Columns[temp].NativeType;

break;

}

return para;

}

</script>