dengjunjie
2024-12-05 46e98339480d853fc78a014c34d7ff9fcaf13890
项目代码/WMS/WIDESEA_WMSServer/WIDESEA_Core/Seed/DBSeed.cs
@@ -27,7 +27,7 @@
        /// <param name="myContext"></param>
        /// <param name="WebRootPath"></param>
        /// <returns></returns>
        public static async Task SeedAsync(DBContext dbContext, string WebRootPath)
        public static void SeedAsync(DBContext dbContext, string WebRootPath)
        {
            try
            {
@@ -40,9 +40,7 @@
                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();
@@ -52,7 +50,7 @@
                if (DBContext.DbType != SqlSugar.DbType.Oracle)
                {
                    dbContext.Db.DbMaintenance.CreateDatabase();
                    ConsoleHelper.WriteSuccessLine($"Database created successfully!");
                    ConsoleHelper.WriteSuccessLine($"Database Created Successfully!");
                }
                else
                {
@@ -65,20 +63,13 @@
                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 referencedAssemblies = 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<SugarTable>() != null)
                    .ToList();
                //{
                //    List<string> models = modelTypes.Select(x => x.Name).ToList();
                //    List<DbTableInfo> dbTableInfos = dbContext.Db.DbMaintenance.GetTableInfoList(false);
                //    string[] dropTables = dbTableInfos.Select(x => x.Name).Where(x => !models.Contains(x)).ToArray();
                //    dbContext.Db.DbMaintenance.DropTable(dropTables);
                //}
                modelTypes.ForEach(t =>
                {
@@ -88,28 +79,62 @@
                    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 != null && 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];
                                        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
                    }
@@ -123,7 +148,7 @@
                            for (int i = 0; i < propertyInfos.Count; i++)
                            {
                                PropertyInfo propertyInfo = propertyInfos[i];
                                SugarColumn sugarColumn = propertyInfo.GetCustomAttribute<SugarColumn>();
                                SugarColumn? sugarColumn = propertyInfo.GetCustomAttribute<SugarColumn>();
                                if (sugarColumn != null)
                                {
                                    if (!sugarColumn.IsIgnore)
@@ -141,7 +166,7 @@
                        }
                    }
                });
                ConsoleHelper.WriteSuccessLine($"Tables created successfully!");
                ConsoleHelper.WriteSuccessLine($"Tables Created Successfully!");
                Console.WriteLine();