当前位置 主页 > 网站技术 > 代码类 >

    利用EF6简单实现多租户的应用(2)

    栏目:代码类 时间:2019-09-11 14:24

    第二步 修改注册用户的代码,注册新用户的时候需要选择所属的公司信息

    [HttpPost] [AllowAnonymous] [ValidateAntiForgeryToken] public async Task<ActionResult> Register(AccountRegistrationModel viewModel) {  var data = this._companyService.Queryable().Select(x => new ListItem() { Value = x.Id.ToString(), Text = x.Name });  this.ViewBag.companylist = data;  // Ensure we have a valid viewModel to work with  if (!this.ModelState.IsValid)  {  return this.View(viewModel);  }  // Try to create a user with the given identity  try  {  // Prepare the identity with the provided information  var user = new ApplicationUser  {   UserName = viewModel.Username,   FullName = viewModel.Lastname + "." + viewModel.Firstname,   CompanyCode = viewModel.CompanyCode,   CompanyName = viewModel.CompanyName,   TenantId=viewModel.TenantId,   Email = viewModel.Email,   AccountType = 0     };  var result = await this.UserManager.CreateAsync(user, viewModel.Password);  // If the user could not be created  if (!result.Succeeded)  {   // Add all errors to the page so they can be used to display what went wrong   this.AddErrors(result);   return this.View(viewModel);  }  // If the user was able to be created we can sign it in immediately  // Note: Consider using the email verification proces  await this.SignInAsync(user, true);  return this.RedirectToLocal();  }  catch (DbEntityValidationException ex)  {  // Add all errors to the page so they can be used to display what went wrong  this.AddErrors(ex);  return this.View(viewModel);  } }AccountController.cs

    第三步 读取登录用户的TenantId 在用户查询和新增修改时把TenantId插入到表中,这里需要引用

    Z.EntityFramework.Plus,这个是免费开源的一个类库,功能强大

    public StoreContext()  : base("Name=DefaultConnection") {  //获取登录用户信息,tenantid  var claimsidentity = (ClaimsIdentity)HttpContext.Current.User.Identity;  var tenantclaim = claimsidentity?.FindFirst("http://schemas.microsoft.com/identity/claims/tenantid");  var tenantid = Convert.ToInt32(tenantclaim?.Value);  //设置当对Work对象进行查询时默认添加过滤条件  QueryFilterManager.Filter<Work>(q => q.Where(x => x.TenantId == tenantid));  //设置当对Order对象进行查询时默认添加过滤条件  QueryFilterManager.Filter<Order>(q => q.Where(x => x.TenantId == tenantid)); } public override Task<int> SaveChangesAsync(CancellationToken cancellationToken) {  var currentDateTime = DateTime.Now;  var claimsidentity = (ClaimsIdentity)HttpContext.Current.User.Identity;  var tenantclaim = claimsidentity?.FindFirst("http://schemas.microsoft.com/identity/claims/tenantid");  var tenantid = Convert.ToInt32(tenantclaim?.Value);  foreach (var auditableEntity in this.ChangeTracker.Entries<Entity>())  {  if (auditableEntity.State == EntityState.Added || auditableEntity.State == EntityState.Modified)  {   //auditableEntity.Entity.LastModifiedDate = currentDateTime;   switch (auditableEntity.State)   {   case EntityState.Added:    auditableEntity.Property("LastModifiedDate").IsModified = false;    auditableEntity.Property("LastModifiedBy").IsModified = false;    auditableEntity.Entity.CreatedDate = currentDateTime;    auditableEntity.Entity.CreatedBy = claimsidentity.Name;    auditableEntity.Entity.TenantId = tenantid;    break;   case EntityState.Modified:    auditableEntity.Property("CreatedDate").IsModified = false;    auditableEntity.Property("CreatedBy").IsModified = false;    auditableEntity.Entity.LastModifiedDate = currentDateTime;    auditableEntity.Entity.LastModifiedBy = claimsidentity.Name;    auditableEntity.Entity.TenantId = tenantid;    //if (auditableEntity.Property(p => p.Created).IsModified || auditableEntity.Property(p => p.CreatedBy).IsModified)    //{    // throw new DbEntityValidationException(string.Format("Attempt to change created audit trails on a modified {0}", auditableEntity.Entity.GetType().FullName));    //}    break;   }  }  }  return base.SaveChangesAsync(cancellationToken); } public override int SaveChanges() {  var currentDateTime = DateTime.Now;  var claimsidentity =(ClaimsIdentity)HttpContext.Current.User.Identity;  var tenantclaim = claimsidentity?.FindFirst("http://schemas.microsoft.com/identity/claims/tenantid");  var tenantid = Convert.ToInt32(tenantclaim?.Value);  foreach (var auditableEntity in this.ChangeTracker.Entries<Entity>())  {  if (auditableEntity.State == EntityState.Added || auditableEntity.State == EntityState.Modified)  {   auditableEntity.Entity.LastModifiedDate = currentDateTime;   switch (auditableEntity.State)   {   case EntityState.Added:    auditableEntity.Property("LastModifiedDate").IsModified = false;    auditableEntity.Property("LastModifiedBy").IsModified = false;    auditableEntity.Entity.CreatedDate = currentDateTime;    auditableEntity.Entity.CreatedBy = claimsidentity.Name;    auditableEntity.Entity.TenantId = tenantid;    break;   case EntityState.Modified:    auditableEntity.Property("CreatedDate").IsModified = false;    auditableEntity.Property("CreatedBy").IsModified = false;    auditableEntity.Entity.LastModifiedDate = currentDateTime;    auditableEntity.Entity.LastModifiedBy = claimsidentity.Name;    auditableEntity.Entity.TenantId = tenantid;    break;   }  }  }  return base.SaveChanges(); }DbContext.cs