在T4中使用一些来自托管项目的类

本文关键字:项目 T4 | 更新日期: 2023-09-27 18:07:19

我在一个项目中有Linq to sql模型,其中.tt也添加了。我想在T4模板中使用这个模型。问题是如何在上面加上参考。

在T4中使用一些来自托管项目的类

让我看看我是否理解对了。

  1. 你有一个项目与Linq2Sql模型
  2. 你想要添加一个。tt文件到这个项目,使用上面的模型来生成一些东西

如果这是正确的,我要做的是加载。dbml文件(它是xml)并从中生成工件。

可能是这样的(从DataClasses1.dbml生成一些类的VS2010模板):

<#@ template    language="C#" hostspecific="true"   #>
<#@ assembly    name="System.Core"                  #>
<#@ assembly    name="System.Xml"                   #>
<#@ assembly    name="System.Xml.Linq"              #>
<#@ import      nameSpace="System.Linq"             #>
<#@ import      nameSpace="System.Xml.Linq"         #>
namespace MyProgram
{
    using System.Data.Linq.Mapping;
<#
    const string ns             = "{http://schemas.microsoft.com/linqtosql/dbml/2007}";
    const string DatabaseName   = ns + "Database";
    const string TableName      = ns + "Table";
    const string TypeName       = ns + "Type";
    const string ColumnName     = ns + "Column";
    var xdoc = XDocument.Load (Host.ResolvePath ("DataClasses1.dbml"));
    var tables = xdoc.Elements (DatabaseName).Elements (TableName);
    foreach (var table in tables)
    {
        var types = table.Elements (TypeName);
        foreach (var @type in types)
        {
            var columns = @type.Elements (ColumnName);
#>
    [Table (Name = "<#=GetAttribute (@type, "Name")#>")]
    partial class <#=GetAttribute (@type, "Name")#>
    {
<#
            foreach (var column in columns)
            {
#>
        [Column (DbType = "<#=GetAttribute (column, "DbType")#>")]
        public <#=GetAttribute (column, "Type")#> <#=GetAttribute (column, "Name")#> { get; set; } 
<#
            }
#>
    }
<#
        }
    }
#>
}
<#+
    static string GetAttribute (XElement element, string name, string defaultValue = null)
    {
        if (defaultValue == null)
        {
            defaultValue = "<" + name + "_attribute_not_found>";
        }
        if (element == null)
        {
            return defaultValue;
        }
        var attribute = element.Attribute (name ?? "");
        if (attribute != null)
        {
            return attribute.Value ?? defaultValue;
        }
        else
        {
            return defaultValue;
        }
    }
#>

您应该会发现这很有帮助:

http://www.olegsych.com/2008/02/t4-assembly-directive/