lbqms.git - Gitblit

			@@ -14,6 +14,7 @@
			import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
			import lombok.RequiredArgsConstructor;
			import lombok.extern.slf4j.Slf4j;
			import org.anyline.metadata.Column;
			import org.anyline.metadata.Table;
			import org.anyline.proxy.ServiceProxy;
			import org.apache.velocity.Template;
			@@ -21,6 +22,7 @@
			import org.apache.velocity.app.Velocity;
			import org.dromara.common.core.constant.Constants;
			import org.dromara.common.core.exception.ServiceException;
			import org.dromara.common.core.utils.SpringUtils;
			import org.dromara.common.core.utils.StreamUtils;
			import org.dromara.common.core.utils.StringUtils;
			import org.dromara.common.core.utils.file.FileUtils;
			@@ -61,6 +63,8 @@
			private final GenTableMapper baseMapper;
			private final GenTableColumnMapper genTableColumnMapper;
			private final IdentifierGenerator identifierGenerator;

			private static final String[] TABLE_IGNORE = new String[]{"sj_", "act_", "flw_", "gen_"};

			/**
			* 查询业务字段列表
			@@ -119,12 +123,27 @@
			// 获取查询条件
			String tableName = genTable.getTableName();
			String tableComment = genTable.getTableComment();
			// 获取分页参数
			Integer pageNum = pageQuery.getPageNum();
			Integer pageSize = pageQuery.getPageSize();

			LinkedHashMap<String, Table<?>> tablesMap = ServiceProxy.metadata().tables();
			if (CollUtil.isEmpty(tablesMap)) {
			return TableDataInfo.build();
			}
			List<String> tableNames = baseMapper.selectTableNameList(genTable.getDataName());
			String[] tableArrays;
			if (CollUtil.isNotEmpty(tableNames)) {
			tableArrays = tableNames.toArray(new String[0]);
			} else {
			tableArrays = new String[0];
			}
			// 过滤并转换表格数据
			List<GenTable> tables = ServiceProxy.metadata().tables().values().stream()
			List<GenTable> tables = tablesMap.values().stream()
			.filter(x -> !StringUtils.containsAnyIgnoreCase(x.getName(), TABLE_IGNORE))
			.filter(x -> {
			if (CollUtil.isEmpty(tableNames)) {
			return true;
			}
			return !StringUtils.equalsAnyIgnoreCase(x.getName(), tableArrays);
			})
			.filter(x -> {
			boolean nameMatches = true;
			boolean commentMatches = true;
			@@ -148,10 +167,10 @@
			return gen;
			}).toList();

			// 创建分页对象，并设置总记录数
			IPage<GenTable> page = new Page<>(pageNum, pageSize, tables.size());
			// 使用CollUtil进行分页，并设置分页记录
			page.setRecords(CollUtil.page(pageNum - 1, pageSize, tables));
			IPage<GenTable> page = pageQuery.build();
			page.setTotal(tables.size());
			// 手动分页 set数据
			page.setRecords(CollUtil.page((int) page.getCurrent() - 1, (int) page.getSize(), tables));
			return TableDataInfo.build(page);
			}

			@@ -166,9 +185,19 @@
			@Override
			public List<GenTable> selectDbTableListByNames(String[] tableNames, String dataName) {
			Set<String> tableNameSet = new HashSet<>(List.of(tableNames));
			List<Table> tableList = ServiceProxy.metadata().tables().values().stream()
			LinkedHashMap<String, Table<?>> tablesMap = ServiceProxy.metadata().tables();

			if (CollUtil.isEmpty(tablesMap)) {
			return new ArrayList<>();
			}

			List<Table<?>> tableList = tablesMap.values().stream()
			.filter(x -> !StringUtils.containsAnyIgnoreCase(x.getName(), TABLE_IGNORE))
			.filter(x -> tableNameSet.contains(x.getName())).toList();

			if (CollUtil.isEmpty(tableList)) {
			return new ArrayList<>();
			}
			return tableList.stream().map(x -> {
			GenTable gen = new GenTable();
			gen.setDataName(dataName);
			@@ -239,7 +268,7 @@
			int row = baseMapper.insert(table);
			if (row > 0) {
			// 保存列信息
			List<GenTableColumn> genTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName, dataName);
			List<GenTableColumn> genTableColumns = SpringUtils.getAopProxy(this).selectDbTableColumnsByName(tableName, dataName);
			List<GenTableColumn> saveColumns = new ArrayList<>();
			for (GenTableColumn column : genTableColumns) {
			GenUtils.initColumnField(column, table);
			@@ -253,6 +282,32 @@
			} catch (Exception e) {
			throw new ServiceException("导入失败：" + e.getMessage());
			}
			}

			/**
			* 根据表名称查询列信息
			*
			* @param tableName 表名称
			* @param dataName 数据源名称
			* @return 列信息
			*/
			@DS("#dataName")
			@Override
			public List<GenTableColumn> selectDbTableColumnsByName(String tableName, String dataName) {
			LinkedHashMap<String, Column> columns = ServiceProxy.metadata().columns(tableName);
			List<GenTableColumn> tableColumns = new ArrayList<>();
			columns.forEach((columnName, column) -> {
			GenTableColumn tableColumn = new GenTableColumn();
			tableColumn.setIsPk(String.valueOf(column.isPrimaryKey()));
			tableColumn.setColumnName(column.getName());
			tableColumn.setColumnComment(column.getComment());
			tableColumn.setColumnType(column.getTypeName().toLowerCase());
			tableColumn.setSort(column.getPosition());
			tableColumn.setIsRequired(column.isNullable() == 0 ? "1" : "0");
			tableColumn.setIsIncrement(column.isAutoIncrement() == -1 ? "0" : "1");
			tableColumns.add(tableColumn);
			});
			return tableColumns;
			}

			/**
			@@ -350,7 +405,7 @@
			List<GenTableColumn> tableColumns = table.getColumns();
			Map<String, GenTableColumn> tableColumnMap = StreamUtils.toIdentityMap(tableColumns, GenTableColumn::getColumnName);

			List<GenTableColumn> dbTableColumns = genTableColumnMapper.selectDbTableColumnsByName(table.getTableName(), table.getDataName());
			List<GenTableColumn> dbTableColumns = SpringUtils.getAopProxy(this).selectDbTableColumnsByName(table.getTableName(), table.getDataName());
			if (CollUtil.isEmpty(dbTableColumns)) {
			throw new ServiceException("同步数据失败，原表结构不存在");
			}