具有代码优先的DbContext Generator T4模板

时间:2014-04-28 17:08:39

标签: entity-framework ef-code-first t4

Database First(和Model First)方法有这个不错的DbContext Generator脚手架文件来生成上下文和模型; Model.Context.tt + Model.tt

由于它们具有内置辅助方法来检索导航属性,因此将它们用于其他目的也非常方便,例如创建控制器,视图等.ASP.NET Scaffolding也做了类似的工作,但在这种情况下,需要为每个模型调用脚手架,相反,这些T4文件一次生成所有文件。

但是,他们只使用“edmx”文件作为输入。是否可以使用/转换它们用于Code First方法?

实体框架版本6.1

1 个答案:

答案 0 :(得分:11)

以下是工作样本 - 2018年6月5日更新:
https://github.com/coni2k/DbContextGeneratorWithCodeFirst

  • Visual Studio 2017
  • 目标框架4.6.1
  • Entity Framework 6.2.0

实现这一目标有几个步骤。

<强> 1。查找并复制EF6.Utility.CS.ttinclude到您的项目

在T4文件(Model.Context.tt + Model.tt)中,inputFile变量(edmx文件)在两个位置使用;

const string inputFile = @"QAModel.edmx";
//var textTransform ...
//var code ...
//var ef ...
//var typeMapper ...
//var fileManager ...
var itemCollection = new EdmMetadataLoader(textTransform.Host, textTransform.Errors).CreateEdmItemCollection(inputFile);
//var codeStringGenerator ...

if (!typeMapper.VerifyCaseInsensitiveTypeUniqueness(typeMapper.GetAllGlobalItems(itemCollection), inputFile))

VerifyCaseInsensitiveTypeUniqueness是一种验证方法,inputFile用作错误的源位置。重要的是EdmMetadataLoader,它来自EF6.Utility.CS.ttinclude文件,该文件在文件开头定义;

<#@ include file="EF6.Utility.CS.ttinclude"#><#@ 

由于需要修改此文件,请找到该文件并复制到项目文件夹中。就我而言,它在这个文件夹下;

%ProgramFiles(x86)%\Microsoft Visual Studio 12.0\Common7\IDE\Extensions\Microsoft\Entity Framework Tools\Templates\Includes\

如您所见,这是一个可选步骤。我们可以修改原始文件,但是使用副本更安全并保留原始文件。

<强> 2。修改ttinclude文件

在include文件中,有三种方法使用edmx文件。

public IEnumerable<GlobalItem> CreateEdmItemCollection(string sourcePath)

public XElement LoadRootElement(string sourcePath)

private void ProcessErrors(IEnumerable<EdmSchemaError> errors, string sourceFilePath)

重要的一个是LoadRootElement方法,它读取xml文件。我们可以从Code First DbContext创建一个内存流,让它从这个流中读取数据,而不是传递物理xml。

一个。将这两个方法添加到包含文件中;

public IEnumerable<GlobalItem> CreateEdmItemCollection(DbContext dbContext)
{
    ArgumentNotNull(dbContext, "dbContext");

    var schemaElement = LoadRootElement(dbContext);
    if (schemaElement != null)
    {
        using (var reader = schemaElement.CreateReader())
        {
            IList<EdmSchemaError> errors;
            var itemCollection = EdmItemCollection.Create(new[] { reader }, null, out errors);

            ProcessErrors(errors, dbContext.Database.Connection.ConnectionString);

            return itemCollection ?? new EdmItemCollection();
        }
    }
    return new EdmItemCollection();
}

public XElement LoadRootElement(DbContext dbContext)
{
    ArgumentNotNull(dbContext, "dbContext");

    XElement root;

    using (var stream = new MemoryStream())
    {
        using (var writer = XmlWriter.Create(stream))
        {
            EdmxWriter.WriteEdmx(dbContext, writer);
        }
        stream.Position = 0;

        root = XElement.Load(stream, LoadOptions.SetBaseUri | LoadOptions.SetLineInfo);
        root = root.Elements()
            .Where(e => e.Name.LocalName == "Runtime")
            .Elements()
            .Where(e => e.Name.LocalName == "ConceptualModels")
            .Elements()
            .Where(e => e.Name.LocalName == "Schema")
            .FirstOrDefault()
                ?? root;
    }

    return root;
}

现在我们在包含文件中使用DbContext,这需要包含System.Data.Entity库。

湾在文件的开头添加这三行;

//<#@ assembly name="%VS120COMNTOOLS%..\IDE\Microsoft.Data.Entity.Design.dll" #>
<#@ assembly name="System.Data.Entity" #>
<#@ import namespace="System.Data.Entity" #>
<#@ import namespace="System.Data.Entity.Infrastructure" #>
//<#@ import namespace="System" #>

第3。通过添加新构造函数

来修改DbContext类

由于T4文件有自己的域,因此无法使用项目的web / app.config文件中的连接字符串初始化DbContext。最简单的解决方法是使用显式连接字符串对其进行初始化。

通过添加一个以连接字符串作为参数的新构造函数来修改您的DBContext类。

public QAContext(string nameOrConnectionString)
    : base(nameOrConnectionString)
{
}

<强> 4。修改T4文件

现在T4已准备好传递Code First DbContext实例。相应地更新文件。

一个。为了能够访问和实例化它,将DbContext类库作为程序集引用添加到T4文件的开头;

//<#@ template language="C#" ...
<#@ assembly name="$(SolutionDir)DbContextGeneratorWithCodeFirst\bin\Debug\DbContextGeneratorWithCodeFirst.dll"#>
//<#@ include file="EF6 ...

湾将inputFile变量替换为connectionString

const string connectionString = @"Server=(LocalDb)\v11.0;Database=QADb;Integrated Security=True;MultipleActiveResultSets=True";
//var textTransform ...

℃。更新CreateEdmItemCollection阻止

//var   fileManager ...
IEnumerable<GlobalItem> itemCollection;
using (var dbContext = new DbContextGeneratorWithCodeFirst.QAContext(connectionString))
{
    itemCollection = new EdmMetadataLoader(textTransform.Host, textTransform.Errors).CreateEdmItemCollection(dbContext);

    if (itemCollection == null)
    {
        return string.Empty;
    }
}
//var codeStringGenerator ...

d。使用VerifyCaseInsensitiveTypeUniqueness参数

更新connectionString方法
if (!typeMapper.VerifyCaseInsensitiveTypeUniqueness(typeMapper.GetAllGlobalItems(itemCollection), connectionString))

现在它已经可以使用了。您可以根据需要修改文件的其余部分,根据您的Code First模型创建任何您想要的文件,如html,javascript,razor等。

这是一项工作,绝对可以改进。例如,include文件可以将DbContext类型作为参数,实例化它,确定它是代码优先还是数据库优先,然后继续处理。或者找到web / app.config文件并从那里读取连接字符串。但是应该足够开始。