From 70233af5426b0d1c343ebe87183303a34a9aaa58 Mon Sep 17 00:00:00 2001 From: hutongqing <hutongqing@hnkhzn.com> Date: 星期三, 20 十一月 2024 09:05:38 +0800 Subject: [PATCH] WIDESEAWCS_QuartzJob更新版本 --- WIDESEAWCS_Server/WIDESEAWCS_Core/Seed/DBSeed.cs | 95 ++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 77 insertions(+), 18 deletions(-) diff --git a/WIDESEAWCS_Server/WIDESEAWCS_Core/Seed/DBSeed.cs b/WIDESEAWCS_Server/WIDESEAWCS_Core/Seed/DBSeed.cs index e56ed82..94924fa 100644 --- a/WIDESEAWCS_Server/WIDESEAWCS_Core/Seed/DBSeed.cs +++ b/WIDESEAWCS_Server/WIDESEAWCS_Core/Seed/DBSeed.cs @@ -1,6 +1,7 @@ 锘縰sing Castle.Components.DictionaryAdapter.Xml; using Microsoft.AspNetCore.Mvc.Rendering; using Newtonsoft.Json; +using SkiaSharp; using SqlSugar; using System; using System.Collections.Generic; @@ -10,6 +11,7 @@ using System.Reflection; using System.Text; using System.Threading.Tasks; +using WIDESEAWCS_Core.Attributes; using WIDESEAWCS_Core.Const; using WIDESEAWCS_Core.DB; using WIDESEAWCS_Core.Helper; @@ -40,7 +42,7 @@ Console.WriteLine("************ WIDESEA DataBase Set *****************"); - Console.WriteLine($"Master DB ConId: {DBContext.ConnId}"); + //Console.WriteLine($"Master DB ConId: {DBContext.ConnId}"); Console.WriteLine($"Master DB Type: {DBContext.DbType}"); //Console.WriteLine($"Master DB ConnectString: {DBContext.ConnectionString}"); @@ -52,7 +54,7 @@ if (DBContext.DbType != SqlSugar.DbType.Oracle) { dbContext.Db.DbMaintenance.CreateDatabase(); - ConsoleHelper.WriteSuccessLine($"Database created successfully!"); + ConsoleHelper.WriteSuccessLine($"Database Created Successfully!"); } else { @@ -81,28 +83,74 @@ IDbMaintenance dbMaintenance = dbContext.Db.DbMaintenance; if (!dbMaintenance.IsAnyTable(t.Name, false)) { - Console.WriteLine(t.Name); + ConsoleHelper.WriteSuccessLine($"Table [{t.Name}] Created Successfully"); dbContext.Db.CodeFirst.InitTables(t); - //dbContext.DbClient.QueryableByObject(t).ToList() == null + string seedData = FileHelper.ReadFile(string.Format(SeedDataFolder, t.Name), Encoding.UTF8); - - #region Sys_User + #region AddSeedData if (seedData != "涓嶅瓨鍦ㄧ浉搴旂殑鐩綍") { - List<Dictionary<string, object>> dic = JsonConvert.DeserializeObject<List<Dictionary<string, object>>>(seedData); - //dbContext.GetEntityDB<BlogArticle>().InsertRange(JsonHelper.ParseFormByJson<List<BlogArticle>>()); - //Console.WriteLine("Table:BlogArticle created success!"); + List<Dictionary<string, object>> dicFile = JsonConvert.DeserializeObject<List<Dictionary<string, object>>>(seedData); - for (int i = 0; i < dic.Count; i++) + if (dicFile.Count > 0) { - if (dic[i].ContainsKey("CreateDate")) - dic[i]["CreateDate"] = DateTime.Now; - else - dic[i].Add("CreateDate", DateTime.Now); - } + List<Dictionary<string, object>> dic = new List<Dictionary<string, object>>(); - dbContext.Db.Insertable(dic).AS(t.Name).ExecuteCommand(); + List<string> columnNames = dbContext.Db.DbMaintenance.GetColumnInfosByTableName(t.Name, false).Select(x => x.DbColumnName).ToList(); + var a = t.GetProperties().FirstOrDefault(x => !columnNames.Contains(x.Name)); + + List<PropertyInfo> propertyInfos = t.GetProperties().Where(x => columnNames.Contains(x.Name)).ToList(); + for (int j = 0; j < dicFile.Count; j++) + { + Dictionary<string, object> keyValuePairs = new Dictionary<string, object>(); + for (int i = 0; i < propertyInfos.Count; i++) + { + PropertyInfo propertyInfo = propertyInfos[i]; + + SequenceAttribute? sequenceAttirbute = propertyInfo.GetCustomAttribute<SequenceAttribute>(); + if (sequenceAttirbute != null) + { + int count = dbContext.Db.Ado.GetScalar($"SELECT COUNT(*) FROM sys.sequences WHERE name = '{sequenceAttirbute.SequenceName}'").ObjToInt(); + if (count == 0) + { + string sql = $"CREATE SEQUENCE {sequenceAttirbute.SequenceName} AS [int] START WITH {sequenceAttirbute.StartWith} INCREMENT BY {sequenceAttirbute.Increment} MINVALUE {sequenceAttirbute.SeqMinValue} MAXVALUE {sequenceAttirbute.SeqMaxValue} {(sequenceAttirbute.IsCycle ? "CYCLE" : "")} CACHE"; + dbContext.Db.Ado.ExecuteCommand(sql); + } + } + + SugarColumn? sugarColumn = propertyInfo.GetCustomAttribute<SugarColumn>(); + 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 } @@ -116,7 +164,18 @@ for (int i = 0; i < propertyInfos.Count; i++) { PropertyInfo propertyInfo = propertyInfos[i]; - SugarColumn sugarColumn = propertyInfo.GetCustomAttribute<SugarColumn>(); + + SequenceAttribute? sequenceAttirbute = propertyInfo.GetCustomAttribute<SequenceAttribute>(); + if (sequenceAttirbute != null) + { + int count = dbContext.Db.Ado.GetScalar($"SELECT COUNT(*) FROM sys.sequences WHERE name = '{sequenceAttirbute.SequenceName}'").ObjToInt(); + if (count == 0) + { + string sql = $"CREATE SEQUENCE {sequenceAttirbute.SequenceName} AS [int] START WITH {sequenceAttirbute.StartWith} INCREMENT BY {sequenceAttirbute.Increment} MINVALUE {sequenceAttirbute.SeqMinValue} MAXVALUE {sequenceAttirbute.SeqMaxValue} {(sequenceAttirbute.IsCycle ? "CYCLE" : "")} CACHE"; + dbContext.Db.Ado.ExecuteCommand(sql); + } + } + SugarColumn? sugarColumn = propertyInfo.GetCustomAttribute<SugarColumn>(); if (sugarColumn != null) { if (!sugarColumn.IsIgnore) @@ -134,7 +193,7 @@ } } }); - ConsoleHelper.WriteSuccessLine($"Tables created successfully!"); + ConsoleHelper.WriteSuccessLine($"Tables Created Successfully!"); Console.WriteLine(); -- Gitblit v1.9.3