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

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

　　

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

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

 1     [Table("UserProfile")]  2 public class UserProfile  3  {  4  [Key]  5  [DatabaseGenerated(DatabaseGeneratedOption.Identity)]  6 [Column(Order = 0)]  7 public int UserId { get; set; }  8  9 [Column(Order = 1)] 10  [Required] 11 public string UserName { get; set; } 12 13 [Column(Order = 2)] 14 public string UserCode { get; set; } 15 16 [Column(Order = 3)] 17 public int Status { get; set; } 18 19 [Column(Order = 4)] 20 public string Email { get; set; } 21 22 [Column(Order = 5)] 23 public int? Gender { get; set; } 24 25 [Column(Order = 6)] 26 public int? CreatorId { get; set; } 27 28 public virtual ICollection<Role> Roles { get; set; } 29  } 30 31 /// <summary> 32 /// 角色信息为树状结构 33 /// </summary> 34 [Table("webpages_Roles")] 35 public class Role 36  { 37  [Key] 38  [DatabaseGenerated(DatabaseGeneratedOption.Identity)] 39 [Column(Order = 0)] 40 public int RoleId { get; set; } 41 42 [Column(Order = 1)] 43 public int? ParentId { get; set; } //父级角色ID 44 45  [Required] 46 [Column(Order = 2)] 47 public string RoleName { get; set; } 48 49 [Column(Order = 3)] 50 public int TreeLevel { get; set; } 51 52 [Column(Order = 4)] 53 public string TreePath { get; set; } 54 55 [Column(Order = 5)] 56 public int Status { get; set; } 57 58 [Column(Order = 6)] 59 public string Description { get; set; } 60 61 [ForeignKey("ParentId")] 62 public virtual Role Parent { get; set; } 63 64 [InverseProperty("Parent")] 65 public virtual ICollection<Role> Children { get; set; } 66 67 public virtual ICollection<UserProfile> Users { get; set; } 68 }

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

 1     public class UsersContext: DbContext, IDisposable  2  {  3 #region 数据表映射实例  4 public DbSet<UserProfile> UserProfiles { get; set; }  5  6 public DbSet<Role> Roles { get; set; }  7 #endregion  8  9 #region 构造 10 public UsersContext() : base("DefaultConnection") { } 11 12 public UsersContext(string connectionString) : base(connectionString) { } 13 #endregion 14 15 #region 事件 16 protected override void OnModelCreating(DbModelBuilder modelBuilder) 17  { 18 //配置用户角色关系表 19 modelBuilder.Entity<Role>().HasMany(p => p.Users).WithMany(p => p.Roles).Map(m => { m.ToTable("webpages_UsersInRoles"); m.MapLeftKey("UserId"); m.MapRightKey("RoleId"); }); 20 base.OnModelCreating(modelBuilder); 21  } 22 #endregion 23 24 #region 析构 25 public event EventHandler Disposed; 26 27 public bool IsDisposed { get; private set; } 28 29 public new void Dispose() 30  { 31 Dispose(true); 32  } 33 34 protected new void Dispose(bool disposing) 35  { 36 lock (this) 37  { 38 if (disposing && !IsDisposed) 39  { 40 base.Dispose(); 41 var evt = Disposed; 42 if (evt != null) evt(this, EventArgs.Empty); 43 Disposed = null; 44 IsDisposed = true; 45 GC.SuppressFinalize(this); 46  } 47  } 48  } 49 #endregion 50 }

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

　　

 1         private class SimpleMembershipInitializer  2  {  3 public SimpleMembershipInitializer()  4  {  5 Database.SetInitializer<Framework.DomainModels.UsersContext>(null); //这里使用我们刚刚定义的数据库上下文  6  7 try  8  {  9 using (var context = new Framework.DomainModels.UsersContext()) //这里使用我们刚刚定义的数据库上下文 10  { 11 if (!context.Database.Exists()) 12  { 13 // Create the SimpleMembership database without Entity Framework migration schema 14  ((IObjectContextAdapter)context).ObjectContext.CreateDatabase(); 15  } 16  } 17 18 WebSecurity.InitializeDatabaseConnection("DefaultConnection", "UserProfile", "UserId", "UserName", autoCreateTables: true); 19  } 20 catch (Exception ex) 21  { 22 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); 23  } 24  } 25 }

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

　　

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

 1  [HttpPost]  2  [AllowAnonymous]  3  [ValidateAntiForgeryToken]  4 public ActionResult Register(RegisterModel model)  5  {  6 if (ModelState.IsValid)  7  {  8 // Attempt to register the user  9 try 10  { 11 WebSecurity.CreateUserAndAccount(model.UserName, model.Password, new { Status = 1 /* 在UserProfile表中，该字段为必填。如还有其它必填字段，只需在此添加即可 */ }); 12  WebSecurity.Login(model.UserName, model.Password); 13 return RedirectToAction("Index", "Home"); 14  } 15 catch (MembershipCreateUserException e) 16  { 17 ModelState.AddModelError("", ErrorCodeToString(e.StatusCode)); 18  } 19  } 20 21 // If we got this far, something failed, redisplay form 22 return View(model); 23 }

　　运行项目，注册用户：

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