OmegaAi/BackEnd/OmegaSource/Services/DatabaseInfoService.cs

using Attribute;
using Model;
using Model.Base;
using Repository;
using Service;
using Microsoft.AspNetCore.Mvc;
using Vo.Admin;
using Dto.Admin;
using Custom;
using Common;
using Infrastructure;
using Tasks;


namespace Services
{
    /// <summary>
    /// 数据库配置Service业务层处理
    /// </summary>
    [AppService(ServiceType = typeof(IDatabaseInfoService), ServiceLifetime = LifeTime.Transient)]
    public class DatabaseInfoService : BaseService<DatabaseInfo>, IDatabaseInfoService
    {
        private readonly IDatabaseTableService _DatabaseTableService;
        private readonly IDatabaseFieldService _DatabaseFieldService;
        private readonly IDatabaseFieldIndexService _DatabaseFieldIndexService;
        private readonly IDatabaseFieldTabsService _DatabaseFieldTabsService;

        public DatabaseInfoService(IDatabaseTableService DatabaseTableService, IDatabaseFieldService DatabaseFieldService, IDatabaseFieldIndexService DatabaseFieldIndexService, IDatabaseFieldTabsService DatabaseFieldTabsService)
        {
            _DatabaseTableService = DatabaseTableService;
            _DatabaseFieldService = DatabaseFieldService;
            _DatabaseFieldIndexService = DatabaseFieldIndexService;
            _DatabaseFieldTabsService = DatabaseFieldTabsService;
        }

        /// <summary>
        /// 数据库配置-列表
        /// </summary>
        /// <param name="param">参数请求体</param>
        /// <param name="page">分页参数</param>
        /// <returns>列表</returns>
        public PagedInfo<GetDatabaseInfoListVo> getDatabaseInfoList([FromQuery] PagerInfo page, [FromQuery] DatabaseInfo param)
        {
            //拼装查询条件
            var predicate = Expressionable.Create<DatabaseInfo>();
            predicate = predicate.AndIF(param.projectId > 0, m => m.projectId == param.projectId);
            predicate = predicate.AndIF(!string.IsNullOrEmpty(param.dbName), m => m.dbName.Contains(param.dbName));
            predicate = predicate.AndIF(!string.IsNullOrEmpty(param.hostAddress), m => m.hostAddress.Contains(param.hostAddress));
            var response = Queryable()
                .Where(predicate.ToExpression())
                .OrderByDescending(m => m.id)
                .ToPage<DatabaseInfo, GetDatabaseInfoListVo>(page);
            return response;
        }



        /// <summary>
        /// 生成SQL脚本
        /// </summary>
        /// <param name="param">参数请求体</param>
        /// <returns>生成</returns>
        public string makeDatabaseInfo([FromBody] MakeDatabaseInfoDto param, List<DatabaseTable> tables, List<DatabaseField> fields, TokenModel loginUser)
        {
            DatabaseInfo databaseInfo = GetById(param.id) ?? new DatabaseInfo();
            foreach(DatabaseTable table in tables)
            {
                table.fields = _DatabaseFieldService.GetList(m => m.tableId == table.id);
                table.fieldIndexes = _DatabaseFieldIndexService.GetList(m => m.tableId == table.id);
            }

            Dictionary<string, object> pushData = new Dictionary<string, object>();
            pushData.Add("since", DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"));
            pushData.Add("author", loginUser.username);
            pushData.Add("databaseInfo", databaseInfo);
            pushData.Add("databaseTable", tables);
            var modePath = "db";
            if(databaseInfo.dbType == "pgsql")
            {
                modePath = "pg-db";
            }
            return MakeHelper.Make(pushData, modePath, "db", loginUser);
        }


        /// <summary>
        /// 数据库配置-根据ID查询列表
        /// </summary>
        /// <param name="param">参数请求体</param>
        /// <param name="page">分页参数</param>
        /// <returns>根据ID查询列表</returns>
        public PagedInfo<GetDatabaseInfoListByIdsVo> getDatabaseInfoListByIds([FromQuery] PagerInfo page, [FromQuery] GetDatabaseInfoListByIdsDto param)
        {
            //拼装查询条件
            var predicate = Expressionable.Create<DatabaseInfo>();
            if(!string.IsNullOrEmpty(param.dbIdList))
            {
                param.dbIdList = param.dbIdList.Replace("[", "").Replace("]", "");
                var dbIdList = Tools.SpitIntArrary(param.dbIdList).ToList();
                predicate = predicate.AndIF(dbIdList.Count > 0, m => dbIdList.Contains(m.id));
            }
            var response = Queryable()
            .Where(predicate.ToExpression())
            .OrderByDescending(m => m.id)
            .ToPage<DatabaseInfo, GetDatabaseInfoListByIdsVo>(page);
            return response;
        }


        /// <summary>
        /// 克隆已有库
        /// </summary>
        /// <param name="param">参数请求体</param>
        /// <returns>克隆已有库</returns>
        public void copyDatabaseInfo([FromBody] CopyDatabaseInfoDto param)
        {
            List<int> dbIdList = param.sourceDatabaseIdList.ToList();
            var sourceDatabaseInfos = GetList(m => dbIdList.Contains(m.id));
            if(sourceDatabaseInfos.Count > 0)
            {
                foreach(DatabaseInfo sourceDatabaseInfo in sourceDatabaseInfos)
                {
                    DatabaseInfo newDatabaseInfo = new DatabaseInfo();
                    newDatabaseInfo.createTime = DateTime.Now;
                    newDatabaseInfo.dbTitle = sourceDatabaseInfo.dbTitle;
                    newDatabaseInfo.dbName = sourceDatabaseInfo.dbName;
                    newDatabaseInfo.serverId = param.targetServiceId;
                    newDatabaseInfo.envId = sourceDatabaseInfo.envId;
                    newDatabaseInfo.hostAddress = sourceDatabaseInfo.hostAddress;
                    newDatabaseInfo.devHostAddress = sourceDatabaseInfo.devHostAddress;
                    newDatabaseInfo.port = sourceDatabaseInfo.port;
                    newDatabaseInfo.username = sourceDatabaseInfo.username;
                    newDatabaseInfo.pwd = sourceDatabaseInfo.pwd;
                    // newDatabaseInfo.projectId = param.targetServiceId;
                    newDatabaseInfo.devPort = sourceDatabaseInfo.devPort;
                    newDatabaseInfo.devUsername = sourceDatabaseInfo.devUsername;
                    newDatabaseInfo.devPwd = sourceDatabaseInfo.devPwd;
                    newDatabaseInfo.dbType = sourceDatabaseInfo.dbType;
                    int newDatabaseInfoId = InsertReturnIdentity(newDatabaseInfo);

                    var sourceTables = _DatabaseTableService.GetList(m => m.databaseId == sourceDatabaseInfo.id).ToList();
                    foreach(DatabaseTable sourceTable in sourceTables)
                    {
                        DatabaseTable newTable = new DatabaseTable();
                        newTable.databaseId = newDatabaseInfoId;
                        newTable.tableName = sourceTable.tableName;
                        newTable.tableTitle = sourceTable.tableTitle;
                        newTable.createTime = DateTime.Now;
                        int newTableId = _DatabaseTableService.InsertReturnIdentity(newTable);

                        var sourceFieldTabs = _DatabaseFieldTabsService.GetList(m => m.tableId == sourceTable.id).ToList();
                        if(sourceFieldTabs.Count == 0)
                        {
                            // 如果没有分组，则创建一个默认分组
                            DatabaseFieldTabs defaultTab = new DatabaseFieldTabs();
                            defaultTab.tabName = "基本信息";
                            defaultTab.databaseId = newDatabaseInfoId;
                            defaultTab.tableId = newTableId;
                            defaultTab.createTime = DateTime.Now;
                            int defaultTabId = _DatabaseFieldTabsService.InsertReturnIdentity(defaultTab);

                            var sourceFieldsNoTab = _DatabaseFieldService.GetList(m => m.tableId == sourceTable.id && m.tabId == 0).ToList();
                            foreach(DatabaseField sourceField in sourceFieldsNoTab)
                            {
                                DatabaseField newField = new DatabaseField();
                                newField.databaseId = newDatabaseInfoId;
                                newField.tableId = newTableId;
                                newField.tabId = defaultTabId;
                                newField.fieldTitle = sourceField.fieldTitle;
                                newField.fieldName = sourceField.fieldName;
                                newField.fieldType = sourceField.fieldType;
                                newField.fieldLength = sourceField.fieldLength;
                                newField.fieldDefaultValue = sourceField.fieldDefaultValue;
                                newField.fieldDetail = sourceField.fieldDetail;
                                newField.editType = sourceField.editType;
                                newField.validateAttribute = sourceField.validateAttribute;
                                newField.primaryKey = sourceField.primaryKey;
                                newField.autoIncrement = sourceField.autoIncrement;
                                newField.splitFlag = sourceField.splitFlag;
                                newField.showList = sourceField.showList;
                                newField.showQuery = sourceField.showQuery;
                                newField.showEdit = sourceField.showEdit;
                                newField.checkRepeat = sourceField.checkRepeat;
                                newField.exportFlag = sourceField.exportFlag;
                                newField.importFlag = sourceField.importFlag;
                                newField.listEditFlag = sourceField.listEditFlag;
                                newField.listEditPlaceholder = sourceField.listEditPlaceholder;
                                newField.statCondition = sourceField.statCondition;
                                newField.statData = sourceField.statData;
                                newField.linkTableId = sourceField.linkTableId;
                                newField.linkFieldId = sourceField.linkFieldId;
                                newField.apiId = sourceField.apiId;
                                newField.createTime = DateTime.Now;
                                _DatabaseFieldService.Add(newField);
                            }
                        }
                        else
                        {
                            foreach(DatabaseFieldTabs sourceFieldTab in sourceFieldTabs)
                            {
                                DatabaseFieldTabs newFieldTab = new DatabaseFieldTabs();
                                newFieldTab.tabName = sourceFieldTab.tabName;
                                newFieldTab.databaseId = newDatabaseInfoId;
                                newFieldTab.tableId = newTableId;
                                newFieldTab.createTime = DateTime.Now;
                                int newFieldTabId = _DatabaseFieldTabsService.InsertReturnIdentity(newFieldTab);

                                var sourceFields = _DatabaseFieldService.GetList(m => m.tabId == sourceFieldTab.id).ToList();
                                foreach(DatabaseField sourceField in sourceFields)
                                {
                                    DatabaseField newField = new DatabaseField();
                                    newField.databaseId = newDatabaseInfoId;
                                    newField.tableId = newTableId;
                                    newField.tabId = newFieldTabId;
                                    newField.fieldTitle = sourceField.fieldTitle;
                                    newField.fieldName = sourceField.fieldName;
                                    newField.fieldType = sourceField.fieldType;
                                    newField.fieldLength = sourceField.fieldLength;
                                    newField.fieldDefaultValue = sourceField.fieldDefaultValue;
                                    newField.fieldDetail = sourceField.fieldDetail;
                                    newField.editType = sourceField.editType;
                                    newField.validateAttribute = sourceField.validateAttribute;
                                    newField.primaryKey = sourceField.primaryKey;
                                    newField.autoIncrement = sourceField.autoIncrement;
                                    newField.splitFlag = sourceField.splitFlag;
                                    newField.showList = sourceField.showList;
                                    newField.showQuery = sourceField.showQuery;
                                    newField.showEdit = sourceField.showEdit;
                                    newField.checkRepeat = sourceField.checkRepeat;
                                    newField.exportFlag = sourceField.exportFlag;
                                    newField.importFlag = sourceField.importFlag;
                                    newField.listEditFlag = sourceField.listEditFlag;
                                    newField.listEditPlaceholder = sourceField.listEditPlaceholder;
                                    newField.statCondition = sourceField.statCondition;
                                    newField.statData = sourceField.statData;
                                    newField.linkTableId = sourceField.linkTableId;
                                    newField.linkFieldId = sourceField.linkFieldId;
                                    newField.apiId = sourceField.apiId;
                                    newField.createTime = DateTime.Now;
                                    _DatabaseFieldService.Add(newField);
                                }
                            }
                        }
                    }
                }
            }
        }
    }
}