快捷導(dǎo)航

利用EF6簡(jiǎn)單實(shí)現(xiàn)多租戶的應(yīng)用

更新時(shí)間：2019年09月10日 09:38:22 作者：阿新

這篇文章主要給大家介紹了關(guān)于如何利用EF6簡(jiǎn)單實(shí)現(xiàn)多租戶應(yīng)用的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或者使用EF6具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

什么是多租戶

網(wǎng)上有好多解釋，有些上升到了架構(gòu)設(shè)計(jì)，讓你覺(jué)得似乎非常高深莫測(cè)，特別是目前流行的ABP架構(gòu)中就有提到多租戶（IMustHaveTenant），其實(shí)說(shuō)的簡(jiǎn)單一點(diǎn)就是再每一張數(shù)據(jù)庫(kù)的表中添加一個(gè)TenantId的字段，用于區(qū)分屬于不同的租戶（或是說(shuō)不同的用戶組）的數(shù)據(jù)。關(guān)鍵是現(xiàn)實(shí)的方式必須對(duì)開發(fā)人員來(lái)說(shuō)是透明的，不需要關(guān)注這個(gè)字段的信息，由后臺(tái)或是封裝在基類中實(shí)現(xiàn)數(shù)據(jù)的篩選和更新。

基本原理

從新用戶注冊(cè)時(shí)就必須指定用戶的TenantId,我的例子是用CompanyId,公司信息做為TenantId,哪些用戶屬于不同的公司，每個(gè)用戶將來(lái)只能修改和查詢屬于本公司的數(shù)據(jù)。

接下來(lái)就是用戶登錄的時(shí)候獲取用戶信息的時(shí)候把TenantId保存起來(lái)，asp.net mvc（不是 core）是通過(guò) Identity 2.0實(shí)現(xiàn)的認(rèn)證和授權(quán)，這里需要重寫部分代碼來(lái)實(shí)現(xiàn)。

最后用戶對(duì)數(shù)據(jù)查詢/修改/新增時(shí)把用戶信息中TenantId，這里就需要設(shè)定一個(gè)Filter(過(guò)濾器)和每次SaveChange的插入TenantId

如何實(shí)現(xiàn)

第一步，擴(kuò)展 Asp.net Identity user 屬性，必須新增一個(gè)TenantId字段,根據(jù)Asp.net Mvc 自帶的項(xiàng)目模板修改IdentityModels.cs 這個(gè)文件

// You can add profile data for the user by adding more properties to your ApplicationUser class, please visit http://go.microsoft.com/fwlink/?LinkID=317594 to learn more.
 public class ApplicationUser : IdentityUser
 {
 public async Task<ClaimsIdentity> GenerateUserIdentityAsync(UserManager<ApplicationUser> manager, string authenticationType)
 {
  // Note the authenticationType must match the one defined in CookieAuthenticationOptions.AuthenticationType
  var userIdentity = await manager.CreateIdentityAsync(this, authenticationType);
  // Add custom user claims here
  userIdentity.AddClaim(new Claim("http://schemas.microsoft.com/identity/claims/tenantid", this.TenantId.ToString()));
  userIdentity.AddClaim(new Claim("CompanyName", this.CompanyName));
  userIdentity.AddClaim(new Claim("EnabledChat", this.EnabledChat.ToString()));
  userIdentity.AddClaim(new Claim("FullName", this.FullName));
  userIdentity.AddClaim(new Claim("AvatarsX50", this.AvatarsX50));
  userIdentity.AddClaim(new Claim("AvatarsX120", this.AvatarsX120));
  return userIdentity;
 }
 public async Task<ClaimsIdentity> GenerateUserIdentityAsync(UserManager<ApplicationUser> manager)
 {
  // Note the authenticationType must match the one defined in CookieAuthenticationOptions.AuthenticationType
  var userIdentity = await manager.CreateIdentityAsync(this, DefaultAuthenticationTypes.ApplicationCookie);
  // Add custom user claims here
  return userIdentity;
 }

 [Display(Name = "全名")]
 public string FullName { get; set; }
 [Display(Name = "性別")]
 public int Gender { get; set; }
 public int AccountType { get; set; }
 [Display(Name = "所屬公司")]
 public string CompanyCode { get; set; }
 [Display(Name = "公司名稱")]
 public string CompanyName { get; set; }
 [Display(Name = "是否在線")]
 public bool IsOnline { get; set; }
 [Display(Name = "是否開啟聊天功能")]
 public bool EnabledChat { get; set; }
 [Display(Name = "小頭像")]
 public string AvatarsX50 { get; set; }
 [Display(Name = "大頭像")]
 public string AvatarsX120 { get; set; }
 [Display(Name = "租戶ID")]
 public int TenantId { get; set; }
 }



 public class ApplicationDbContext : IdentityDbContext<ApplicationUser>
 {
 public ApplicationDbContext()
  : base("DefaultConnection", throwIfV1Schema: false) => Database.SetInitializer<ApplicationDbContext>(null);

 public static ApplicationDbContext Create() => new ApplicationDbContext();


 }

第二步修改注冊(cè)用戶的代碼，注冊(cè)新用戶的時(shí)候需要選擇所屬的公司信息

[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 在用戶查詢和新增修改時(shí)把TenantId插入到表中,這里需要引用

Z.EntityFramework.Plus，這個(gè)是免費(fèi)開源的一個(gè)類庫(kù)，功能強(qiáng)大

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);
  //設(shè)置當(dāng)對(duì)Work對(duì)象進(jìn)行查詢時(shí)默認(rèn)添加過(guò)濾條件
  QueryFilterManager.Filter<Work>(q => q.Where(x => x.TenantId == tenantid));
  //設(shè)置當(dāng)對(duì)Order對(duì)象進(jìn)行查詢時(shí)默認(rèn)添加過(guò)濾條件
  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

經(jīng)過(guò)以上3步就實(shí)現(xiàn)一個(gè)簡(jiǎn)單的多租戶查詢數(shù)據(jù)的功能。

總結(jié)

以上就是這篇文章的全部?jī)?nèi)容了，希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，謝謝大家對(duì)腳本之家的支持。

您可能感興趣的文章: