【WTM-多租户改造】「建议收藏」

public class Tenant : BasePoco
    {
        [Display(Name = "编号")]
        [Required(ErrorMessage = "{0}是必填项")]
        public string Code { get; set; }

        [Display(Name = "域名")]
        [Required(ErrorMessage = "{0}是必填项")]
        public string DomainName { get; set; }

        [Display(Name = "租户角色")]
        [Required(ErrorMessage = "{0}是必填项")]
        public Guid RoleId { get; set; }
        [Display(Name = "租户角色")]
        public FrameworkRole Role { get; set; }

        [Display(Name = "账号")]
        [Required(ErrorMessage = "{0}是必填项")]
        public string Account { get; set; }

        [Display(Name = "名称")]
        [Required(ErrorMessage = "{0}是必填项")]
        public string Name { get; set; }
    }

        [HttpPost]
[ActionDescription("Sys.Create")]
public ActionResult Create(TenantVM vm)
{
using (var trans = DC.BeginTransaction())
{
if (!ModelState.IsValid)
{
return PartialView(vm);
}
else
{
vm.DoAdd();
if (!ModelState.IsValid)
{
vm.DoReInit();
return PartialView(vm);
}
else
{
//我这代码直接写这了 生成租户库和基本信息 随便写了下简单的系统表数据
var NDC = new DataContext(Wtm.ConfigInfo.Connections[0].Value.Replace("SAASDEMODB", "SAASDEMODB" + vm.Entity.Code), DBTypeEnum.SqlServer);
var Result = NDC.Database.EnsureCreated();
if (Result)
{
var role = DC.Set<FrameworkRole>().Where(x => x.ID == vm.Entity.RoleId).FirstOrDefault();
//角色拥有的菜单权限
var pr = DC.Set<FunctionPrivilege>().Where(x => x.RoleCode == role.RoleCode).ToList();
var user = new FrameworkUser
{
ITCode = vm.Entity.Account,
Password = Utils.GetMD5String("000000"),
IsValid = true,
Name = vm.Entity.Name
};
var userrole = new FrameworkUserRole
{
UserCode = vm.Entity.Account,
RoleCode = role.RoleCode
};
NDC.Set<FrameworkUser>().Add(user);
NDC.Set<FrameworkRole>().Add(role);
NDC.Set<FrameworkUserRole>().Add(userrole);
//这里框架自带角色表  页面权限FunctionPrivilege表没有加父级菜单数据 会导致约束冲突。后期自己添加租户角色表吧
NDC.Set<FrameworkMenu>().AddRange(DC.Set<FrameworkMenu>().CheckContain(pr.Select(x => x.MenuItemId).ToList(), x => x.ID).ToList());
NDC.Set<FunctionPrivilege>().AddRange(pr);
NDC.SaveChanges();
//云解析DNS-添加解析记录
if (!new CommonHelp().DomainNameResolution(vm.Entity.Code))
{
trans.Rollback();
return FFResult().CloseDialog().RefreshGrid().Alert("域名解析失败！");
}
}
else
{
trans.Rollback();
return FFResult().CloseDialog().RefreshGrid().Alert("租户信息初始化失败！");
}
trans.Commit();
return FFResult().CloseDialog().RefreshGrid();
}
}
}
}

        #region 云解析DNS-添加解析记录  可以去阿里云地址看文档 https://help.aliyun.com/document_detail/29772.html  但是这种方式服务器要求 80端口只允许部署这一套系统，因为现在这种方式是域名直接指向服务器IP
public bool DomainNameResolution(string Name)
{
IClientProfile profile = DefaultProfile.GetProfile("cn-hangzhou", "", "");//域名 AccessKeyID Secret
DefaultAcsClient client = new DefaultAcsClient(profile);
var request = new AddDomainRecordRequest();
request._Value = "";  //指向服务器IP
request.Type = "A";
request.RR = Name;  //随便定义   
request.DomainName = "xxx.com";  //域名
try
{
var response = client.GetAcsResponse(request);
return true;
//Console.WriteLine(System.Text.Encoding.Default.GetString(response.HttpResponse.Content));
}
catch (ServerException e)
{
return false;
}
catch (ClientException e)
{
return false;
}
}
#endregion

        #region 获取当前url
public string GetAbsoluteUri(HttpRequest request)
{
return new StringBuilder()
.Append(request.Scheme)
.Append("://")
.Append(request.Host)
.Append(request.PathBase)
.Append(request.Path)
.Append(request.QueryString)
.ToString();
}
#endregion
[Public]
[ActionDescription("Login")]
public IActionResult Login()
{
LoginVM vm = Wtm.CreateVM<LoginVM>();
string TenantKey = "default";
string displayUrl = GetAbsoluteUri(HttpContext.Request);
displayUrl = displayUrl.Replace("http://", "").Replace("https://", "") + "/";
displayUrl = displayUrl.Substring(0, displayUrl.IndexOf("/"));
var ZDC = new DataContext(Wtm.ConfigInfo.Connections[0].Value, DBTypeEnum.SqlServer);
var Tenant = ZDC.Set<Tenant>().Where(x => x.DomainName.Replace("http://", "").Replace("https://", "") == displayUrl).FirstOrDefault();
if (Tenant != null)
{
TenantKey = "SAASDEMODB" + Tenant.Code;
Wtm.Session.Set("TenantKey", "SAASDEMODB" + Tenant.Code);
}
else
{
Wtm.Session.Set("TenantKey", TenantKey);
}
int i = 0;
foreach (var item in Wtm.ConfigInfo.Connections)
{
if (item.Key == TenantKey)
{
i++;
break;
}
}
if (i == 0)
{
CS cs = new CS();
cs.DbContext = "DataContext";
cs.DbType = DBTypeEnum.SqlServer;
cs.Key = TenantKey;
cs.Value = Wtm.ConfigInfo.Connections[0].Value.Replace("SAASDEMODB", cs.Key);
cs.DcConstructor = Wtm.ConfigInfo.Connections[0].DcConstructor;
Wtm.ConfigInfo.Connections.Add(cs);
}
vm.Redirect = HttpContext.Request.Query["ReturnUrl"];
if (Wtm.ConfigInfo.IsQuickDebug == true)
{
vm.ITCode = "admin";
vm.Password = "000000";
}
return View(vm);
}

        public string CSSelector(ActionExecutingContext context)
{
var wtm = (context.Controller as IBaseController)?.Wtm;
var TenantKey = wtm.Session.Get<string>("TenantKey");
return TenantKey;
}