.NET MVC4 实训记录之二（扩展WebSecurity模型下的UserProfile表）

　　使用VS2013创建MVC4项目后，自动生成的代码中默认使用WebSecurity模型创建用户管理，生成以下数据库：

　　

　　用户信息只有ID和UserName，角色信息也只有两个基础字段。通常情况下这样的数据表不能满足我们的需求，因此对其进行扩展。

　　首先定义自己的用户信息、角色信息结构。

    [Table("UserProfile")]
    public class UserProfile
    {
        [Key]
        [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
        [Column(Order = 0)]
        public int UserId { get; set; }

        [Column(Order = 1)]
        [Required]
        public string UserName { get; set; }

        [Column(Order = 2)]
        public string UserCode { get; set; }

        [Column(Order = 3)]
        public int Status { get; set; }

        [Column(Order = 4)]
        public string Email { get; set; }

        [Column(Order = 5)]
        public int? Gender { get; set; }

        [Column(Order = 6)]
        public int? CreatorId { get; set; }

        public virtual ICollection<Role> Roles { get; set; }
    }

    /// <summary>
    /// 角色信息为树状结构
    /// </summary>
    [Table("webpages_Roles")]
    public class Role
    {
        [Key]
        [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
        [Column(Order = 0)]
        public int RoleId { get; set; }

        [Column(Order = 1)]
        public int? ParentId { get; set; }      //父级角色ID

        [Required]
        [Column(Order = 2)]
        public string RoleName { get; set; }

        [Column(Order = 3)]
        public int TreeLevel { get; set; }

        [Column(Order = 4)]
        public string TreePath { get; set; }

        [Column(Order = 5)]
        public int Status { get; set; }

        [Column(Order = 6)]
        public string Description { get; set; }

        [ForeignKey("ParentId")]
        public virtual Role Parent { get; set; }

        [InverseProperty("Parent")]
        public virtual ICollection<Role> Children { get; set; }

        public virtual ICollection<UserProfile> Users { get; set; }
    }

　　创建数据库上下文类型UsersContext：

    public class UsersContext: DbContext, IDisposable
    {
        #region 数据表映射实例
        public DbSet<UserProfile> UserProfiles { get; set; }

        public DbSet<Role> Roles { get; set; }
        #endregion

        #region 构造
        public UsersContext() : base("DefaultConnection") { }

        public UsersContext(string connectionString) : base(connectionString) { }
        #endregion

        #region 事件
        protected override void OnModelCreating(DbModelBuilder modelBuilder)
        {
            //配置用户角色关系表
            modelBuilder.Entity<Role>().HasMany(p => p.Users).WithMany(p => p.Roles).Map(m => { m.ToTable("webpages_UsersInRoles"); m.MapLeftKey("UserId"); m.MapRightKey("RoleId"); });
            base.OnModelCreating(modelBuilder);
        }
        #endregion

        #region 析构
        public event EventHandler Disposed;

        public bool IsDisposed { get; private set; }

        public new void Dispose()
        {
            Dispose(true);
        }

        protected new void Dispose(bool disposing)
        {
            lock (this)
            {
                if (disposing && !IsDisposed)
                {
                    base.Dispose();
                    var evt = Disposed;
                    if (evt != null) evt(this, EventArgs.Empty);
                    Disposed = null;
                    IsDisposed = true;
                    GC.SuppressFinalize(this);
                }
            }
        }
        #endregion
    }

　　修改站点项目Filters目录下的InitializeSimpleMembershipAttribute.cs文件中的构造器。

　　

        private class SimpleMembershipInitializer
        {
            public SimpleMembershipInitializer()
            {
                Database.SetInitializer<Framework.DomainModels.UsersContext>(null); //这里使用我们刚刚定义的数据库上下文

                try
                {
                    using (var context = new Framework.DomainModels.UsersContext()) //这里使用我们刚刚定义的数据库上下文
                    {
                        if (!context.Database.Exists())
                        {
                            // Create the SimpleMembership database without Entity Framework migration schema
                            ((IObjectContextAdapter)context).ObjectContext.CreateDatabase();
                        }
                    }

                    WebSecurity.InitializeDatabaseConnection("DefaultConnection", "UserProfile", "UserId", "UserName", autoCreateTables: true);
                }
                catch (Exception ex)
                {
                    throw new InvalidOperationException("The ASP.NET Simple Membership database could not be initialized. For more information, please see http://go.microsoft.com/fwlink/?LinkId=256588", ex);
                }
            }
        }

　　删除原有数据库，重新运行项目，点击注册按钮，这时生成新的数据库。

　　

　　在提交注册用户之前，让我们再做一些修改。找到AccountController下的Register方法（Post类型)，对其进行以下修改：

        [HttpPost]
        [AllowAnonymous]
        [ValidateAntiForgeryToken]
        public ActionResult Register(RegisterModel model)
        {
            if (ModelState.IsValid)
            {
                // Attempt to register the user
                try
                {
                    WebSecurity.CreateUserAndAccount(model.UserName, model.Password, new { Status = 1 /* 在UserProfile表中，该字段为必填。如还有其它必填字段，只需在此添加即可 */ });
                    WebSecurity.Login(model.UserName, model.Password);
                    return RedirectToAction("Index", "Home");
                }
                catch (MembershipCreateUserException e)
                {
                    ModelState.AddModelError("", ErrorCodeToString(e.StatusCode));
                }
            }

            // If we got this far, something failed, redisplay form
            return View(model);
        }

　　运行项目，注册用户：

　　OK，我们的用户信息及角色信息扩展成功！