修改

Net6Demo_Api.Entity/Input/HomeInput.cs

@@ -22,6 +22,16 @@ namespace Net6Demo_Api.Entity
         /// </summary>
         [DefaultValue("")]
         public string tablename { get; set; } = "";
+        /// <summary>
+        /// 只生成model
+        /// </summary>
+        [DefaultValue(0)]
+        public int isOnlyEntity { get; set; }
+        /// <summary>
+        /// 是否覆盖
+        /// </summary>
+        [DefaultValue(0)]
+        public int isCover { get; set; }
     }
 
     public class LoginInput

Net6Demo_Api.Util/Helper/CreateCodeFileHelper.cs

@@ -11,7 +11,7 @@ using System.Text;
 
 namespace Net6Demo_Api.Util
 {
-    public class CreateCodeFileHelper: ITransientDependency
+    public class CreateCodeFileHelper : ITransientDependency
     {
         private readonly IDbAccessor _db;
         private readonly IHostEnvironment _env;
@@ -33,19 +33,24 @@ namespace Net6Demo_Api.Util
             }
         }
 
+        private int _isCover = 0;
         /// <summary>
         /// 生成代码文件
         /// </summary>
         /// <param name="tableName"></param>
-        public void CreateCodeFile(string tableName, string controllerName, string moduleName)
+        public void CreateCodeFile(string tableName, string controllerName, string moduleName, int isOnlyEntity = 0, int isCover = 0)
         {
-            var list= GetDataTableWithSql(DbTableSqlDic[_db.DbType]).ToJson().ToObject<List<TableInfo>>();
+            _isCover = isCover;
+            var list = GetDataTableWithSql(DbTableSqlDic[_db.DbType]).ToJson().ToObject<List<TableInfo>>();
             if (!string.IsNullOrEmpty(tableName)) list = list.Where(p => p.Name.ToLower() == tableName.ToLower()).ToList();
             foreach (var item in list)
             {
-                CreateEntityFile(item.Name,item.Description, controllerName, moduleName);
-                CreateBusinessFile(item.Name, controllerName, moduleName);
-                CreateControllerFile(item.Name, controllerName, moduleName);
+                CreateEntityFile(item.Name, item.Description, controllerName, moduleName);
+                if (isOnlyEntity == 0)
+                {
+                    CreateBusinessFile(item.Name, controllerName, moduleName);
+                    CreateControllerFile(item.Name, controllerName, moduleName);
+                }
             }
         }
 
@@ -59,7 +64,7 @@ namespace Net6Demo_Api.Util
             string pname = _env.ApplicationName;
 
             var dbtype = _db.DbType;
-            var columns = GetDataTableWithSql(DbTableColumnSqlDic[dbtype], new Dictionary<string, string>() { { "@tablename" , tableName } } ).ToJson().ToObject<List<TableColumn>>();
+            var columns = GetDataTableWithSql(DbTableColumnSqlDic[dbtype], new Dictionary<string, string>() { { "@tablename", tableName } }).ToJson().ToObject<List<TableColumn>>();
             int index = 0;
             string properties = string.Empty;
             string inputproperties = string.Empty;
@@ -75,15 +80,16 @@ namespace Net6Demo_Api.Util
                 }
 
                 Type type = DbTypeToCsharpType(column.Type);
-                string isNullable = column.IsNullable && type.IsValueType ? "?" : "";
-                string description =string.IsNullOrEmpty(column.Description) ? column.Name : column.Description;
+                //string isNullable = column.IsNullable && type.IsValueType ? "?" : "";
+                string isNullable = column.IsNullable ? "?" : "";
+                string description = string.IsNullOrEmpty(column.Description) ? column.Name : column.Description;
                 properties += $@"
         /// <summary>{description}</summary>{key}
         public {type.Name}{isNullable} {column.Name} {{ get; set; }}
 ";
                 inputproperties += $@"
         /// <summary>{description}</summary>
-        public {type.Name}{isNullable} {(column.Name.StartsWith("F_")? column.Name.Substring(2) : column.Name)} {{ get; set; }}
+        public {type.Name}{isNullable} {(column.Name.StartsWith("F_") ? column.Name.Substring(2) : column.Name)} {{ get; set; }}
 ";
                 viewproperties += $@"
         /// <summary>{description}</summary>
@@ -308,6 +314,7 @@ namespace {pname}.Business
         {
             string path = filepath.Substring(0, filepath.LastIndexOf("\\"));
             if (!Directory.Exists(path)) Directory.CreateDirectory(path);
+            if (File.Exists(filepath) && _isCover == 1) File.Delete(filepath);
             if (!File.Exists(filepath))
             {
                 FileInfo file = new FileInfo(filepath);
@@ -540,7 +547,7 @@ namespace {pname}.Business
         /// <summary>
         /// 数据库类型表sql字典
         /// </summary>
-        private Dictionary<DatabaseType, string> DbTableSqlDic=new Dictionary<DatabaseType, string>()
+        private Dictionary<DatabaseType, string> DbTableSqlDic = new Dictionary<DatabaseType, string>()
         {
             { DatabaseType.SqlServer,
                 "select [Name] = a.name, [Description] = g.value from sys.tables a left join sys.extended_properties g on (a.object_id = g.major_id AND g.minor_id = 0 AND g.name= 'MS_Description')"
@@ -694,5 +701,6 @@ order by a.attnum asc;"
 
             private string _description { get; set; }
         }
+
     }
 }

Net6Demo_Api/Controllers/HomeController.cs

@@ -95,7 +95,7 @@ namespace Net6Demo_Api.Controllers
         [HttpPost]
         public ActionResult BulidCodeFile(CodeInput input)
         {
-            _createHelper.CreateCodeFile(input.tablename, input.controllername, input.modulename);
+            _createHelper.CreateCodeFile(input.tablename, input.controllername, input.modulename, input.isOnlyEntity, input.isCover);
             return Success("成功");
         }