ABP框架入門踩坑-配置數據庫表前綴

配置數據庫表前綴

ABP踩坑記錄-目錄

本篇其實和ABP關係並不大，主要是EF Core的一些應用-.-。

原由

支持數據庫表前綴應該是不少應用中比較常見的功能，而在ABP中並沒直接提供這一功能，因此在咱們的應用中，咱們轉而藉助EF Core的配置來實現數據庫表前綴的配置。

解決方案

這裏我結合了Fluent API和數據註解的形式進行配置。

首先，約定全部自定義的表，在其實體類型上都標註了[Table("tablename")]屬性。

而後在QincaiDbContext中重載OnModelCreating方法。

protected override void OnModelCreating(ModelBuilder modelBuilder)
{
    var entityTypes = modelBuilder.Model.GetEntityTypes().ToList();
    // 設置自定義表前綴
    foreach (var entityType in entityTypes)
    {
        if (entityType.ClrType
            .GetCustomAttributes(typeof(TableAttribute), true)
            .FirstOrDefault() is TableAttribute table)
        {
            // 若是你的表名就是實體類型名的話，能夠修改成以下形式，就沒必要給出[table]的Name參數
            // string tableName = tablePrefix + entityType.ClrType.Name;
            // 如如有其餘需求也可在此進行修改
            string tableName = tablePrefix + table.Name;
            modelBuilder.Entity(entityType.ClrType)
                .ToTable(tableName);
        }
    }
    // 設置內置表前綴
    modelBuilder.ChangeAbpTablePrefix<Tenant, Role, User>(tablePrefix);
}

經歷

由於採用的Module Zero模板，其數據庫上下文中已包含部份內置表，ABP Module Zero也提供了拓展方法ModelBuilder.ChangeAbpTablePrefix<TTenant, TRole, TUser>以便於修改表前綴，其默認的表前綴爲Abp。

剩下就是配置咱們本身定義的表的前綴，這裏咱們考慮將表前綴以字符串常量的形式儲存。

參考資料：EF Core文檔

若是你喜歡用Fluent API來配置數據庫，那麼解決方案就很簡單了，直接拼接字符串生成新表名便可。

能夠參考如下代碼：

// 這裏tablePrefix爲字符串常量
modelBuilder.Entity<Blog>().ToTable(tablePrefix + "Blogs");

可是，若是每個類型都須要配置的話，難免顯得有些冗長。所以，我便考慮經過反射的方式來統一配置表前綴。

首先，咱們須要獲取到全部的實體類型。

// 這裏須要注意須要將迭代對象做臨時存儲
// 直接foreach的話會報錯，即不能修改迭代中的對象
var entityTypes = modelBuilder.Model.GetEntityTypes().ToList();

foreach (var entityType in entityTypes)
{
    // 配置表前綴
    // ...
}

而後即是表前綴的配置，一開始個人想法是直接經過實體類型名來配置表名，但在考慮實際狀況後，我認爲仍是須要有必定的靈活度使得該解決方案更通用。

最終，這裏我選擇用數據註解的方式來爲指定表名，即利用[Table]特性。

由於，這是EF Core官方所提供配置方式之一，不會讓你們由於配置表前綴而增刪過多代碼。

咱們會用到[Table]特性的Name參數，例如這樣：

using System.ComponentModel.DataAnnotations.Schema;

[Table("blogs")]
public class Blog
{
    public int BlogId { get; set; }
    public string Url { get; set; }
}

由於特性並不支持在其參數中引用字符串常量，因此[Table($"{tablePrefix}blogs")]這種寫法是不存在的，你們清醒一下。

而後，就是使用Fluent API來配置表前綴了。

// 經過反射獲取該實體類上定義的TableAttribute
// 這裏沒有處理table爲null的狀況
TableAttribute table = (TableAttribute)entityType.ClrType
    .GetCustomAttributes(typeof(TableAttribute), true)
    .FirstOrDefault();

// 利用TableAttribute中的Name參數來配置表名
modelBuilder.Entity(entityType.ClrType)
    .ToTable(tablePrefix + table.Name);

結合模式匹配語法，咱們能夠寫出以下形式：

// 若table爲null，模式匹配結果將爲false
if (entityType.ClrType
    .GetCustomAttributes(typeof(TableAttribute), true)
    .FirstOrDefault() is TableAttribute table)
{
    modelBuilder.Entity(entityType.ClrType)
        .ToTable(tablePrefix + table.Name);
}

最後，尚未完！！

還記得在一開始，我就說了ABP Module Zero內置表有默認的前綴Abp，也就是說在上述反射得到的內置表table.Name將是Abpxxx的形式，若是你不但願表名中出現Abp，能夠在反射以後再使用Abp提供的拓展方法將表名修改一下。