[原創]EF架構隨心所欲打造屬于你自己的DbModel

2019-11-14 16:08:26

字體：大中小

來源：轉載

供稿：網友

前言

我們都知道EF可以生成Dbmodel，系統生成的Model有時候并不是我們想要的，如何我們要生成自己的Model，那么久需要我們手動的去修改T4模版，T4是對“Text Template Transformation Toolkit”（4個T）的簡稱。如果你對T4不怎么了解可以去看蔣金楠（Artech）文章從數據到代碼——基于T4的代碼生成方式。

1.0先看看我們要達到的效果圖吧

2.0首先我們來修改T4模版吧

打開T4模版，找到代碼 WriteHeader(codeStringGenerator, fileManager);

我們首先定義變量（圖中黃色代碼為我們自己添加的代碼）

WriteHeader(codeStringGenerator, fileManager);string summary=string.Empty; 定義變量foreach (var entity in typeMapper.GetItemsToGenerate<EntityType>(itemCollection)){    fileManager.StartNewFile(entity.Name + ".cs");    BeginNamespace(code);       if(entity.Documentation !=null && entity.Documentation.Summary!=null)            summary=entity.Documentation.Summary;        else            summary=entity.Name;#><#=codeStringGenerator.UsingDirectives(inHeader: false)#>using System.ComponentModel.DataAnnotations; 導入你需要的命名空間/// <summary>/// <#=            summary#>給類加注釋/// </summary>[Serializable]<#=codeStringGenerator.EntityClassOpening(entity)#>

看看效果圖如下：

類上面的注釋已經加好了，接下來就是刪除構造函數，刪除以下代碼即可：

    public <#=code.Escape(entity)#>()    {<#        foreach (var edmPRoperty in propertiesWithDefaultValues)        {#>        this.<#=code.Escape(edmProperty)#> = <#=typeMapper.CreateLiteral(edmProperty.DefaultValue)#>;<#        }        foreach (var navigationProperty in collectionNavigationProperties)        {#>        this.<#=code.Escape(navigationProperty)#> = new HashSet<<#=typeMapper.GetTypeName(navigationProperty.ToEndMember.GetEntityType())#>>();<#        }        foreach (var complexProperty in complexProperties)        {#>        this.<#=code.Escape(complexProperty)#> = new <#=typeMapper.GetTypeName(complexProperty.TypeUsage)#>();<#        }#>    }

接下來我們把這些可空類型還原成本來面目，已經去掉virtual關鍵字，修改代碼如下:

public string GetTypeName(EdmType edmType, bool? isNullable, string modelNamespace)    {        if (edmType == null)        {            return null;        }        var collectionType = edmType as CollectionType;        if (collectionType != null)        {            return String.Format(CultureInfo.InvariantCulture, "ICollection<{0}>", GetTypeName(collectionType.TypeUsage, modelNamespace));        }        var typeName = _code.Escape(edmType.MetadataProperties                                .Where(p => p.Name == ExternalTypeNameAttributeName)                                .Select(p => (string)p.Value)                                .FirstOrDefault())            ?? (modelNamespace != null && edmType.NamespaceName != modelNamespace ?                _code.CreateFullName(_code.EscapeNamespace(edmType.NamespaceName), _code.Escape(edmType)) :                _code.Escape(edmType));        if (edmType is StructuralType)        {            return typeName;        }        if (edmType is SimpleType)        {            var clrType = UnderlyingClrType(edmType);            if (!IsEnumType(edmType))            {                typeName = _code.Escape(clrType);            }            typeName = FixNamespaces(typeName);            return clrType.IsValueType && isNullable == true ?               // String.Format(CultureInfo.InvariantCulture, "Nullable<{0}>", typeName) :原來的代碼
                String.Format(CultureInfo.InvariantCulture, "{0}?", typeName) :自己修改的代碼
                typeName;        }        throw new ArgumentException("edmType");    }

   public string NavigationProperty(NavigationProperty navigationProperty)    {        var endType = _typeMapper.GetTypeName(navigationProperty.ToEndMember.GetEntityType());        return string.Format(            CultureInfo.InvariantCulture,            "public {1} {2} {{ {3}get; {4}set; }}",            accessibilityAndVirtual(Accessibility.ForProperty(navigationProperty)),            navigationProperty.ToEndMember.RelationshipMultiplicity == RelationshipMultiplicity.Many ? ("ICollection<" + endType + ">") : endType,            _code.Escape(navigationProperty),            _code.SpaceAfter(Accessibility.ForGetter(navigationProperty)),            _code.SpaceAfter(Accessibility.ForSetter(navigationProperty)));    }

接下來來給屬性上添加注釋：(橙色代碼刪除，皇色代碼添加)

/// <summary>/// <#=            summary#>/// </summary>[Serializable]<#=codeStringGenerator.EntityClassOpening(entity)#>{<#    var propertiesWithDefaultValues = typeMapper.GetPropertiesWithDefaultValues(entity);    var collectionNavigationProperties = typeMapper.GetCollectionNavigationProperties(entity);    var complexProperties = typeMapper.GetComplexProperties(entity);    if (propertiesWithDefaultValues.Any() || collectionNavigationProperties.Any() || complexProperties.Any())    {#><#    }刪除掉代碼    var simpleProperties = typeMapper.GetSimpleProperties(entity);    if (simpleProperties.Any())    {        foreach (var edmProperty in simpleProperties)        {                if (edmProperty.Documentation != null && edmProperty.Documentation.Summary != null)                {                    if(!entity.KeyMembers.Contains(edmProperty.Name))                        summary=edmProperty.Documentation.Summary.ToLower().Replace("id","名稱");                    else                        summary=edmProperty.Documentation.Summary;                }                else                {                    summary="";                }#><#                //if(edmProperty.Name.ToLower() == "id")                // continue;var a=edmProperty.Nullable;                                var keyName="";                 if(entity.KeyMembers.Contains(edmProperty.Name))                              keyName="[Key]";                var required="";                if(!edmProperty.Nullable)                    required="[Required(ErrorMessage = /"請輸入{0}/")]";                string facetName = "MaxLength";                var lengthDes="";                var stringLengthDes="";                int maxLength = 0;                 if (code.Escape(edmProperty.TypeUsage).ToLower().Contains("string") && Int32.TryParse(edmProperty.TypeUsage.Facets[facetName].Value.ToString(), out maxLength)){                    lengthDes="[MaxLength("+maxLength+")]";                    stringLengthDes="[StringLength("+maxLength+")]";                }                var dataType="";                if (code.Escape(edmProperty.TypeUsage).ToLower().Contains("datetime"))                    dataType="[DataType(DataType.DateTime)]";                else if (edmProperty.Name.ToLower().Contains("passWord"))                    dataType="[DataType(DataType.Password)]";      #>    /// <summary>    /// <#=                    summary#>    /// </summary><#                if(!string.IsNullOrWhiteSpace(required)){ #>    <#=                        required #><#                    } if(!string.IsNullOrWhiteSpace(summary)){ #>    <#=                        "[Display(Name = /""+summary+"/")]" #><#                    } if(!string.IsNullOrWhiteSpace(lengthDes)){ #>    <#=                        lengthDes #><#                    } if(!string.IsNullOrWhiteSpace(stringLengthDes)){ #>    <#=                        stringLengthDes #><#                    } if(!string.IsNullOrWhiteSpace(dataType)){ #>    <#=                        dataType #><#                    } if(!string.IsNullOrWhiteSpace(keyName)){ #>    <#=                        keyName #><#                } #>    <#=codeStringGenerator.Property(edmProperty)#>

效果基本已經差不多，可是這里為什么沒有注釋，園子里已經有其他文章來處理這個問題：

1.0EF架構~將數據庫注釋添加導入到模型實體類中 2.0entity framework框架生成摘要文檔為空(沒有元數據文檔可用)的bug解決方案

按照步驟做了，可是問題還是沒有解決，怎么辦，其實根本原因是：

主要原因是這里的摘要沒有數據。不斷的嘗試啊，entity framework框架生成摘要文檔為空(沒有元數據文檔可用)的bug解決方案試了幾次還是沒有從根本上解決問題，怎么辦了...

3.0解決bug

沒辦法我們查看EFTSQLDocumentation.Generator的源碼終于找到問題所在

        public String ConnectionString { get; set; }        public String InputFileName { get; set; }        public String OutputFileName { get; set; }        private SqlConnection _connection;        public Program(String connectionString, String inputFileName, String outputFileName)        {            this.ConnectionString = connectionString;            this.InputFileName = inputFileName;            this.OutputFileName = outputFileName;            this._connection = new SqlConnection(connectionString);            this._connection.Open();        }        public void Dispose()        {            this._connection.Dispose();        }        private void CreateDocumentation()        {            XDocument doc = XDocument.Load(this.InputFileName);            IEnumerable<XElement> entityTypeElements = doc.Descendants("{http://schemas.microsoft.com/ado/2008/09/edm}EntityType");            int i = 0;            foreach (XElement entityTypeElement in entityTypeElements)            {                String tableName = entityTypeElement.Attribute("Name").Value;                IEnumerable<XElement> propertyElements = entityTypeElement.Descendants("{http://schemas.microsoft.com/ado/2008/09/edm}Property");                Console.Clear();                Console.WriteLine("Analyzing table {0} of {1}", i++, entityTypeElements.Count());                Console.WriteLine(" => TableName : {0}" +                                  "/n => property count : {1}", tableName, propertyElements.Count());                this.AddNodeDocumentation(entityTypeElement, GetTableDocumentation(tableName));                foreach (XElement propertyElement in propertyElements)                {                    String columnName = propertyElement.Attribute("Name").Value;                    this.AddNodeDocumentation(propertyElement, GetColumnDocumentation(tableName, columnName));                }            }            Console.WriteLine("Writing result to {0}", this.OutputFileName);            if (File.Exists(this.OutputFileName))                File.Delete(this.OutputFileName);            doc.Save(this.OutputFileName);        }        private void AddNodeDocumentation(XElement element, String documentation)        {            if (String.IsNullOrEmpty(documentation))                return;            element.Descendants("{http://schemas.microsoft.com/ado/2008/09/edm}Documentation").Remove();            element.AddFirst(new XElement("{http://schemas.microsoft.com/ado/2008/09/edm}Documentation", new XElement("{http://schemas.microsoft.com/ado/2008/09/edm}Summary", documentation)));        }        private String GetTableDocumentation(String tableName)        {            using (SqlCommand command = new SqlCommand(@" SELECT [value]                                                           FROM fn_listextendedproperty (                                                                'MS_Description',                                                                 'schema', 'dbo',                                                                 'table',  @TableName,                                                                 null, null)", this._connection))            {                command.Parameters.AddWithValue("TableName", tableName);                return command.ExecuteScalar() as String;            }        }        private String GetColumnDocumentation(String tableName, String columnName)        {            using (SqlCommand command = new SqlCommand(@"SELECT [value]                                                          FROM fn_listextendedproperty (                                                                'MS_Description',                                                                 'schema', 'dbo',                                                                 'table', @TableName,                                                                 'column', @columnName)", this._connection))            {                command.Parameters.AddWithValue("TableName", tableName);                command.Parameters.AddWithValue("ColumnName", columnName);                return command.ExecuteScalar() as String;            }        }

我們的edmx中的代碼如下:

      <Schema Namespace="yaochufaNewTestModel.Store" Provider="System.Data.SqlClient" ProviderManifestToken="2008" Alias="Self" xmlns:store="http://schemas.microsoft.com/ado/2007/12/edm/EntityStoreSchemaGenerator" xmlns:customannotation="http://schemas.microsoft.com/ado/2013/11/edm/customannotation" xmlns="http://schemas.microsoft.com/ado/2009/11/edm/ssdl">        <EntityType Name="Activities">          <Key>            <PropertyRef Name="ActivityId" />          </Key>          <Property Name="ActivityId" Type="int" StoreGeneratedPattern="Identity" Nullable="false" />          <Property Name="ActivityType" Type="int" Nullable="false" />          <Property Name="ProvinceId" Type="int" />          <Property Name="CityId" Type="int" />          <Property Name="Description" Type="nvarchar" MaxLength="50" />          <Property Name="IsActive" Type="bit" Nullable="false" />          <Property Name="EndDate" Type="datetime" />          <Property Name="StartDate" Type="datetime" />          <Property Name="DrawDate" Type="datetime" />          <Property Name="DrawInfo" Type="nvarchar" MaxLength="1000" />          <Property Name="LimitTime" Type="int" />          <Property Name="ShowStartDate" Type="datetime" />          <Property Name="ShowEndDate" Type="datetime" />          <Property Name="PrizeCount" Type="int" />          <Property Name="ModifiedById" Type="int" />          <Property Name="ModifiedByName" Type="nvarchar" MaxLength="50" />          <Property Name="ModifiedDate" Type="datetime" />          <Property Name="CreatedById" Type="int" Nullable="false" />          <Property Name="CreatedByName" Type="nvarchar" MaxLength="50" Nullable="false" />          <Property Name="CreatedDate" Type="datetime" Nullable="false" />        </EntityType>

需要修改的就是EFTSQLDocumentation.Generator源碼中的xml命名空間我們要替換成 http://schemas.microsoft.com/ado/2009/11/edm最終在cmd中運行如下代碼：

EFTSQLDocumentation.Generator.exe -c "data source=.;initial catalog=yaochufaNewTest;user id=sa;password=123;" -i " D:/Feng.Test/Feng.Test/Model1.edmx"

得到效果圖如下：