Ado.Net 实体框架学习笔记2

上一遍我介绍了如何使用引用对象的方法，这一篇我介绍一个工具，用来专门生成上一篇中各个实体类的对应部分类：不多说，直接上代码：

class Program
{
static void Main(string[] args)
{
foreach (var fileName in System.IO.Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory))
{
if (fileName.ToLower().EndsWith(".dll"))
{
Assembly assembly = Assembly.LoadFile(fileName);
EdmRelationshipAttribute[] atrrs = (EdmRelationshipAttribute[])Attribute.GetCustomAttributes(assembly, typeof(EdmRelationshipAttribute));
string newFoldeName = fileName.Remove(fileName.LastIndexOf('.')) + "PartialClass";
var directInfo = System.IO.Directory.CreateDirectory(newFoldeName);

//获取程序集的命名空间
string namespa = (fileName.Replace(AppDomain.CurrentDomain.BaseDirectory, ""));
namespa = namespa.Remove(namespa.LastIndexOf('.'));
foreach (var type in assembly.GetTypes())
{

if (type.IsSubclassOf(typeof(System.Data.Objects.DataClasses.EntityObject)))
{
StringBuilder strBuild = new StringBuilder();
strBuild.Append(@"using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;");
strBuild.AppendLine();
strBuild.AppendLine("namespace " + namespa);
strBuild.AppendLine("{");
strBuild.AppendLine("\t#region " + type.Name);
strBuild.AppendLine("\tpublic partial class " + type.Name);
strBuild.AppendLine("\t{");

if (atrrs != null)
{
foreach (var r in atrrs)
{
if (r.Role2Type == type)
{
//添加注释
strBuild.AppendLine("\t\t/// <summary>");
strBuild.AppendLine("\t\t/// 获取" + r.Role1Type.Name);
strBuild.AppendLine("\t\t/// </summary>");
//添加方法
strBuild.AppendLine("\t\tpublic " + r.Role1Type.Name + " Get" + r.Role1Type.Name + "()");
strBuild.AppendLine("\t\t{");
strBuild.AppendLine("\t\t\tif (this." + r.Role1Type.Name + " == null)");
strBuild.AppendLine("\t\t\t\tthis." + r.Role1Type.Name + "Reference.Load();");
strBuild.AppendLine("\t\t\t return this." + r.Role1Type.Name + ";");
strBuild.AppendLine("\t\t}");

}
}
}
strBuild.AppendLine("\t}");
strBuild.AppendLine("\t#endregion");
strBuild.AppendLine();
strBuild.AppendLine("}");
using (var stream = System.IO.File.Create(System.IO.Path.Combine(directInfo.Name, type.Name + ".PartialClass.cs")))
{
byte[] buffer = System.Text.Encoding.UTF8.GetBytes(strBuild.ToString());
stream.Write(buffer, 0, buffer.Length);
stream.Close();
}
}

}

}
}

}
}

使用方法是，直接将实体框架的实体类项目编译成dll，然后将dll放在本工具的同级目录下就可以了