CodeSmith使用总结--调用自定义方法

上一篇读取了一个表的内容，但是到了真正应用的时候还是不够用的，我们很容易可以对比出来，SQL里边的数据类型的定义和C#中有所不同，比如Saler--String，大写的String在C#中不是一个类型，所以在这里我们要进行转换一下，还有有的时候我们要加一些操作的逻辑，但这个逻辑不只在一个地方会使用到，这时候我们就可以建一个方法来调用。

在CodeSmith中创建方法要把方法放在<script runat="template">……</script>这个标签中，template我理解为运行在模板中的，所以他可以用来定义方法。如下方式：

<script runat="template">

public string Test()

{

    return "测试";

}

</script>

这样就创建了一个Test的方法，调用的时候和之前一样使用<%= Test() %>，需要注意的是，这里括号后边没有分号，<%= Test(); %>是不对的。

下面是一个数据库类型转C#数据类型的方法，这个是从网上搜集来的。

/// <summary>
/// 将数据库类型转化为C#类型
/// </summary>
/// <param name="dbType">数据库类型</param>
/// <returns>C#类型</returns>
public string DataTypeToCSharpType(System.Data.DbType dbType)
{
    switch (dbType)
    {
        case DbType.AnsiString:
            return "string";
        case DbType.AnsiStringFixedLength:
            return "string";
        case DbType.Binary:
            return "byte[]";
        case DbType.Boolean:
            return "bool";
        case DbType.Byte:
            return "byte";
        case DbType.Currency:
            return "decimal";
        case DbType.Date:
            return "DateTime";
        case DbType.DateTime:
            return "DateTime";
        case DbType.DateTime2:
            return "DateTime";
        case DbType.DateTimeOffset:
            return "DateTime";
        case DbType.Decimal:
            return "decimal";
        case DbType.Double:
            return "double";
        case DbType.Guid:
            return "Guid";
        case DbType.Int16:
            return "short";
        case DbType.Int32:
            return "int";
        case DbType.Int64:
            return "long";
        case DbType.Object:
            return "object";
        case DbType.SByte:
            return "sbyte";
        case DbType.Single:
            return "float";
        case DbType.String:
            return "string";
        case DbType.StringFixedLength:
            return "string";
        case DbType.Time:
            return "TimeSpan";
        case DbType.UInt16:
            return "ushort";
        case DbType.UInt32:
            return "uint";
        case DbType.UInt64:
            return "ulong";
        case DbType.VarNumeric:
            return "decimal";
        case DbType.Xml:
            return "string";
        default:
            return "object";
    }
}

有了它我们就可以像这样调用了

<%foreach(ColumnSchema col in SourceTable.Columns){ %>

public <%= DataTypeToCSharpType(col.DataType) %> <%= col.Name %> { get;set; }

<% } %>

本篇示例模板：

<%-- 

Name: 测试模板3

Author: GodFinal

Description: 

--%>

<%@ Template Language="C#" TargetLanguage="SQL" Description="连接SQL数据库，并读取表结构" Debug="True"%>

<%@ Assembly Name="SchemaExplorer" %>

<%@ Import Namespace="SchemaExplorer" %>

<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="表" Description="要操作的表" %>

using System;

 

namespace GodFinal

{

    /// <summary>

    /// <%=ClassName %>[实体]

    /// </summary>

public class <%=ClassName %> 

{

        <%foreach(ColumnSchema col in SourceTable.Columns){ %>

        public <%= DataTypeToCSharpType(col.DataType) %> <%= col.Name %> { get;set; }

        <% } %>

    }

}

<script runat="template"> 

/// <summary>

/// 将数据库类型转化为C#类型

/// </summary>

/// <param name="dbType">数据库类型</param>

/// <returns>C#类型</returns>

public string DataTypeToCSharpType(System.Data.DbType dbType)

{

    switch (dbType)

    {

        case DbType.AnsiString:

            return "string";

        case DbType.AnsiStringFixedLength:

            return "string";

        case DbType.Binary:

            return "byte[]";

        case DbType.Boolean:

            return "bool";

        case DbType.Byte:

            return "byte";

        case DbType.Currency:

            return "decimal";

        case DbType.Date:

            return "DateTime";

        case DbType.DateTime:

            return "DateTime";

        case DbType.DateTime2:

            return "DateTime";

        case DbType.DateTimeOffset:

            return "DateTime";

        case DbType.Decimal:

            return "decimal";

        case DbType.Double:

            return "double";

        case DbType.Guid:

            return "Guid";

        case DbType.Int16:

            return "short";

        case DbType.Int32:

            return "int";

        case DbType.Int64:

            return "long";

        case DbType.Object:

            return "object";

        case DbType.SByte:

            return "sbyte";

        case DbType.Single:

            return "float";

        case DbType.String:

            return "string";

        case DbType.StringFixedLength:

            return "string";

        case DbType.Time:

            return "TimeSpan";

        case DbType.UInt16:

            return "ushort";

        case DbType.UInt32:

            return "uint";

        case DbType.UInt64:

            return "ulong";

        case DbType.VarNumeric:

            return "decimal";

        case DbType.Xml:

            return "string";

        default:

            return "object";

    }

}

</script>

最终结果：

using System;

 

namespace GodFinal

{

    /// <summary>

    /// tb_sys_user[实体]

    /// </summary>

　　public class tb_sys_user 

　　{

        public int id { get;set; }

        public string CPlace { get;set; }

        public string Saler { get;set; }

        public string Company { get;set; }

        public string Department { get;set; }

        public string Account { get;set; }

        public string telephone { get;set; }

        public string email { get;set; }

        public string Address { get;set; }

        public string Zipcode { get;set; }

        public string Level { get;set; }

    }

}