还在为身份验证引入的Microsoft.AspNet.Identity.EntityFramework导致多上下文，生成的DB改名困扰吗？

　　最近一个项目需要在API里面集成身份验证，所以选择了Microsoft.AspNet.Identity.Core，以及Microsoft.AspNet.Identity.Owin来实现，但是在实现过程中发现了需要引入Microsoft.AspNet.Identity.EntityFramework ，导致在代码内出现了多个EF的上下文，虽然网上到处有改名的方式，但是对于强迫症患者，一个工程里多个上下文始终不能接受。果断查看有无方案解决

　　通过获取源码发现，Microsoft.AspNet.Identity.EntityFramework只是封装了EF，完全可以自己来做，下面简单说明下实现思路。

　　本文是基于web api5.2+unity+ef来实现

　　1.通过源码可以查看的继承体系

　　

　　可以看到其核心是Core中的UserManager来操作，并且Microsoft.AspNet.Identity.EntityFramework封装了EF的上下文和操作的模型。

　　2.按照需要自定义模型

　　Microsoft.AspNet.Identity.EntityFramework   IdentityRole  -> RoleInfo ：IRole   只需要继承IRole即可，默认的左联按需取舍

　　IdentityUser  -> UserInfo : IUser

　　自定义 UserRoleInfo，UserClaimInfo，AccountLoginInfo

　　3.把新的模型添加到自己的EF上下文

　　通过codefirst在Map里自定义表名字段类型等，想怎么做怎么做。

　　4.添加身份操作的Store替换Microsoft.AspNet.Identity.EntityFramework里的UserStore和RoleStore

　　定义一个新的接口，方便注入

　　

/// <summary>
    /// 身份验证Store接口
    /// </summary>
    public interface ICustomerIdentityStore :
        IUserLoginStore<UserInfo, string>,
        IUserClaimStore<UserInfo, string>,
        IUserRoleStore<UserInfo, string>,
        IUserPasswordStore<UserInfo, string>,
        IUserSecurityStampStore<UserInfo, string>,
        IQueryableUserStore<UserInfo, string>,
        IUserEmailStore<UserInfo, string>,
        IUserPhoneNumberStore<UserInfo, string>,
        IUserTwoFactorStore<UserInfo, string>,
        IUserLockoutStore<UserInfo, string>,
        IUserStore<UserInfo>
    {
    }

　　然后实现其接口，思路的通过注入把上下文的仓储注入到实现里即可对DB进行操作

　　5. 自定义身份验证仓储接口和实现（样例代码）

　　

/// <summary>
    /// 身份验证仓储
    /// </summary>
    public interface IAuthRepository
    {
        /// <summary>
        /// 注册用户
        /// </summary>
        /// <param name="userModel"></param>
        /// <returns></returns>
        Task<IdentityResult> RegisterUser(UserModel userModel);

        /// <summary>
        /// 查找用户
        /// </summary>
        /// <param name="userName"></param>
        /// <param name="password"></param>
        /// <returns></returns>
        Task<UserInfo> FindUser(string userName, string password);

        /// <summary>
        /// 查找AppClient信息
        /// </summary>
        /// <param name="clientId"></param>
        /// <returns></returns>
        AppClientInfo FindClient(string clientId);

        /// <summary>
        /// 添加Token信息
        /// </summary>
        /// <param name="token"></param>
        /// <returns></returns>
        Task<bool> AddRefreshToken(RefreshTokenInfo token);

        /// <summary>
        /// 移除token信息
        /// </summary>
        /// <param name="refreshTokenId"></param>
        /// <returns></returns>
        Task<bool> RemoveRefreshToken(string refreshTokenId);

        /// <summary>
        /// 移除token信息
        /// </summary>
        /// <param name="refreshToken"></param>
        /// <returns></returns>
        Task<bool> RemoveRefreshToken(RefreshTokenInfo refreshToken);

        /// <summary>
        /// 查找token信息
        /// </summary>
        /// <param name="refreshTokenId"></param>
        /// <returns></returns>
        Task<RefreshTokenInfo> FindRefreshToken(string refreshTokenId);

        /// <summary>
        /// 查找所有刷新token信息
        /// </summary>
        /// <returns></returns>
        List<RefreshTokenInfo> GetAllRefreshTokens();

        /// <summary>
        /// 通过登录信息查找用户信息
        /// </summary>
        /// <param name="loginInfo"></param>
        /// <returns></returns>
        Task<UserInfo> FindAsync(UserLoginInfo loginInfo);

        /// <summary>
        /// 创建用户信息
        /// </summary>
        /// <param name="user"></param>
        /// <returns></returns>
        Task<IdentityResult> CreateAsync(UserInfo user);

        /// <summary>
        /// 添加用户登录信息
        /// </summary>
        /// <param name="userId"></param>
        /// <param name="login"></param>
        /// <returns></returns>
        Task<IdentityResult> AddLoginAsync(string userId, UserLoginInfo login);
    }

　　然后添加其实现

　　

/// <summary>
    /// 身份验证仓储
    /// </summary>
    public class AuthRepository : IAuthRepository
    {
        /// <summary>
        /// 仓储接口
        /// </summary>
        private readonly IUnitRepository _repository;

        /// <summary>
        /// 工作单元
        /// </summary>
        private readonly IUnitOfWork _unitOfWork;

        /// <summary>
        /// 用户管理
        /// </summary>
        private readonly UserManager<UserInfo> _userManager;

        /// <summary>
        /// ctor
        /// </summary>
        /// <param name="repository"></param>
        /// <param name="unitOfWork"></param>
        /// <param name="userStore"></param>
        public AuthRepository(IUnitRepository repository, IUnitOfWork unitOfWork, ITaurusIdentityStore userStore)
        {
            _repository = repository;
            _unitOfWork = unitOfWork;
            _userManager = new UserManager<UserInfo>(userStore);
        }


        public async Task<IdentityResult> AddLoginAsync(string userId, UserLoginInfo login)
        {
            var result = await _userManager.AddLoginAsync(userId, login);

            return result;
        }

        public async Task<bool> AddRefreshToken(RefreshTokenInfo token)
        {
            var existingToken = _repository.FirstOrDefault<RefreshTokenInfo>(r => r.Subject == token.Subject
            && r.AppClientId == token.AppClientId);

            if (existingToken != null)
            {
                var result = await RemoveRefreshToken(existingToken);
            }

            _repository.Insert(token);

            return _unitOfWork.SaveChanges() > 0;
        }

        public async Task<IdentityResult> CreateAsync(UserInfo user)
        {
            var result = await _userManager.CreateAsync(user);

            return result;
        }

        public async Task<UserInfo> FindAsync(UserLoginInfo loginInfo)
        {
            var user = await _userManager.FindAsync(loginInfo);

            return user;
        }

        public AppClientInfo FindClient(string clientId)
        {
            var client = _repository.FirstOrDefault<AppClientInfo>(s => s.Id == clientId);

            return client;
        }

        public Task<RefreshTokenInfo> FindRefreshToken(string refreshTokenId)
        {
            var refreshToken = _repository.FirstOrDefault<RefreshTokenInfo>(s => s.TokenId == refreshTokenId);

            return Task.FromResult(refreshToken);
        }

        public async Task<UserInfo> FindUser(string userName, string password)
        {
            var user = await _userManager.FindAsync(userName, password);

            return user;
        }

        public List<RefreshTokenInfo> GetAllRefreshTokens()
        {
            return _repository.All<RefreshTokenInfo>().ToList();
        }

        public async Task<IdentityResult> RegisterUser(UserModel userModel)
        {
            var user = new UserInfo
            {
                UserName = userModel.UserName,
                Id = Guid.NewGuid().ToString(),
                FullName = "test",
                CreateTime = DateTime.Now,
                CreateBy = Guid.Empty.ToString(),
                UpdateBy = Guid.Empty.ToString(),
                UpdateTime = DateTime.Now
            };

            var result = await _userManager.CreateAsync(user, userModel.Password);

            return result;
        }

        public Task<bool> RemoveRefreshToken(RefreshTokenInfo refreshToken)
        {
            _repository.DeleteItem(refreshToken);

            var result = _unitOfWork.SaveChanges() > 0;
            return Task.FromResult(result);
        }

        public Task<bool> RemoveRefreshToken(string refreshTokenId)
        {
            var refreshToken = _repository.FirstOrDefault<RefreshTokenInfo>(s => s.Id == refreshTokenId);
            var result = false;
            if (refreshToken != null)
            {
                _repository.DeleteItem(refreshToken);
                result = _unitOfWork.SaveChanges() > 0;
            }

            return Task.FromResult(result);
        }
    }

　　思路是通过注入ICustomerIdentityStore(IUserStore)构造出UserManager实例

　　6.使用时直接使用IAuthRepository接口就好。

　　这样所有的操作都控制自己的代码里，不再需要去改表名，重新映射字段，维护多个上下文了。

有问题欢迎讨论

谢谢