[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();
    }
}
           

這裡主要的就是

ToSnakeCase

這個擴充方法，他将 NameSnake 這種形式的名稱替換為 name_snake。