您的位置:首页 > 其它

[EFCore]EntityFrameworkCore Code First 当中批量自定义列名

2017-12-10 11:06 393 查看
在使用.NET CORE 进行 Web 开发的时候会考虑到使用不同数据库的情况,并且在每种数据库建立表结构的时候会采用不同的命名规则。之前的解决方法是使用
[ColumnAttribute]
或者
[TableAttribute]
这种特性来显式标注不同的列名。

[Table("bas_stock_address")]
public class BasStockAddress : FullAuditedEntity<int, User>
{
/// <summary>
/// 仓库ID
/// </summary>
[Column("stock_id")]
public int StockId { get; set; }
/// <summary>
/// 备注
/// </summary>
[Column("remark")]
[StringLength(200)]
public string Remark { get; set; }
}

这种情况的话就很尴尬,如果实体一多,就要对每个属性进行标注的话,工作量确实会很大。

这个时候就可以使用 Conventions 来处理这种情况,如果是 EntityFramework 6.x 的话可以使用:

https://msdn.microsoft.com/en-us/library/jj819164(v=vs.113).aspx

提到的方法来进行操作。

如果是 EntityFrameworkCore 的话可以通过
ModelBuilder.Model.GetEntityTypes()
获得所有实体对象的类型以及他的相关属性,并且使用
Relational()
方法来获得实体对象的 Annotation ,这样就可以对生成的表结构进行统一的配置。

public class ApplicationDbContext : IdentityDbContext<ApplicationUser>
{
public ApplicationDbContext(DbContextOptions<ApplicationDbContext> options)
: base(options)
{ }

protected override void OnModelCreating(ModelBuilder builder)
{
base.OnModelCreating(builder);

foreach(var entity in builder.Model.GetEntityTypes())
{
// 覆写表名
entity.Relational().TableName = entity.Relational().TableName.ToSnakeCase();

// 覆写列名
foreach(var property in entity.GetProperties())
{
property.Relational().ColumnName = property.Name.ToSnakeCase();
}

foreach(var key in entity.GetKeys())
{
key.Relational().Name = key.Relational().Name.ToSnakeCase();
}

foreach(var key in entity.GetForeignKeys())
{
key.Relational().Name = key.Relational().Name.ToSnakeCase();
}

foreach(var index in entity.GetIndexes())
{
index.Relational().Name = index.Relational().Name.ToSnakeCase();
}
}
}
}

public static class StringExtensions
{
public static string ToSnakeCase(this string input)
{
if (string.IsNullOrEmpty(input)) { return input; }
return Regex.Replace(input, @"([a-z0-9])([A-Z])", "$1_$2").ToLower();
}
}

这里主要的就是
ToSnakeCase
这个扩展方法,他将 NameSnake 这种形式的名称替换为 name_snake。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: