717 lines
27 KiB
C#
717 lines
27 KiB
C#
using System.ArrayExtensions;
|
||
using System.Collections;
|
||
using System.ComponentModel.DataAnnotations.Schema;
|
||
using System.Data;
|
||
using System.Text;
|
||
using System.Text.RegularExpressions;
|
||
using JNPF.Common.Configuration;
|
||
using JNPF.Common.Contracts;
|
||
using JNPF.Common.Core.Manager;
|
||
using JNPF.Common.Core.Manager.Files;
|
||
using JNPF.Common.Dtos.DataBase;
|
||
using JNPF.Common.Enums;
|
||
using JNPF.Common.Extension;
|
||
using JNPF.Common.Filter;
|
||
using JNPF.Common.Security;
|
||
using JNPF.DatabaseAccessor;
|
||
using JNPF.DependencyInjection;
|
||
using JNPF.DynamicApiController;
|
||
using JNPF.FriendlyException;
|
||
using JNPF.Systems.Entitys.Dto.Database;
|
||
using JNPF.Systems.Entitys.Model.DataBase;
|
||
using JNPF.Systems.Entitys.System;
|
||
using JNPF.Systems.Interfaces.System;
|
||
using JNPF.ViewEngine;
|
||
using Mapster;
|
||
using Microsoft.AspNetCore.Authorization;
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.AspNetCore.Mvc;
|
||
using NPOI.SS.Formula.Functions;
|
||
using Org.BouncyCastle.Asn1.Cms;
|
||
using SqlSugar;
|
||
|
||
namespace JNPF.Systems;
|
||
|
||
/// <summary>
|
||
/// 数据管理
|
||
/// 版 本：V3.2
|
||
/// 版 权：拓通智联科技有限公司（http://www.tuotong-tech.com）
|
||
/// 日 期：2021-06-01.
|
||
/// </summary>
|
||
[ApiDescriptionSettings(Tag = "System", Name = "DataModel", Order = 208)]
|
||
[Route("api/system/[controller]")]
|
||
public class DataBaseService : IDynamicApiController, ITransient
|
||
{
|
||
/// <summary>
|
||
/// 服务基础仓储.
|
||
/// </summary>
|
||
public readonly ISqlSugarRepository<DbLinkEntity> _repository;
|
||
|
||
/// <summary>
|
||
/// 数据连接服务.
|
||
/// </summary>
|
||
private readonly IDbLinkService _dbLinkService;
|
||
|
||
/// <summary>
|
||
/// 视图引擎.
|
||
/// </summary>
|
||
private readonly IViewEngine _viewEngine;
|
||
|
||
|
||
/// <summary>
|
||
/// 文件服务.
|
||
/// </summary>
|
||
private readonly IFileManager _fileManager;
|
||
|
||
/// <summary>
|
||
/// 数据库管理.
|
||
/// </summary>
|
||
private readonly IDataBaseManager _dataBaseManager;
|
||
|
||
/// <summary>
|
||
/// 用户管理.
|
||
/// </summary>
|
||
private readonly IUserManager _userManager;
|
||
|
||
/// <summary>
|
||
/// 初始化一个<see cref="DataBaseService"/>类型的新实例.
|
||
/// </summary>
|
||
public DataBaseService(
|
||
ISqlSugarRepository<DbLinkEntity> repository,
|
||
IDbLinkService dbLinkService,
|
||
IFileManager fileManager,
|
||
IDataBaseManager dataBaseManager,
|
||
IUserManager userManager,
|
||
IViewEngine viewEngine)
|
||
{
|
||
_repository = repository;
|
||
_dbLinkService = dbLinkService;
|
||
_fileManager = fileManager;
|
||
_dataBaseManager = dataBaseManager;
|
||
_userManager = userManager;
|
||
_viewEngine = viewEngine;
|
||
}
|
||
|
||
#region GET
|
||
|
||
/// <summary>
|
||
/// 表名列表.
|
||
/// </summary>
|
||
/// <param name="id">连接Id.</param>
|
||
/// <param name="input">过滤条件.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{id}/Tables")]
|
||
public async Task<dynamic> GetList(string id, [FromQuery] KeywordInput input)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(id);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var tables = _dataBaseManager.GetDBTableList(tenantLink);
|
||
//tables = tables.Where((x, i) => tables.FindIndex(z => z.Name == x.Name) == i).ToList();
|
||
//var output = tables.Adapt<List<DatabaseTableListOutput>>();
|
||
if (!string.IsNullOrEmpty(input.keyword))
|
||
{
|
||
var keyword = input.keyword.ToLower();
|
||
tables = tables.FindAll(d => d.table.ToLower().Contains(keyword) || (d.tableName.IsNotEmptyOrNull() && d.tableName.ToLower().Contains(keyword)));
|
||
}
|
||
if (tenantLink.DbType.ToLower().Equals("dm"))
|
||
{
|
||
GetTableCount(tables, tenantLink);
|
||
}
|
||
return new { list = tables.OrderBy(x => x.table).ToList() };
|
||
//try
|
||
//{
|
||
//}
|
||
//catch (Exception ex)
|
||
//{
|
||
// return new { list = new List<DatabaseTableListOutput>() };
|
||
//}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 预览数据.
|
||
/// </summary>
|
||
/// <param name="input">请求参数.</param>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="tableName">表名.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{linkId}/Table/{tableName}/Preview")]
|
||
public async Task<dynamic> GetData([FromQuery] DatabaseTablePreviewQuery input, string linkId, string tableName)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
if (string.IsNullOrEmpty(tableName)) return new PageResult();
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
StringBuilder dbSql = new StringBuilder();
|
||
dbSql.AppendFormat("SELECT * FROM {0} WHERE 1=1", tableName);
|
||
if (!string.IsNullOrEmpty(input.field) && !string.IsNullOrEmpty(input.keyword))
|
||
dbSql.AppendFormat(" AND {0} like '%{1}%'", input.field, input.keyword);
|
||
|
||
return await _dataBaseManager.GetDataTablePage(tenantLink, dbSql.ToString(), input.currentPage, input.pageSize);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 字段列表.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="tableName">表名.</param>
|
||
/// <param name="type">字段类型.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{linkId}/Tables/{tableName}/Fields")]
|
||
public async Task<dynamic> GetFieldList(string linkId, string tableName, [FromQuery] string type)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
if (string.IsNullOrEmpty(tableName)) return new PageResult();
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var data = _dataBaseManager.GetFieldList(tenantLink, tableName).Adapt<List<TableFieldOutput>>();
|
||
if (type.Equals("1"))
|
||
{
|
||
data.ForEach(item =>
|
||
{
|
||
item.field = item.field.ReplaceRegex("^f_", string.Empty).ParseToPascalCase().ToLowerCase();
|
||
});
|
||
}
|
||
|
||
return new { list = data };
|
||
}
|
||
|
||
/// <summary>
|
||
/// 信息.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="tableName">主键值.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{linkId}/Table/{tableName}")]
|
||
public async Task<dynamic> GetInfo(string linkId, string tableName)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
if (string.IsNullOrEmpty(tableName)) return new PageResult();
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var output = _dataBaseManager.GetDataBaseTableInfo(tenantLink, tableName);
|
||
var data = _dataBaseManager.GetData(tenantLink, tableName);
|
||
output.hasTableData = data.Rows.Count > 0;
|
||
return output;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取数据库表字段下拉框列表(弃用).
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="tableName">表名.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{linkId}/Tables/{tableName}/Fields/Selector")]
|
||
public async Task<dynamic> SelectorData(string linkId, string tableName)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
if (string.IsNullOrEmpty(tableName)) return new PageResult();
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var data = _dataBaseManager.GetDBTableList(tenantLink).FindAll(m => m.table == tableName).Adapt<List<DatabaseTableFieldsSelectorOutput>>();
|
||
return new { list = data };
|
||
}
|
||
|
||
/// <summary>
|
||
/// 导出.
|
||
/// </summary>
|
||
/// <param name="linkId">连接ID.</param>
|
||
/// <param name="tableName">表名称.</param>
|
||
/// <returns></returns>
|
||
[HttpGet("{linkId}/Table/{tableName}/Action/Export")]
|
||
public async Task<dynamic> ActionsExport(string linkId, string tableName)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var data = _dataBaseManager.GetDataBaseTableInfo(tenantLink, tableName);
|
||
var jsonStr = data.ToJsonString();
|
||
return await _fileManager.Export(jsonStr, data.tableInfo.table, ExportFileType.bdb);
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region POST
|
||
|
||
/// <summary>
|
||
/// 删除.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="tableName">表名.</param>
|
||
/// <returns></returns>
|
||
[HttpDelete("{linkId}/Table/{tableName}")]
|
||
[UnitOfWork]
|
||
public async Task Delete(string linkId, string tableName)
|
||
{
|
||
if (IsSysTable(tableName))
|
||
throw Oops.Oh(ErrorCode.D1504);
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var data = _dataBaseManager.GetData(tenantLink, tableName);
|
||
if (data.Rows.Count > 0)
|
||
throw Oops.Oh(ErrorCode.D1508);
|
||
if (!_dataBaseManager.Delete(tenantLink, tableName))
|
||
throw Oops.Oh(ErrorCode.D1500);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 新建.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="input">请求参数.</param>
|
||
/// <returns></returns>
|
||
[HttpPost("{linkId}/Table")]
|
||
[UnitOfWork]
|
||
public async Task Create(string linkId, [FromBody] DatabaseTableInfoOutput input)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
if (_dataBaseManager.IsAnyTable(tenantLink, input.tableInfo.newTable))
|
||
throw Oops.Oh(ErrorCode.D1503);
|
||
var tableInfo = input.tableInfo.Adapt<DbTableModel>();
|
||
tableInfo.table = input.tableInfo.newTable;
|
||
var tableFieldList = input.tableFieldList.Adapt<List<DbTableFieldModel>>();
|
||
if (!await _dataBaseManager.Create(tenantLink, tableInfo, tableFieldList))
|
||
throw Oops.Oh(ErrorCode.D1501);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 更新.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="input">请求参数.</param>
|
||
/// <returns></returns>
|
||
[HttpPut("{linkId}/Table")]
|
||
[UnitOfWork]
|
||
public async Task Update(string linkId, [FromBody] DatabaseTableUpInput input)
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
var oldFieldList = _dataBaseManager.GetFieldList(tenantLink, input.tableInfo.table).Adapt<List<TableFieldOutput>>();
|
||
oldFieldList = _dataBaseManager.ViewDataTypeConversion(oldFieldList, _dataBaseManager.ToDbType(tenantLink.DbType));
|
||
var oldTableInfo = _dataBaseManager.GetDBTableList(tenantLink).Find(m => m.table == input.tableInfo.table).Adapt<DbTableModel>();
|
||
var data = _dataBaseManager.GetData(tenantLink, input.tableInfo.table);
|
||
if (data.Rows.Count > 0)
|
||
throw Oops.Oh(ErrorCode.D1508);
|
||
var tableInfo = input.tableInfo.Adapt<DbTableModel>();
|
||
tableInfo.table = input.tableInfo.newTable;
|
||
if (IsSysTable(tableInfo.table))
|
||
throw Oops.Oh(ErrorCode.D1504);
|
||
var tableFieldList = input.tableFieldList.Adapt<List<DbTableFieldModel>>();
|
||
if (!input.tableInfo.table.Equals(input.tableInfo.newTable) && _dataBaseManager.IsAnyTable(tenantLink, input.tableInfo.newTable))
|
||
throw Oops.Oh(ErrorCode.D1503);
|
||
if (!await _dataBaseManager.Update(tenantLink, input.tableInfo.table, tableInfo, tableFieldList))
|
||
{
|
||
await _dataBaseManager.Create(tenantLink, oldTableInfo, oldFieldList.Adapt<List<DbTableFieldModel>>());
|
||
throw Oops.Oh(ErrorCode.D1502);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 更新.
|
||
/// </summary>
|
||
/// <param name="linkId">连接Id.</param>
|
||
/// <param name="input">请求参数.</param>
|
||
/// <returns></returns>
|
||
[HttpPut("{linkId}/addFields")]
|
||
[UnitOfWork]
|
||
public async Task AddFields(string linkId, [FromBody] DatabaseTableUpInput input)
|
||
{
|
||
try
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
var tenantLink = link ?? _dataBaseManager.GetTenantDbLink(_userManager.TenantId, _userManager.TenantDbName);
|
||
_dataBaseManager.AddTableColumn(tenantLink, input.tableInfo.table, input.tableFieldList.Adapt<List<DbTableFieldModel>>());
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
throw Oops.Oh(ErrorCode.D1510);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 导入.
|
||
/// </summary>
|
||
/// <param name="linkid"></param>
|
||
/// <param name="file"></param>
|
||
/// <returns></returns>
|
||
[HttpPost("{linkid}/Action/Import")]
|
||
public async Task ActionsImport(string linkid, IFormFile file)
|
||
{
|
||
var fileType = Path.GetExtension(file.FileName).Replace(".", string.Empty);
|
||
if (!fileType.ToLower().Equals(ExportFileType.bdb.ToString()))
|
||
throw Oops.Oh(ErrorCode.D3006);
|
||
var josn = _fileManager.Import(file);
|
||
var data = josn.ToObject<DatabaseTableInfoOutput>();
|
||
if (data == null || data.tableFieldList == null || data.tableInfo == null)
|
||
throw Oops.Oh(ErrorCode.D3006);
|
||
data.tableInfo.newTable = data.tableInfo.table;
|
||
await Create(linkid, data);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清除表数据
|
||
/// </summary>
|
||
/// <param name="linkId">数据库配置主键</param>
|
||
/// <param name="input"></param>
|
||
/// <returns></returns>
|
||
[HttpPost("{linkId}/clean-data")]
|
||
[AllowAnonymous]
|
||
public async Task<dynamic> CleanTableData(string linkId, DatabaseTableDataCleanInput input)
|
||
{
|
||
ISqlSugarClient? sugarClient = null!;
|
||
if (linkId == "0")
|
||
{
|
||
sugarClient = _repository.AsSugarClient();
|
||
}
|
||
else
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
sugarClient = _dataBaseManager.ChangeDataBase(link);
|
||
}
|
||
return await sugarClient!.Deleteable<object>().AS(input.TableName).ExecuteCommandAsync();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 生成实体类代码
|
||
/// </summary>
|
||
/// <param name="linkId">数据库配置表主键</param>
|
||
/// <param name="input">post输入参数</param>
|
||
/// <returns></returns>
|
||
/// <exception cref="ArgumentNullException"></exception>
|
||
/// <exception cref="ArgumentException"></exception>
|
||
[HttpPost("{linkId}/gen-code")]
|
||
public async Task<dynamic> GenerateCode(string linkId, DatabaseTableDataCleanInput input)
|
||
{
|
||
if (input.TableName == "bas_customer")
|
||
{
|
||
return await GenerateAllEntity(linkId);
|
||
}
|
||
if (input.TableName.IsNullOrWhiteSpace())
|
||
{
|
||
throw new ArgumentNullException(nameof(input.TableName));
|
||
}
|
||
if (!input.TableName.Contains("_"))
|
||
{
|
||
throw new ArgumentException($"表【{input.TableName}】,表名之间必须包含_");
|
||
}
|
||
ISqlSugarClient sugarClient = _repository.AsSugarClient();
|
||
if (linkId != "0")
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
sugarClient = _dataBaseManager.ChangeDataBase(link);
|
||
}
|
||
List<DbEntityInfo> entities = new();
|
||
Dictionary<string, string> nsMapper = GetNsMapper();
|
||
foreach (var tbl in sugarClient.DbMaintenance.GetTableInfoList().Where(t => t.Name == input.TableName))
|
||
{
|
||
var entityName = string.Join("", tbl.Name.Split('_').Select(a => a.ToPascalCase()));
|
||
sugarClient.MappingTables.Add(entityName, tbl.Name);
|
||
DbEntityInfo model = new() { tableName = tbl.Name, descrip = tbl.Description, clsName = entityName };
|
||
var key = tbl.Name.Split('_')[0] + "_";
|
||
if (nsMapper.ContainsKey(key)) model.nsName = nsMapper[key];
|
||
foreach (var field in sugarClient.DbMaintenance.GetColumnInfosByTableName(tbl.Name))
|
||
{
|
||
DbEntityPropInfo col = field.Adapt<DbEntityPropInfo>();
|
||
col.csType = sugarClient.Ado.DbBind.GetPropertyTypeName(field.DataType);
|
||
col.propName = field.DbColumnName;
|
||
model.columns.Add(col);
|
||
}
|
||
var primaryKey = model.columns.FirstOrDefault(a => a.primaryKey);
|
||
if (primaryKey != null)
|
||
{
|
||
model.pkType = primaryKey.csType;
|
||
model.pkName = primaryKey.propName;
|
||
}
|
||
if (model.pkName == "id") model.ignoreCols.Add(model.pkName);
|
||
entities.Add(model);
|
||
}
|
||
var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template", "Entity.vue.vm");
|
||
string tContent = File.ReadAllText(templatePath);
|
||
foreach (var item in entities)
|
||
{
|
||
var tResult = _viewEngine.RunCompileFromCached(tContent, item, builderAction: builder =>
|
||
{
|
||
builder.AddUsing("System.Collections.Generic");
|
||
builder.AddAssemblyReferenceByName("System.Collections");
|
||
});
|
||
var dir = Path.Combine(FileVariable.GenerateCodePath, item.nsName);
|
||
Directory.CreateDirectory(dir);
|
||
File.WriteAllText(Path.Combine(dir, item.clsName + ".cs"), tResult);
|
||
|
||
return tResult;
|
||
}
|
||
|
||
var prefix = input.TableName.Split('_')[0];
|
||
var nsName = nsMapper.ContainsKey(prefix) ? nsMapper[prefix] : "Tnb.Entities";
|
||
var dict = GenerateEntityConfig(sugarClient.DbFirst.Where(input.TableName)).ToClassStringList(nsName);
|
||
return dict.Count > 0 ? dict.First().Value.Replace("\r\n", "\n") : string.Empty;
|
||
}
|
||
#endregion
|
||
|
||
#region PrivateMethod
|
||
|
||
/// <summary>
|
||
/// 是否系统表.
|
||
/// </summary>
|
||
/// <param name="table"></param>
|
||
/// <returns></returns>
|
||
private bool IsSysTable(string table)
|
||
{
|
||
string[] byoTable =
|
||
{
|
||
"base_appdata",
|
||
"base_authorize",
|
||
"base_billrule",
|
||
"base_columnspurview",
|
||
"base_comfields",
|
||
"base_datainterface",
|
||
"base_datainterfacelog",
|
||
"base_dbbackup",
|
||
"base_dblink",
|
||
"base_dictionarydata",
|
||
"base_dictionarytype",
|
||
"base_group",
|
||
"base_imcontent",
|
||
"base_imreply",
|
||
"base_message",
|
||
"base_message_template",
|
||
"base_messagereceive",
|
||
"base_module",
|
||
"base_modulebutton",
|
||
"base_modulecolumn",
|
||
"base_moduledataauthorize",
|
||
"base_moduledataauthorizescheme",
|
||
"base_moduleform",
|
||
"base_organize",
|
||
"base_organize_relation",
|
||
"base_organizeadministrator",
|
||
"base_portal",
|
||
"base_position",
|
||
"base_printdev",
|
||
"base_province",
|
||
"base_role",
|
||
"base_sms_template",
|
||
"base_synthirdinfo",
|
||
"base_sysconfig",
|
||
"base_syslog",
|
||
"base_timetask",
|
||
"base_timetasklog",
|
||
"base_user",
|
||
"base_userrelation",
|
||
"base_visualdev",
|
||
"base_visualdev_modeldata",
|
||
"blade_visual",
|
||
"blade_visual_category",
|
||
"blade_visual_config",
|
||
"blade_visual_db",
|
||
"blade_visual_map",
|
||
"crm_busines",
|
||
"crm_businesproduct",
|
||
"crm_clue",
|
||
"crm_contract",
|
||
"crm_contractinvoice",
|
||
"crm_contractmoney",
|
||
"crm_contractproduct",
|
||
"crm_customer",
|
||
"crm_customercontacts",
|
||
"crm_followlog",
|
||
"crm_invoice",
|
||
"crm_product",
|
||
"crm_receivable",
|
||
"ext_bigdata",
|
||
"ext_document",
|
||
"ext_documentshare",
|
||
"ext_emailconfig",
|
||
"ext_emailreceive",
|
||
"ext_emailsend",
|
||
"ext_employee",
|
||
"ext_order",
|
||
"ext_orderentry",
|
||
"ext_orderreceivable",
|
||
"ext_projectgantt",
|
||
"ext_schedule",
|
||
"ext_tableexample",
|
||
"ext_worklog",
|
||
"ext_worklogshare",
|
||
"flow_delegate",
|
||
"flow_engine",
|
||
"flow_engineform",
|
||
"flow_enginevisible",
|
||
"flow_task",
|
||
"flow_taskcirculate",
|
||
"flow_tasknode",
|
||
"flow_taskoperator",
|
||
"flow_taskoperatorrecord",
|
||
"wechat_mpeventcontent",
|
||
"wechat_mpmaterial",
|
||
"wechat_mpmessage",
|
||
"wechat_qydepartment",
|
||
"wechat_qymessage",
|
||
"wechat_qyuser",
|
||
"wform_applybanquet",
|
||
"wform_applydelivergoods",
|
||
"wform_applydelivergoodsentry",
|
||
"wform_applymeeting",
|
||
"wform_archivalborrow",
|
||
"wform_articleswarehous",
|
||
"wform_batchpack",
|
||
"wform_batchtable",
|
||
"wform_conbilling",
|
||
"wform_contractapproval",
|
||
"wform_contractapprovalsheet",
|
||
"wform_debitbill",
|
||
"wform_documentapproval",
|
||
"wform_documentsigning",
|
||
"wform_expenseexpenditure",
|
||
"wform_finishedproduct",
|
||
"wform_finishedproductentry",
|
||
"wform_incomerecognition",
|
||
"wform_leaveapply",
|
||
"wform_letterservice",
|
||
"wform_materialrequisition",
|
||
"wform_materialrequisitionentry",
|
||
"wform_monthlyreport",
|
||
"wform_officesupplies",
|
||
"wform_outboundorder",
|
||
"wform_outboundorderentry",
|
||
"wform_outgoingapply",
|
||
"wform_paydistribution",
|
||
"wform_paymentapply",
|
||
"wform_postbatchtab",
|
||
"wform_procurementmaterial",
|
||
"wform_procurementmaterialentry",
|
||
"wform_purchaselist",
|
||
"wform_purchaselistentry",
|
||
"wform_quotationapproval",
|
||
"wform_receiptprocessing",
|
||
"wform_receiptsign",
|
||
"wform_rewardpunishment",
|
||
"wform_salesorder",
|
||
"wform_salesorderentry",
|
||
"wform_salessupport",
|
||
"wform_staffovertime",
|
||
"wform_supplementcard",
|
||
"wform_travelapply",
|
||
"wform_travelreimbursement",
|
||
"wform_vehicleapply",
|
||
"wform_violationhandling",
|
||
"wform_warehousereceipt",
|
||
"wform_warehousereceiptentry",
|
||
"wform_workcontactsheet"
|
||
};
|
||
return ((IList)byoTable).Contains(table.ToLower());
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取表条数.
|
||
/// </summary>
|
||
/// <param name="tableList"></param>
|
||
private void GetTableCount(List<DatabaseTableListOutput> tableList, DbLinkEntity link)
|
||
{
|
||
var _sqlSugarClient = _dataBaseManager.ChangeDataBase(link);
|
||
foreach (var item in tableList)
|
||
{
|
||
try
|
||
{
|
||
item.sum = _sqlSugarClient.Queryable<dynamic>().AS(item.table).Count();
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
item.sum = 0;
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 生成所有表的实体类文件
|
||
/// </summary>
|
||
/// <param name="linkId"></param>
|
||
/// <returns></returns>
|
||
private async Task<dynamic> GenerateAllEntity(string linkId)
|
||
{
|
||
ISqlSugarClient sugarClient = _repository.AsSugarClient();
|
||
if (linkId != "0")
|
||
{
|
||
var link = await _dbLinkService.GetInfo(linkId);
|
||
sugarClient = _dataBaseManager.ChangeDataBase(link);
|
||
}
|
||
foreach (var item in sugarClient.DbMaintenance.GetTableInfoList())
|
||
{
|
||
string entityName = string.Join("", item.Name.Split('_').Select(a => a.ToPascalCase()));
|
||
sugarClient.MappingTables.Add(entityName, item.Name);
|
||
}
|
||
Dictionary<string, string> nsMapper = GetNsMapper();
|
||
|
||
List<string> dirs = new List<string>();
|
||
foreach (string key in nsMapper.Keys)
|
||
{
|
||
if (key != "eqp_") continue;
|
||
var nsName = nsMapper[key];
|
||
var dir = Path.Combine(FileVariable.GenerateCodePath, nsName);
|
||
dirs.Add(dir);
|
||
if (!Directory.Exists(dir))
|
||
Directory.CreateDirectory(dir);
|
||
else
|
||
Directory.GetFiles(dir).ToList().ForEach(File.Delete);
|
||
GenerateEntityConfig(sugarClient.DbFirst.Where(t => t.StartsWith(key))).CreateClassFile(dir, nsName);
|
||
}
|
||
string content = string.Empty;
|
||
foreach (var dir in dirs)
|
||
{
|
||
var files = Directory.GetFiles(dir, "*.cs").ToList();
|
||
files.ForEach(f => File.WriteAllText(f, File.ReadAllText(f).Replace("\r\n", "\n")));
|
||
if (files.Count > 0 && string.IsNullOrEmpty(content))
|
||
{
|
||
content = File.ReadAllText(files[0]);
|
||
}
|
||
}
|
||
return content;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 表前缀与命名空间的映射关系
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
private Dictionary<string, string> GetNsMapper()
|
||
{
|
||
Dictionary<string, string> nsMapper = new()
|
||
{
|
||
{"bas_", "Tnb.BasicData.Entities" },
|
||
{"prd_", "Tnb.ProductionMgr.Entities" },
|
||
{"wms_", "Tnb.WarehouseMgr.Entities" },
|
||
{"eqp_", "Tnb.EquipMgr.Entities" },
|
||
{"tool_", "Tnb.EquipMgr.Entities" },
|
||
{"qc_", "Tnb.QcMgr.Entities" },
|
||
};
|
||
return nsMapper;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 生成实体类配置
|
||
/// </summary>
|
||
/// <param name="db"></param>
|
||
/// <returns></returns>
|
||
private IDbFirst GenerateEntityConfig(IDbFirst db)
|
||
{
|
||
db.IsCreateAttribute()
|
||
.StringNullable()
|
||
.SettingPropertyTemplate((col, tpl, type) =>
|
||
{
|
||
string sugarColumnStr = "";
|
||
List<string> sugarAttrs = new List<string>();
|
||
if (col.IsPrimarykey)
|
||
sugarAttrs.Add("IsPrimaryKey=true");
|
||
if (col.IsIdentity)
|
||
sugarAttrs.Add("IsIdentity=true");
|
||
if (sugarAttrs.Count > 0)
|
||
sugarColumnStr = $"\r\n [SugarColumn({string.Join(", ", sugarAttrs)})]";
|
||
var txt = tpl.Replace("{PropertyType}", type).Replace("{PropertyName}", col.DbColumnName).Replace("{SugarColumn}", sugarColumnStr);
|
||
if (col.DbColumnName == "id")
|
||
{
|
||
if (type == "string") txt = txt.TrimEnd('\n', '\r') + " = SnowflakeIdHelper.NextId();\r\n";
|
||
else if (type == "long") txt = txt.TrimEnd('\n', '\r') + " = YitIdHelper.NextId();\r\n";
|
||
}
|
||
else if (type == "string") txt = txt.TrimEnd('\n', '\r') + " = string.Empty;\r\n";
|
||
return txt;
|
||
})
|
||
.SettingNamespaceTemplate(tpl => tpl + "using JNPF.Common.Security;\r\n");
|
||
return db;
|
||
}
|
||
|
||
#endregion
|
||
} |