using Castle.Components.DictionaryAdapter.Xml;
using Microsoft.AspNetCore.Mvc.Rendering;
using Newtonsoft.Json;
using SkiaSharp;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.Dynamic;
using System.Linq;
using System.Net.Sockets;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using WIDESEAWCS_Core.Const;
using WIDESEAWCS_Core.DB;
using WIDESEAWCS_Core.Helper;
using WIDESEAWCS_Core.Tenants;
namespace WIDESEAWCS_Core.Seed
{
public class DBSeed
{
private static string SeedDataFolder = "WIDESEAWCS_DB.DBSeed.Json/{0}.tsv";
///
/// 异步添加种子数据
///
///
///
///
public static async Task SeedAsync(DBContext dbContext, string WebRootPath)
{
try
{
if (string.IsNullOrEmpty(WebRootPath))
{
throw new Exception("获取wwwroot路径时,异常!");
}
SeedDataFolder = Path.Combine(WebRootPath, SeedDataFolder);
Console.WriteLine("************ WIDESEA DataBase Set *****************");
//Console.WriteLine($"Master DB ConId: {DBContext.ConnId}");
Console.WriteLine($"Master DB Type: {DBContext.DbType}");
//Console.WriteLine($"Master DB ConnectString: {DBContext.ConnectionString}");
Console.WriteLine();
// 创建数据库
Console.WriteLine($"Create Database(The Db Id:{DBContext.ConnId})...");
if (DBContext.DbType != SqlSugar.DbType.Oracle)
{
dbContext.Db.DbMaintenance.CreateDatabase();
ConsoleHelper.WriteSuccessLine($"Database Created Successfully!");
}
else
{
//Oracle 数据库不支持该操作
ConsoleHelper.WriteSuccessLine($"Oracle 数据库不支持该操作,可手动创建Oracle数据库!");
}
// 创建数据库表,遍历指定命名空间下的class,
// 注意不要把其他命名空间下的也添加进来。
Console.WriteLine("Create Tables...");
var path = AppDomain.CurrentDomain.RelativeSearchPath ?? AppDomain.CurrentDomain.BaseDirectory;
var referencedAssemblies = System.IO.Directory.GetFiles(path, MainDb.AssemblyName).Select(Assembly.LoadFrom).ToArray();
var modelTypes = referencedAssemblies
.SelectMany(a => a.DefinedTypes)
.Select(type => type.AsType())
.Where(x => x.IsClass && x.Namespace is MainDb.EntityNameSpace && x.GetCustomAttribute() != null)
.ToList();
modelTypes.ForEach(t =>
{
//var diffString = dbContext.Db.CodeFirst.GetDifferenceTables(t).ToDiffString();
// 这里只支持添加表,不支持删除
// 如果想要删除,数据库直接右键删除,或者联系SqlSugar作者;
IDbMaintenance dbMaintenance = dbContext.Db.DbMaintenance;
if (!dbMaintenance.IsAnyTable(t.Name, false))
{
ConsoleHelper.WriteSuccessLine($"Table [{t.Name}] Created Successfully");
dbContext.Db.CodeFirst.InitTables(t);
string seedData = FileHelper.ReadFile(string.Format(SeedDataFolder, t.Name), Encoding.UTF8);
#region AddSeedData
if (seedData != "不存在相应的目录")
{
List> dicFile = JsonConvert.DeserializeObject>>(seedData);
if (dicFile.Count > 0)
{
List> dic = new List>();
List columnNames = dbContext.Db.DbMaintenance.GetColumnInfosByTableName(t.Name, false).Select(x => x.DbColumnName).ToList();
var a = t.GetProperties().FirstOrDefault(x => !columnNames.Contains(x.Name));
List propertyInfos = t.GetProperties().Where(x => columnNames.Contains(x.Name)).ToList();
for (int j = 0; j < dicFile.Count; j++)
{
Dictionary keyValuePairs = new Dictionary();
for (int i = 0; i < propertyInfos.Count; i++)
{
PropertyInfo propertyInfo = propertyInfos[i];
SugarColumn sugarColumn = propertyInfo.GetCustomAttribute();
if (sugarColumn != null)
{
if (!sugarColumn.IsIgnore)
{
keyValuePairs.Add(propertyInfo.Name, dicFile[j][propertyInfo.Name]);
}
}
}
dic.Add(keyValuePairs);
}
if (dic.Count > 0)
{
for (int i = 0; i < dic.Count; i++)
{
if (dic[i].ContainsKey("CreateDate"))
dic[i]["CreateDate"] = DateTime.Now;
else
dic[i].Add("CreateDate", DateTime.Now);
}
string str = $"SET IDENTITY_INSERT {t.Name} ON;";
str += dbContext.Db.Insertable(dic).AS(t.Name).ToSqlString();
str += ($"SET IDENTITY_INSERT {t.Name} OFF;");
dbContext.Db.Ado.ExecuteCommand(str);
ConsoleHelper.WriteSuccessLine($"Table [{t.Name}] SeedData Added Successfully");
}
}
}
#endregion
}
else
{
List columnNames = dbContext.Db.DbMaintenance.GetColumnInfosByTableName(t.Name, false).Select(x => x.DbColumnName).ToList();
if (t.GetProperties().FirstOrDefault(x => !columnNames.Contains(x.Name)) != null)
{
bool isChange = true;
List propertyInfos = t.GetProperties().Where(x => !columnNames.Contains(x.Name)).ToList();
for (int i = 0; i < propertyInfos.Count; i++)
{
PropertyInfo propertyInfo = propertyInfos[i];
SugarColumn sugarColumn = propertyInfo.GetCustomAttribute();
if (sugarColumn != null)
{
if (!sugarColumn.IsIgnore)
{
if (!sugarColumn.IsNullable)
{
isChange = false;
break;
}
}
}
}
if (isChange)
dbContext.Db.CodeFirst.InitTables(t);
}
}
});
ConsoleHelper.WriteSuccessLine($"Tables Created Successfully!");
Console.WriteLine();
}
catch (Exception ex)
{
// 1、若是Mysql,查看常见问题:https://github.com/anjoy8/Blog.Core/issues/148#issue-776281770
//2、若是Oracle,查看常见问题:https://github.com/anjoy8/Blog.Core/issues/148#issuecomment-752340231
throw new Exception("错误:" + ex.Message);
}
}
///
/// 初始化 多租户
///
///
///
public static async Task TenantSeedAsync(DBContext dbContext)
{
if (BaseDBConfig.MutiConnectionString.Where(x => x.ConnId != MainDb.CurrentDbConnId).Any())
{
Console.WriteLine($@"Init Multi Tenant Db");
foreach (MutiDBOperate tenant in BaseDBConfig.MutiConnectionString.Where(x => x.ConnId != MainDb.CurrentDbConnId))
{
Console.WriteLine($@"Init Multi Tenant Db : {tenant.ConnId}");
ConnectionConfig connectionConfig = new ConnectionConfig()
{
ConfigId = tenant.ConnId,
ConnectionString = tenant.Connection,
IsAutoCloseConnection = true,
MoreSettings = new ConnMoreSettings()
{
IsAutoRemoveDataCache = true
},
DbType = (DbType)tenant.DbType,
};
await InitTenantSeedAsync(dbContext.Db.AsTenant(), connectionConfig);
}
Console.WriteLine(DateTime.Now + $@"Init Multi Tenant Db Finish");
}
//tenants = await myContext.Db.Queryable().Where(s => s.TenantType == TenantTypeEnum.Tables).ToListAsync();
//if (tenants.Any())
//{
// await InitTenantSeedAsync(myContext, tenants);
//}
}
#region 多租户 多库 初始化
///
/// 初始化多库
///
///
///
///
public static async Task InitTenantSeedAsync(ITenant itenant, ConnectionConfig config)
{
Console.WriteLine(DateTime.Now + $@"Init Multi Tenant Db");
//itenant.RemoveConnection(config.ConfigId);
itenant.AddConnection(config);
var db = itenant.GetConnectionScope(config.ConfigId);
db.DbMaintenance.CreateDatabase();
ConsoleHelper.WriteSuccessLine($"Init Multi Tenant Db : {config.ConfigId} Database created successfully!");
Console.WriteLine($@"Init Multi Tenant Db : {config.ConfigId} Create Tables");
// 获取所有实体表-初始化租户业务表
var entityTypes = TenantUtil.GetTenantEntityTypes(TenantTypeEnum.Db);
if (!entityTypes.Any()) return;
foreach (var entityType in entityTypes)
{
//var splitTable = entityType.GetCustomAttribute();
//if (splitTable == null)
db.CodeFirst.InitTables(entityType);
//else
// db.CodeFirst.SplitTables().InitTables(entityType);
Console.WriteLine(entityType.Name);
}
Console.WriteLine(DateTime.Now + $@"Init Multi Tenant Db Finish");
//多租户初始化种子数据
//await TenantSeedDataAsync(db, TenantTypeEnum.Db);
}
#endregion
//private static async Task TenantSeedDataAsync(ISqlSugarClient db, TenantTypeEnum tenantType)
//{
// // 获取所有种子配置-初始化数据
// var seedDataTypes = AssemblysExtensions.GetAllAssemblies().SelectMany(s => s.DefinedTypes)
// .Where(u => !u.IsInterface && !u.IsAbstract && u.IsClass)
// .Where(u =>
// {
// var esd = u.GetInterfaces().FirstOrDefault(i => i.HasImplementedRawGeneric(typeof(IEntitySeedData<>)));
// if (esd is null)
// {
// return false;
// }
// var eType = esd.GenericTypeArguments[0];
// return eType.IsTenantEntity(tenantType);
// });
// if (!seedDataTypes.Any()) return;
// foreach (var seedType in seedDataTypes)
// {
// dynamic instance = Activator.CreateInstance(seedType);
// //初始化数据
// {
// var seedData = instance.InitSeedData();
// if (seedData != null && Enumerable.Any(seedData))
// {
// var entityType = seedType.GetInterfaces().First().GetGenericArguments().First();
// var entity = db.EntityMaintenance.GetEntityInfo(entityType);
// if (!await db.Queryable(entity.DbTableName, "").AnyAsync())
// {
// await db.Insertable(Enumerable.ToList(seedData)).ExecuteCommandAsync();
// Console.WriteLine($"Table:{entity.DbTableName} init success!");
// }
// }
// }
// //种子数据
// {
// var seedData = instance.SeedData();
// if (seedData != null && Enumerable.Any(seedData))
// {
// var entityType = seedType.GetInterfaces().First().GetGenericArguments().First();
// var entity = db.EntityMaintenance.GetEntityInfo(entityType);
// await db.Storageable(Enumerable.ToList(seedData)).ExecuteCommandAsync();
// Console.WriteLine($"Table:{entity.DbTableName} seedData success!");
// }
// }
// //自定义处理
// {
// await instance.CustomizeSeedData(db);
// }
// }
//}
}
}