|
|
|
|
using zzz.Common;
|
|
|
|
|
using SqlSugar;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Reflection;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace zzz.Services
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 种子数据 服务
|
|
|
|
|
/// </summary>
|
|
|
|
|
public class DBSeed: IDBSeed
|
|
|
|
|
{
|
|
|
|
|
private readonly ISqlSugarClient _sqlSugarClient;
|
|
|
|
|
private static string SeedDataFolder = "BlogCore.Data.json/{0}.tsv";
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 构造
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="sqlSugarClient"></param>
|
|
|
|
|
public DBSeed(ISqlSugarClient sqlSugarClient)
|
|
|
|
|
{
|
|
|
|
|
_sqlSugarClient = sqlSugarClient;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 异步添加种子数据
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="myContext"></param>
|
|
|
|
|
/// <param name="WebRootPath"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
public async static Task SeedAsync(ISqlSugarClient myContext,string WebRootPath)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
if (string.IsNullOrEmpty(WebRootPath))
|
|
|
|
|
{
|
|
|
|
|
throw new Exception("获取wwwroot路径时,异常!");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//创建数据库
|
|
|
|
|
if (Appsettings.app("SeedDBEnabled").ObjToBool())
|
|
|
|
|
{
|
|
|
|
|
var dbtype = Appsettings.app("DBType").ObjToInt();
|
|
|
|
|
Console.WriteLine($"Create Database...");
|
|
|
|
|
|
|
|
|
|
if (dbtype != Convert.ToInt32(SqlSugar.DbType.Oracle))
|
|
|
|
|
{
|
|
|
|
|
myContext.DbMaintenance.CreateDatabase();
|
|
|
|
|
ConsoleHelper.WriteSuccessLine($"Database created successfully!");
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
//Oracle 数据库不支持该操作
|
|
|
|
|
ConsoleHelper.WriteSuccessLine($"Oracle 数据库不支持该操作,可手动创建Oracle数据库!");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//初始化数据
|
|
|
|
|
if (Appsettings.app("SeedDBDataEnabled").ObjToBool())
|
|
|
|
|
{
|
|
|
|
|
// 创建数据库表,遍历指定命名空间下的class,
|
|
|
|
|
// 注意不要把其他命名空间下的也添加进来。
|
|
|
|
|
Console.WriteLine("Create Tables...");
|
|
|
|
|
|
|
|
|
|
var path = AppDomain.CurrentDomain.RelativeSearchPath ?? AppDomain.CurrentDomain.BaseDirectory;
|
|
|
|
|
var referencedAssemblies = System.IO.Directory.GetFiles(path, "zzz.Model.dll").Select(Assembly.LoadFrom).ToArray();
|
|
|
|
|
var modelTypes = referencedAssemblies
|
|
|
|
|
.SelectMany(a => a.DefinedTypes)
|
|
|
|
|
.Select(type => type.AsType())
|
|
|
|
|
.Where(x => x.IsClass && x.Namespace != null && x.Namespace.Equals("zzz.Model.Entity")).ToList();
|
|
|
|
|
modelTypes.ForEach(t =>
|
|
|
|
|
{
|
|
|
|
|
// 这里只支持添加表,不支持删除
|
|
|
|
|
// 如果想要删除,数据库直接右键删除,或者联系SqlSugar作者;
|
|
|
|
|
if (!myContext.DbMaintenance.IsAnyTable(t.Name))
|
|
|
|
|
{
|
|
|
|
|
Console.WriteLine(t.Name);
|
|
|
|
|
myContext.CodeFirst.InitTables(t);
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
ConsoleHelper.WriteSuccessLine($"Tables created successfully!");
|
|
|
|
|
Console.WriteLine();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Console.WriteLine();
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
throw new Exception(
|
|
|
|
|
$"1、若是Mysql,查看常见问题:https://github.com/anjoy8/zzz/issues/148#issue-776281770 \n" +
|
|
|
|
|
$"2、若是Oracle,查看常见问题:https://github.com/anjoy8/zzz/issues/148#issuecomment-752340231 \n" +
|
|
|
|
|
"3、其他错误:" + ex.Message);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 同步表结构
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="tableName"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
public async Task CreateTable(string tableName)
|
|
|
|
|
{
|
|
|
|
|
if (tableName.IsNotEmptyOrNull())
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
var path = AppDomain.CurrentDomain.RelativeSearchPath ?? AppDomain.CurrentDomain.BaseDirectory;
|
|
|
|
|
var referencedAssemblies = System.IO.Directory.GetFiles(path, "zzz.Model.dll").Select(Assembly.LoadFrom).ToArray();
|
|
|
|
|
var modelTypes = referencedAssemblies
|
|
|
|
|
.SelectMany(a => a.DefinedTypes)
|
|
|
|
|
.Select(type => type.AsType())
|
|
|
|
|
.Where(x => x.IsClass && x.Namespace != null && x.Namespace.Equals("zzz.Model.Entity") && x.Name == tableName)
|
|
|
|
|
.FirstOrDefault();
|
|
|
|
|
_sqlSugarClient.CodeFirst.InitTables(modelTypes);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
throw;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|