EntityFramework Core 2.x/3.x (ef core) 在迁移中自动生成数据库表和列说明
前言
在项目开发中有没有用过拼音首字母做列名或者接手这样的项目?
看见xmspsqb(项目审批申请表)这种表名时是否有一种无法抑制的想肛了取名的老兄的冲动?
更坑爹的是这种数据库没有文档(或者文档老旧不堪早已无用)也没有数据库内部说明,是不是很无奈?
但是,凡事就怕有但是,有些表和列名字确实太专业(奇葩),用英文不是太长就是根本不知道用什么(英文差……),似乎也只能用拼音。好吧,那就用吧,写个说明凑活用用。这个时候问题就来了,如何用sql生成表和列说明?在ef core中又怎样生成表和列说明?
正文
基础准备
以sqlserver为例。
- 使用ssms管理器编辑说明:不瞎的都知道吧。
- 使用sql生成说明:
-- 添加
exec sys.sp_addextendedproperty
@name=N'MS_Description'
, @value=N'<说明>'
, @level0type=N'SCHEMA'
, @level0name=N'<dbo>'
, @level1type=N'TABLE'
, @level1name=N'<表名>'
, @level2type=N'COLUMN'
, @level2name=N'<列名>'
尖括号中的内容根据情况修改,需要注意,如果说明已经存在会报错(MSDN的描述为返回值0为成功,1为失败,并不抛出异常,为避免麻烦,我选择先判断再操作,以下相同)。如果需要添加的是表说明,那么@level2type
和@level2name
填NULL
即可。
-- 删除
exec sys.sp_dropextendedproperty
@name=N'MS_Description'
, @level0type=N'SCHEMA'
, @level0name=N'<dbo>'
, @level1type=N'TABLE'
, @level1name=N'<表名>'
, @level2type=N'COLUMN'
, @level2name=N'<列名>'
很好,只需要这两个内置存储过程就可以用sql管理说明了,修改虽然也有,但是先删再加也一样,就不写了。还有一个遗留问题,上面的存储过程可能会报错,后面的sql就得不到执行,为避免麻烦,这需要解决一下,思路很直接,查询下是否存在,存在的话先删再加,不存在就直接加。
-- 查询说明是否存在
select exists (
select t.name as tname,c.name as cname, d.value as Description
from sysobjects t
left join syscolumns c
on c.id=t.id and t.xtype='U' and t.name <> 'dtproperties'
left join sys.extended_properties d
on c.id=d.major_id and c.colid=d.minor_id and d.name = 'MS_Description'
where t.name = '<表名>' and c.name = '<列名>' and d.value is not null)
尖括号中的内容根据情况修改,如果要查询的是表说明,删除and c.name = '<列名>'
的部分即可。
EF Core 扩展
不错,判断问题也解决了,直接使用sql管理基本上也就够用了,那么如果使用ef core托管数据库该怎么办呢?思路也很清晰,使用ef迁移。在迁移中管理sql,MigrationBuilder.Sql(string sql)
方法可以在迁移中执行任何自定义sql。把上面的sql传给方法就可以让ef迁移自动生成说明,并且生成的独立迁移脚本文件也包含说明相关sql。
问题看似解决了,但是(又是但是),这个解决方案实在是太难用了。
- 这样的字符串没有智能提示和代码着色,怎么写错的都不知道。
- 后续管理困难,很可能跟数据库文档一样的下场,没人维护更新,随着版本推移逐渐沦为垃圾。
- 不好用!不优雅!
接下来就是个人思考尝试后得到的解决方案:
- 将说明分散到说明对象的脸上,让查阅和修改都能随手完成,降低维护成本,利用C#的特性可以优雅的解决这个问题。
[AttributeUsage(AttributeTargets.Class | AttributeTargets.Enum | AttributeTargets.Property | AttributeTargets.Field, Inherited = true, AllowMultiple = false)]
public class DbDescriptionAttribute : Attribute
{
/// <summary>
/// 初始化新的实例
/// </summary>
/// <param name="description">说明内容</param>
public DbDescriptionAttribute(string description) => Description = description;
/// <summary>
/// 说明
/// </summary>
public virtual string Description { get; }
}
- 读取特性并应用到迁移中。不过我并不打算让迁移直接读取特性,首先在迁移过程中实体类型并不会载入,从模型获取实体类型结果是null,需要自己想办法把模型类型传入迁移。其次我希望迁移能时刻与模型匹配,ef迁移会生成多个迁移类代码,追踪整个实体模型的变更历史,而特性一旦修改,就会丢失旧的内容,无法充分利用ef迁移的跟踪能力。基于以上考虑,可以把模型的说明写入模型注解,ef迁移会将模型注解写入迁移快照。最后就是在适当的时机读取特性并写入注解,很显然,这个时机就是
OnModelCreating
方法。
public static ModelBuilder ConfigDatabaseDescription(this ModelBuilder modelBuilder)
{
foreach (var entityType in modelBuilder.Model.GetEntityTypes())
{
//添加表说明
if (entityType.FindAnnotation(DbDescriptionAnnotationName) == null && entityType.ClrType?.CustomAttributes.Any(
attr => attr.AttributeType == typeof(DbDescriptionAttribute)) == true)
{
entityType.AddAnnotation(DbDescriptionAnnotationName,
(entityType.ClrType.GetCustomAttribute(typeof(DbDescriptionAttribute)) as DbDescriptionAttribute
)?.Description);
}
//添加列说明
foreach (var property in entityType.GetProperties())
{
if (property.FindAnnotation(DbDescriptionAnnotationName) == null && property.PropertyInfo?.CustomAttributes
.Any(attr => attr.AttributeType == typeof(DbDescriptionAttribute)) == true)
{
var propertyInfo = property.PropertyInfo;
var propertyType = propertyInfo?.PropertyType;
//如果该列的实体属性是枚举类型,把枚举的说明追加到列说明
var enumDbDescription = string.Empty;
if (propertyType.IsEnum
|| (propertyType.IsDerivedFrom(typeof(Nullable<>)) && propertyType.GenericTypeArguments[0].IsEnum))
{
var @enum = propertyType.IsDerivedFrom(typeof(Nullable<>))
? propertyType.GenericTypeArguments[0]
: propertyType;
var descList = new List<string>();
foreach (var field in @enum?.GetFields() ?? new FieldInfo[0])
{
if (!field.IsSpecialName)
{
var desc = (field.GetCustomAttributes(typeof(DbDescriptionAttribute), false)
.FirstOrDefault() as DbDescriptionAttribute)?.Description;
descList.Add(
$@"{field.GetRawConstantValue()} : {(desc.IsNullOrWhiteSpace() ? field.Name : desc)}");
}
}
var isFlags = @enum?.GetCustomAttribute(typeof(FlagsAttribute)) != null;
var enumTypeDbDescription =
(@enum?.GetCustomAttributes(typeof(DbDescriptionAttribute), false).FirstOrDefault() as
DbDescriptionAttribute)?.Description;
enumTypeDbDescription += enumDbDescription + (isFlags ? " [是标志位枚举]" : string.Empty);
enumDbDescription =
$@"( {(enumTypeDbDescription.IsNullOrWhiteSpace() ? "" : $@"{enumTypeDbDescription}; ")}{string.Join("; ", descList)} )";
}
property.AddAnnotation(DbDescriptionAnnotationName,
$@"{(propertyInfo.GetCustomAttribute(typeof(DbDescriptionAttribute)) as DbDescriptionAttribute)
?.Description}{(enumDbDescription.IsNullOrWhiteSpace() ? "" : $@" {enumDbDescription}")}");
}
}
}
return modelBuilder;
}
在OnModelCreating
方法中调用ConfigDatabaseDescription
方法即可将说明写入模型注解。其中的关键是AddAnnotation
这个ef core提供的API,不清楚1.x和ef 6.x有没有这个功能。其中DbDescriptionAnnotationName
就是个名称,随便取,只要不和已有注解重名即可。可以看到,这个方法同时支持扫描并生成枚举类型的说明,包括可空枚举。
- 在迁移中读取模型注解并生成说明。有了之前的准备工作,到这里就好办了。
public static MigrationBuilder ApplyDatabaseDescription(this MigrationBuilder migrationBuilder, Migration migration)
{
var defaultSchema = "dbo";
var descriptionAnnotationName = ModelBuilderExtensions.DbDescriptionAnnotationName;
foreach (var entityType in migration.TargetModel.GetEntityTypes())
{
//添加表说明
var tableName = entityType.Relational().TableName;
var schema = entityType.Relational().Schema;
var tableDescriptionAnnotation = entityType.FindAnnotation(descriptionAnnotationName);
if (tableDescriptionAnnotation != null)
{
migrationBuilder.AddOrUpdateTableDescription(
tableName,
tableDescriptionAnnotation.Value.ToString(),
schema.IsNullOrEmpty() ? defaultSchema : schema);
}
//添加列说明
foreach (var property in entityType.GetProperties())
{
var columnDescriptionAnnotation = property.FindAnnotation(descriptionAnnotationName);
if (columnDescriptionAnnotation != null)
{
migrationBuilder.AddOrUpdateColumnDescription(
tableName,
property.Relational().ColumnName,
columnDescriptionAnnotation.Value.ToString(),
schema.IsNullOrEmpty() ? defaultSchema : schema);
}
}
}
return migrationBuilder;
}
在迁移的Up
和Down
方法末尾调用ApplyDatabaseDescription
方法即可取出模型注解中的说明并生成和执行相应的sql。
至此,一个好用的数据库说明管理就基本完成了。因为这个方法使用了大量ef core提供的API,所以基本上是完整支持ef core的各种实体映射,实测包括与实体类名、属性名不一致的表名、列名,(嵌套的)Owned类型属性(类似ef 6.x的复杂类型属性 Complex Type)、表拆分等。可以说基本上没有什么后顾之忧。这里的sql是以sqlserver为例,如果使用的是mysql或其他关系数据库,需要自行修改sql以及AddOrUpdateColumnDescription
和AddOrUpdateTableDescription
的逻辑。
其中Owned类型属性在生成迁移时可能会生成错误代码,导致编译错误CS1061 "ReferenceOwnershipBuilder"未包含"HasAnnotation"的定义且……;,只需要把HasAnnotation
替换成HasEntityTypeAnnotation
即可。估计是微软的老兄粗心没注意这个问题。(貌似EF Core 3.0后这个问题消失了,HasEntityTypeAnnotation
方法被删除)
ps:为什么不直接使用Description
或者DisplayName
之类的内置特性而要使用自定义特性,因为Description
在语义上是指广泛的说明,并不能明确表明这是数据库说明,同时避免与现存代码纠缠不清影响使用,为加强语义性,使用新增的自定义特性。
ps2:为什么不使用xml注释文档,因为这会让这个功能产生对/doc编译选项和弱类型文本的依赖,甚至需要对文档配置嵌入式资源,也会增加编码难度,同时会影响现存代码的xml注释,为避免影响现存代码,对非代码和编译器不可检查行为的依赖,保证代码健壮性,不使用xml文档注释。
效果预览
更新
2019-12-14
已更新 EF core 3.x 版本代码,具体代码请看 Github 项目库中的 NetCore_3.0 分支(实际上已经更新到 .Net Core 3.1,懒得改名了。好麻烦 (¬_¬") )。
转载请完整保留以下内容,未经授权删除以下内容进行转载盗用的,保留追究法律责任的权利!
本文地址:https://www.cnblogs.com/coredx/p/10026783.html
完整源代码:Github
里面有各种小东西,这只是其中之一,不嫌弃的话可以Star一下。