Update Demo

Src/Asp.Net/SugarCodeGeneration/Program.cs

@@ -5,34 +5,139 @@ using System.Diagnostics;
 using System.IO;
 using System.Linq;
 using System.Text;
+using System.Threading.Tasks;
 using static System.Net.Mime.MediaTypeNames;
 
 namespace SugarCodeGeneration
 {
     class Program
     {
-        private const SqlSugar.DbType sqlServer = SqlSugar.DbType.SqlServer;
-        private const string projectName = "SugarCodeGeneration";
-        private const string classPath= "Models"; 
-        private const string classNamespace = "MyTest";
-        private const string connectionString = "server=.;uid=sa;pwd=@jhl85661501;database=SqlSugar4XTest";
 
+        const SqlSugar.DbType dbType = SqlSugar.DbType.SqlServer;
+        const string connectionString = "server=.;uid=sa;pwd=@jhl85661501;database=SqlSugar4XTest";
+        static List<Task> CsprojList = new List<Task>();
+        static SqlSugar.SqlSugarClient GetDB()
+        {
+
+            return new SqlSugar.SqlSugarClient(new SqlSugar.ConnectionConfig()
+            {
+                DbType = dbType,
+                ConnectionString = connectionString,
+                IsAutoCloseConnection = true,
+                InitKeyType=SqlSugar.InitKeyType.Attribute
+            });
+        }
         static void Main(string[] args)
         {
-            //Generation model
-            SqlSugar.SqlSugarClient db = new SqlSugar.SqlSugarClient(new SqlSugar.ConnectionConfig() {
-                DbType = sqlServer,
+            //生成实体
+            GenerationClass();
+            Console.WriteLine("实体创建成功");
+            Console.WriteLine("");
+
+            //生成DbContext
+            GenerationDContext();
+            Console.WriteLine("DbContext创建成功");
+            Console.WriteLine("");
+
+            //生成BLL类
+            GenerationBLL();
+            Console.WriteLine("Bll创建成功");
+            Console.WriteLine("");
+
+            //修改解决方案
+            UpdateCsproj();
+            Console.WriteLine("项目解决方案修改成功");
+            Console.ReadKey();
+
+            //test bll
+             
+        }
+
+
+        /// <summary>
+        /// 生成BLL
+        /// </summary>
+        private static void GenerationBLL()
+        {
+            var db = GetDB();
+
+            //配置参数
+            var templatePath = Methods.GetCurrentProjectPath + "\\Template\\Bll.txt";//bll模版地址
+            var bllProjectName = "SugarCodeGeneration";//具体项目
+            var bllPath = "BLL";//文件目录
+            var savePath = Methods.GetSlnPath + "\\" + bllProjectName + "\\" + bllPath;//保存目录
+            var tables = db.DbMaintenance.GetTableInfoList().Where(it => it.Name == "Student" || it.Name == "School").Select(it => it.Name).ToList();
+
+            //下面代码不动
+            Methods.CreateBLL(templatePath, savePath, tables);
+            AddTask(bllProjectName, bllPath);
+        }
+
+
+
+        /// <summary>
+        /// 生成DbContext
+        /// </summary>
+        private static void GenerationDContext()
+        {
+            var db = GetDB();
+
+
+            //配置参数
+            var templatePath = Methods.GetCurrentProjectPath + "\\Template\\DbContext.txt";//dbcontexts模版文件
+            var contextProjectName = "SugarCodeGeneration";//DbContext所在项目
+            var contextPath = "DbContext";//dbcontext存储目录
+            var savePath = Methods.GetSlnPath + "\\" + contextProjectName + "\\" + contextPath+"\\DbContext.cs";//具体文件名
+            var tables = db.DbMaintenance.GetTableInfoList().Where(it => it.Name == "Student" || it.Name == "School").Select(it => it.Name).ToList();
+
+            //下面代码不动
+            var model = new DbContextParameter{
                 ConnectionString = connectionString,
-                IsAutoCloseConnection = true
+                DbType = dbType,
+                Tables = tables
+            };
+
+
+            Methods.CreateDbContext(templatePath,savePath,model);
+            AddTask(contextPath, contextProjectName);
+        }
+
+        /// <summary>
+        /// 生成实体类
+        /// </summary>
+        private static void GenerationClass()
+        {
+
+            string classProjectName = "SugarCodeGeneration";//实体类项目名称
+            string classPath = "Models";//生成的目录
+            string classNamespace = "MyTest";//实体命名空间
+            var db = GetDB();
+            var classDirectory = Methods.GetSlnPath + "\\" + classProjectName + "\\" + classPath.TrimStart('\\');
+
+            //如果生成全部可以把Where去掉
+            db.DbFirst.Where("Student", "School").IsCreateAttribute().CreateClassFile(classDirectory, classNamespace);
+
+            AddTask(classPath, classProjectName);
+        }
+
+        /// <summary>
+        ///  修改解决方案
+        /// </summary>
+        private static void UpdateCsproj()
+        {
+            foreach (var item in CsprojList)
+            {
+                item.Start();
+            }
+        }
+
+        private static void AddTask(string bllProjectName, string bllPath)
+        {
+            var task = new Task(() =>
+            {
+                Methods.AddCsproj(bllPath, bllProjectName);
             });
-            var classDirectory = Methods.GetSlnPath +"\\"+projectName+"\\"+ classPath.TrimStart('\\');
-           
-            //if all then remove .Where
-            db.DbFirst.Where("Student","School").CreateClassFile(classDirectory, classNamespace);
-
-            Methods.AddCsproj(classPath, projectName);
-
-            //Generation  DbContext
+            CsprojList.Add(task);
         }
     }
 }