使用CodeSmith生成数据实体类

最新推荐文章于 2023-06-17 16:24:42 发布

麻烦的一笔

最新推荐文章于 2023-06-17 16:24:42 发布

阅读量2k

点赞数 1

分类专栏： aspnet CSharp

本文链接：https://blog.youkuaiyun.com/qwe8254/article/details/51798726

版权

aspnet 同时被 2 个专栏收录

12 篇文章

订阅专栏

CSharp

5 篇文章

订阅专栏

本文介绍如何利用CodeSmith工具快速生成C#模型类。通过设置模板参数，可以自动生成包含属性及getter和setter方法的类文件，适用于从数据库表结构生成对应的C#实体类。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先到网上下载codesmith工具（必须是破解版）

然后在CodeSmithStudio新建一个Csharp的Template

最后拷贝代码：

<%@ CodeTemplate Language="C#" TargetLanguage="C#" Description="Template description here." %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" %>
<%@ Property Name="NameSpace" Type="String" Category="参数" Description="命名空间" Default="" Optional="True"%>
<%@ Import Namespace="System.Text.RegularExpressions" %>
using System;
using System.Collections.Generic;
using System.Text;

namespace <%=NameSpace%>
{
    [Serializable]
    public class <%=GetClassName(SourceTable)+"Model" %>
    {
        <% foreach (ColumnSchema column in SourceTable.Columns) {%>
        <% if (column.IsForeignKeyMember) {%>
        private <%=GetPascalName(GetForeignKeyColumnType(column))%> <%=GetCamelName(GetForeignKeyColumnName(column))%> = new <%=GetPascalName(GetForeignKeyColumnType(column))%>();
        <% }else{%>
        private <%=GetCSharpVariableType(column)%> <%=GetCamelName(column)%>;
        <% }%>
        <% }%>    
        
        <% foreach (ColumnSchema column in SourceTable.Columns) {%>
        <% if (column.IsForeignKeyMember) {%>
        public <%=GetPascalName(GetForeignKeyColumnType(column)) %> <%=GetPascalName(GetForeignKeyColumnName(column))%>
        {
            get { return <%=GetCamelName(GetForeignKeyColumnName(column)) %>; }
            set { <%=GetCamelName(GetForeignKeyColumnName(column)) %> = value; }
        }
        <% } else {%>
        public <%=GetCSharpVariableType(column) %> <%=GetPascalName(column) %>
        {
            get { return <%= GetCamelName(column) %>; }
            set { <%= GetCamelName(column) %> = value; }
        }
        <% }%>
        <% }%>
    }
}
<script runat="template">

public string MakeSingle(string name)
{
    return name;
}
public string GetCamelName(ColumnSchema column)
{
    return column.Name.Substring(0, 1).ToLower() + column.Name.Substring(1);
}
public string GetCamelName(string value)
{
    return value.Substring(0, 1).ToLower() + value.Substring(1);
}
public string GetPascalName(ColumnSchema column)
{
    return column.Name.Substring(0, 1).ToUpper() + column.Name.Substring(1);
}
public string GetPascalName(string value)
{
    return value.Substring(0, 1).ToUpper() + value.Substring(1);
}
public string GetClassName(TableSchema table)
{
    return GetPascalName(MakeSingle(table.Name));
}    
public string GetForeignKeyColumnType(ColumnSchema column)
{        
    return column.Table.ForeignKeys[0].PrimaryKeyTable.Name;
}
public string GetForeignKeyColumnName(ColumnSchema column)
{    
    if(column.Name.Substring(column.Name.Length-2).ToLower() == "id")
    {
        return column.Name.Substring(0,column.Name.Length-2);
    }
    else
    {
        return column.Name;
    }
}

public string GetStructureParameter()
{//int id,string name
        string temp = "";
        foreach(ColumnSchema column in SourceTable.NonPrimaryKeyColumns) 
        {
                if(column.IsForeignKeyMember) 
                {
                    temp += GetPascalName(GetForeignKeyColumnType(column)) + " " +GetCamelName(GetForeignKeyColumnName(column)) + ",";
                }
                else
                {
                    temp += GetCSharpVariableType(column) + " " + GetCamelName(column) + ",";
                }
        }
        return temp.Substring(0,temp.Length-1);
}
public string GetPrimaryKeyType(TableSchema table)
{
    if (table.PrimaryKey != null)
    {
        if (table.PrimaryKey.MemberColumns.Count == 1)
        {
            return GetCSharpVariableType(table.PrimaryKey.MemberColumns[0]);
        }
        else
        {
            throw new ApplicationException("This template will not work on primary keys with more than one member column.");
        }
    }
    else
    {
        throw new ApplicationException("This template will only work on tables with a primary key.");
    }
}
public string GetCSharpVariableType(ColumnSchema column)
{
    if (column.Name.EndsWith("TypeCode")) return column.Name;
    
    switch (column.DataType)
    {
        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.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";
        default:
        {
            return "__UNKNOWN__" + column.NativeType;
        }
    }
}    
public override string GetFileName()
{
    return     GetClassName(SourceTable) + ".cs";
}
</script>