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

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

在使用.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();

    }

}