初创

liaoxujun@qq.com

2024-02-28 9f8e542b9b74fe18a6a4e0a00fef4b50e3e62581

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/***********************************************************************
 *            Project: CoreCms
 *        ProjectName: 核心内容管理系统                                
 *                Web: https://www.corecms.net                      
 *             Author: 大灰灰                                          
 *              Email: jianweie@163.com                                
 *         CreateTime: 2021/1/31 21:45:10
 *        Description: 暂无
 ***********************************************************************/
 
using System.Collections.Generic;
using System.Linq;
using CoreCms.Net.CodeGenerator;
using CoreCms.Net.IRepository;
using CoreCms.Net.IRepository.UnitOfWork;
using CoreCms.Net.Model.ViewModels.Basics;
using SqlSugar;
 
namespace CoreCms.Net.Repository.CodeGenerator
{
    public class CodeGeneratorRepository : BaseRepository<object>, ICodeGeneratorRepository
    {
        public CodeGeneratorRepository(IUnitOfWork unitOfWork) : base(unitOfWork)
        {
        }
 
        /// <summary>
        ///     获取所有的表
        /// </summary>
        /// <returns></returns>
        public List<DbTableInfo> GetDbTables()
        {
            var tables = DbClient.DbMaintenance.GetTableInfoList(false).OrderBy(p => p.Name).ToList();
            var views = DbClient.DbMaintenance.GetViewInfoList(false).OrderBy(p => p.Name).ToList();
            if (!views.Any()) return tables;
            var newList = tables.Union(views).ToList();
            return newList;
        }
 
        /// <summary>
        ///     获取表下面所有的字段
        /// </summary>
        /// <param name="tableName"></param>
        /// <returns></returns>
        public List<DbColumnInfo> GetDbTablesColumns(string tableName)
        {
            var columns = DbClient.DbMaintenance.GetColumnInfosByTableName(tableName, false);
            return columns;
        }
 
 
        /// <summary>
        ///     自动生成代码
        /// </summary>
        /// <param name="tableName"></param>
        /// <param name="fileType"></param>
        /// <returns></returns>
        public byte[] CodeGen(string tableName, string fileType, bool isVue = false)
        {
            var tables = DbClient.DbMaintenance.GetTableInfoList(false);
            var views = DbClient.DbMaintenance.GetViewInfoList(false);
            var tb = tables.Find(p => p.Name == tableName) ?? views.Find(p => p.Name == tableName);
            if (tb == null)
            {
                return null;
            }
 
            var columns = DbClient.DbMaintenance.GetColumnInfosByTableName(tb.Name, false);
            return GeneratorCodeHelper.CodeGenerator(tb.Name, tb.Description, columns, fileType, isVue);
        }
 
 
        /// <summary>
        ///     自动生成类型的所有数据库代码
        /// </summary>
        /// <param name="tableName"></param>
        /// <param name="fileType"></param>
        /// <returns></returns>
        public byte[] CodeGenByAll(string fileType, bool isVue = false)
        {
            var tables = DbClient.DbMaintenance.GetTableInfoList(false);
            var views = DbClient.DbMaintenance.GetViewInfoList(false);
            var newList = tables;
            if (views.Any()) newList = tables.Union(views).ToList();
 
            var allDb = new List<DbTableInfoAndColumns>();
            newList.ForEach(p =>
            {
                var model = new DbTableInfoAndColumns();
                model.Name = p.Name;
                model.DbObjectType = p.DbObjectType;
                model.Description = p.Description;
                model.columns = DbClient.DbMaintenance.GetColumnInfosByTableName(p.Name, false);
                allDb.Add(model);
            });
 
            if (!allDb.Any())
                return null;
            return GeneratorCodeHelper.CodeGeneratorAll(allDb, fileType, isVue);
        }
    }
}