sql-server - 让 IdentityManager 在两个不同的数据库上进行更改

Question

我们有两个数据库，其中的User 表如下:

• MasterDB.AspNetUsers
• Customer1DB.AspNetUsers

我们使用 MasterDb 登录用户，然后将他们连接到客户数据库，例如 Customer1DB(或其他一些客户数据库，具体取决于用户) .两个用户表都具有基于 ASP.NET Identity Framework 的相同模式，我们可以使用 IdentityManager管理 MasterDb 上的用户。

现在我想要的是对 MasterDb 用户记录执行的任何更改，将其镜像到客户数据库中的用户表(具有相同的用户 ID)。我想知道最好的方法是什么？我是否需要修改对 UserStore、RoleStore、UserManager、RoleManager 的所有操作？我有一个获取 UserId 并将其从第一个数据库添加或更新到第二个数据库的函数，但我不确定我应该如何将它准确地集成到身份框架实现中。

谢谢!

Answer 1

我认为您有 2 个选择:

1. 为您需要的商店和功能创建适配器。适配器将在两个数据库上调用相同的方法。

2. 在主 DbContext 中，当您保存更改时，您会查看 ChangeTracker 并将更改复制到另一个 DbContext。

每种方法各有优缺点。

1

示例适配器:

class MultiDatabaseUserStore<T> : IUserStore<T>
    where T : IdentityUser
{
    private readonly UserStore<T>[] stores;

    public MultiDatabaseUserStore(params IdentityDbContext[] dbContexts)
    {
        if (dbContexts == null || dbContexts.Length <= 0)
        {
            throw new ArgumentException("At least one db context is required.", "dbContexts");
        }

        this.stores = dbContexts.Select(x => new UserStore<T>(x)).ToArray();
    }

    public void Dispose()
    {
        foreach (var store in this.stores)
        {
            store.Dispose();
        }
    }

    public Task CreateAsync(T user)
    {
        return this.ExecuteOnAll(x => x.CreateAsync(user));
    }

    public Task UpdateAsync(T user)
    {
        return this.ExecuteOnAll(x => x.UpdateAsync(user));
    }

    public Task DeleteAsync(T user)
    {
        return this.ExecuteOnAll(x => x.DeleteAsync(user));
    }

    public Task<T> FindByIdAsync(string userId)
    {
        return this.stores.First().FindByIdAsync(userId);
    }

    public Task<T> FindByNameAsync(string userName)
    {
        return this.stores.First().FindByNameAsync(userName);
    }

    private Task ExecuteOnAll(Func<UserStore<T>, Task> function)
    {
        return Task.WhenAll(this.stores.Select(function));
    }
}

它像这样集成:

var store = new MultiDatabaseUserStore<IdentityUser>(new MasterDb(), new Customer1DB());
var userManager = new UserManager<IdentityUser>(store);

这可行，但根据您的需要，您将有更多接口(interface)需要实现:

• IUserLoginStore
• IUserClaimStore
• IUserRoleStore
• IUserPasswordStore
• IUserSecurityStampStore
• IQueryableUserStore
• IUserEmailStore
• IUserPhoneNumberStore
• IUserTwoFactorStore
• IUserLockoutStore

2

示例主 DbContext:

class MasterDb : IdentityDbContext
{
    public MasterDb()
    {
    }

    public override int SaveChanges()
    {
        var changes = this.GetChangesToReplicate();

        var i = base.SaveChanges();

        this.SaveReplicatedChanges(changes);

        return i;
    }

    public override Task<int> SaveChangesAsync()
    {
        return Task.Run(async () =>
        {
            var changes = this.GetChangesToReplicate();

            var i = await base.SaveChangesAsync();

            this.SaveReplicatedChanges(changes);

            return i;
        });
    }

    private void SaveReplicatedChanges(IEnumerable<Action<IdentityDbContext>> changes)
    {
        if (changes != null)
        {
            using (var db = new Customer1DB())
            {
                foreach (var change in changes)
                {
                    change(db);
                }

                db.SaveChanges();
            }
        }
    }

    private IEnumerable<Action<IdentityDbContext>> GetChangesToReplicate()
    {
        var actions = new List<Action<IdentityDbContext>>();
        var userTye = typeof(IdentityUser);
        var changedEntries = this.ChangeTracker.Entries();
        var users = changedEntries.Where(x => x.Entity.GetType() == userTye).ToList();

        foreach (var u in users)
        {
            switch (u.State)
            {
                case EntityState.Added:
                    var userToAdd = (IdentityUser)u.Entity;
                    actions.Add(db => db.Users.Add(userToAdd));
                    break;

                case EntityState.Modified:
                    var userToUpdate = (IdentityUser)u.Entity; ;
                    actions.Add(db =>
                    {
                        db.Users.Attach(userToUpdate);
                        db.Entry(userToUpdate).State = EntityState.Modified;
                    });
                    break;

                case EntityState.Deleted:
                    var userToDelete = (IdentityUser)u.Entity; ;
                    actions.Add(db =>
                    {
                        db.Users.Attach(userToDelete);
                        db.Entry(userToDelete).State = EntityState.Deleted;
                    });
                    break;
            }
        }

        return actions;
    }
}

这也有效，但仅限于用户实体。您将必须根据需要添加角色、声明等的复制。

2 改进

考虑一下，您可以跳过类型检查并直接将所有更改复制到另一个 DbContext。这大大简化了代码，并且适用于角色/声明。您还需要添加 bool 属性 IsReplicatingChanges 以仅在必要时启用复制。

您必须仅为身份存储创建一个实例并将标志设置为 true:

var dbContext = new MasterDb { IsReplicatingChanges = true };
var userManager = new UserManager<IdentityUser>(new UserStore<IdentityUser>(dbContext));
var roleManager = new RoleManager<IdentityRole>(new RoleStore<IdentityRole>(dbContext));

GetChangesToReplicate() 变为:

private IEnumerable<Action<IdentityDbContext>> GetChangesToReplicate()
{
    if (!this.IsReplicatingChanges) // Flag
    {
        // Return null when not activated
        return null;
    }

    var actions = new List<Action<IdentityDbContext>>();
    var changedEntries = this.ChangeTracker.Entries().Where(x => x.State != EntityState.Unchanged);

    foreach (var u in changedEntries)
    {
        var entity = u.Entity;
        var state = u.State;

        actions.Add(db =>
        {
            db.Set(entity.GetType()).Attach(entity);
            db.Entry(entity).State = state;
        });
    }

    return actions;
}

并且不要忘记将 null 检查添加到 SaveReplicatedChanges()。