zongzhilinglan/Magic.Core/Service/DataBase/DataBaseManager.cs

using Furion;
using Furion.DependencyInjection;
using Furion.DynamicApiController;
using Furion.FriendlyException;
using Furion.ViewEngine;
using Magic.Core.Service.DataBase.Extension;
using Mapster;
using Microsoft.AspNetCore.Mvc;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;

namespace Magic.Core.Service;

/// <summary>
/// 数据库管理
/// </summary>
[ApiDescriptionSettings(Name = "DataBase", Order = 200)]
public class DataBaseManager : IDataBaseManager, IDynamicApiController, ITransient
{
    private readonly ISqlSugarClient _sqlSugarClient;
    private readonly IViewEngine _viewEngine;

    public DataBaseManager(ISqlSugarClient sqlSugarClient, IViewEngine viewEngine)
    {
        _sqlSugarClient = sqlSugarClient;
        _viewEngine = viewEngine;
    }

    /// <summary>
    /// 添加列
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/column/add")]
    public void ColumnAdd(DbColumnInfoInput input)
    {
        DbColumnInfo column = new DbColumnInfo();
        column.ColumnDescription = input.ColumnDescription;
        column.DbColumnName = input.DbColumnName;
        column.IsIdentity = input.IsIdentity == 1;
        column.IsNullable = input.IsNullable == 1;
        column.IsPrimarykey = input.IsPrimarykey == 1;
        column.Length = input.Length;
        column.DecimalDigits = input.DecimalDigits;
        column.DataType = input.DataType;
        _sqlSugarClient.DbMaintenance.AddColumn(input.TableName, column);
        _sqlSugarClient.DbMaintenance.AddColumnRemarkEx(input.DbColumnName, input.TableName, input.ColumnDescription);
        if (column.IsPrimarykey)
        {
            _sqlSugarClient.DbMaintenance.AddPrimaryKey(input.TableName, input.DbColumnName);
        }
    }

    /// <summary>
    /// 删除列
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/column/delete")]
    public void ColumnDelete(DbColumnInfoOutput input)
    {
        _sqlSugarClient.DbMaintenance.DropColumn(input.TableName, input.DbColumnName);
    }

    /// <summary>
    /// 编辑列
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/column/edit")]
    public void ColumnEdit(EditColumnInput input)
    {
        _sqlSugarClient.DbMaintenance.RenameColumn(input.TableName, input.OldName, input.DbColumnName);
        if (_sqlSugarClient.DbMaintenance.IsAnyColumnRemarkEx(input.DbColumnName, input.TableName))
        {
            _sqlSugarClient.DbMaintenance.DeleteColumnRemark(input.DbColumnName, input.TableName);
        }
        _sqlSugarClient.DbMaintenance.AddColumnRemarkEx(input.DbColumnName, input.TableName, string.IsNullOrWhiteSpace(input.ColumnDescription) ? input.DbColumnName : input.ColumnDescription);
    }

    /// <summary>
    /// 获取表字段
    /// </summary>
    /// <param name="tableName"></param>
    /// <returns></returns>
    [HttpGet("/dataBase/columnInfoList")]
    public List<DbColumnInfoOutput> GetColumnInfosByTableName(string tableName)
    {
        if (string.IsNullOrWhiteSpace(tableName))
            return new List<DbColumnInfoOutput>();
        return _sqlSugarClient.DbMaintenance.GetColumnInfosByTableName(tableName, false).Adapt<List<DbColumnInfoOutput>>();
    }

    /// <summary>
    /// 获取所有表
    /// </summary>
    /// <returns></returns>
    [HttpGet("/dataBase/tableInfoList")]
    public List<DbTableInfo> GetTableInfoList()
    {
        return _sqlSugarClient.DbMaintenance.GetTableInfoList(false);
    }

    /// <summary>
    /// 新增表
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/table/add")]
    public void TableAdd(DbTableInfoInput input)
    {
        List<DbColumnInfo> columns = new List<DbColumnInfo>();
        if (input.DbColumnInfoList == null || !input.DbColumnInfoList.Any())
        {
            throw Oops.Oh(ErrorCode.db1000);
        }
        input.DbColumnInfoList.ForEach(m =>
        {
            columns.Add(new DbColumnInfo
            {
                DbColumnName = m.DbColumnName,
                DataType = m.DataType,
                Length = m.Length,
                ColumnDescription = m.ColumnDescription,
                IsNullable = m.IsNullable == 1,
                IsIdentity = m.IsIdentity == 1,
                IsPrimarykey = m.IsPrimarykey == 1,
                DecimalDigits = m.DecimalDigits
            });
        });
        _sqlSugarClient.DbMaintenance.CreateTable(input.Name, columns, false);
        _sqlSugarClient.DbMaintenance.AddTableRemark(input.Name, input.Description);
        if (columns.Any(m => m.IsPrimarykey))
        {
            _sqlSugarClient.DbMaintenance.AddPrimaryKey(input.Name, columns.FirstOrDefault(m => m.IsPrimarykey).DbColumnName);
        }
        input.DbColumnInfoList.ForEach(m =>
        {
            _sqlSugarClient.DbMaintenance.AddColumnRemarkEx(m.DbColumnName, input.Name, m.ColumnDescription);
        });
    }

    /// <summary>
    /// 删除表
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/table/delete")]
    public void TableDelete(DbTableInfo input)
    {
        _sqlSugarClient.DbMaintenance.DropTable(input.Name);
    }

    /// <summary>
    /// 编辑表
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/table/edit")]
    public void TableEdit(EditTableInput input)
    {
        _sqlSugarClient.DbMaintenance.RenameTable(input.OldName, input.Name);
        if (_sqlSugarClient.DbMaintenance.IsAnyTableRemarkEx(input.Name))
        {
            _sqlSugarClient.DbMaintenance.DeleteTableRemark(input.Name);
        }
        _sqlSugarClient.DbMaintenance.AddTableRemark(input.Name, input.Description);
    }

    /// <summary>
    /// 生成实体
    /// </summary>
    /// <param name="input"></param>
    [HttpPost("/table/createEntity")]
    public void CreateEntity(CreateEntityInput input)
    {
        Type baseType = Type.GetType($"Magic.Core.Entity.{input.BaseClassName}");

        if (baseType.IsNullOrZero())
            throw Oops.Oh("父类不存在");

        var baseTypeProperties = baseType.GetProperties().Select(m => m.Name).ToList();

        input.Position = string.IsNullOrWhiteSpace(input.Position) ? "Magic.Application" : input.Position;
        input.BaseClassName = string.IsNullOrWhiteSpace(input.BaseClassName) ? "" : $" : {input.BaseClassName}";
        var templatePath = GetTemplatePath();
        var targetPath = GetTargetPath(input);
        DbTableInfo dbTableInfo = _sqlSugarClient.DbMaintenance.GetTableInfoList(false).FirstOrDefault(m => m.Name == input.TableName);
        if (dbTableInfo == null)
            throw Oops.Oh(ErrorCode.db1001);
        List<DbColumnInfo> dbColumnInfos = _sqlSugarClient.DbMaintenance.GetColumnInfosByTableName(input.TableName, false).Where(m => !baseTypeProperties.Contains(m.DbColumnName)).ToList();
        dbColumnInfos.ForEach(m =>
        {
            m.DataType = CodeGenUtil.ConvertDataType(m.DataType);
        });
        var tContent = File.ReadAllText(templatePath);
        var tResult = _viewEngine.RunCompileFromCached(tContent, new
        {
            input.TableName,
            input.EntityName,
            input.BaseClassName,
            dbTableInfo.Description,
            TableField = dbColumnInfos
        });

        #region 检查目录是否存在

        FileInfo fileInfo = new FileInfo(targetPath);
        string entityPath = fileInfo.DirectoryName;
        if (!Directory.Exists(entityPath))
        {
            Directory.CreateDirectory(entityPath);
        }

        #endregion 检查目录是否存在

        File.WriteAllText(targetPath, tResult, Encoding.UTF8);
    }

    /// <summary>
    /// 获取模板文件路径集合
    /// </summary>
    /// <returns></returns>
    private string GetTemplatePath()
    {
        var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
        return Path.Combine(templatePath, "Entity.cs.vm");
    }

    /// <summary>
    /// 设置生成文件路径
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    private string GetTargetPath(CreateEntityInput input)
    {
        var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.Position, "Entity");
        var entityPath = Path.Combine(backendPath, input.EntityName + ".cs");
        return entityPath;
    }
}