diff --git a/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java b/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java index 2ed62b7..00a9018 100644 --- a/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java +++ b/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java @@ -5,7 +5,6 @@ import cn.bunny.domain.entity.DatabaseInfoMetaData; import cn.bunny.domain.entity.TableMetaData; import cn.bunny.service.TableService; import lombok.RequiredArgsConstructor; -import lombok.SneakyThrows; import org.springframework.stereotype.Service; import java.util.List; @@ -22,7 +21,6 @@ public class TableServiceImpl implements TableService { * * @return 当前连接的数据库信息属性 */ - @SneakyThrows @Override public DatabaseInfoMetaData databaseInfoMetaData() { List databaseTableList = databaseMetadataProvider.getTableMetadataBatch(null); diff --git a/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java b/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java index 541a070..3f1117b 100644 --- a/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java +++ b/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java @@ -10,7 +10,6 @@ import cn.bunny.utils.ResourceFileUtil; import cn.bunny.utils.VmsUtil; import cn.hutool.crypto.digest.MD5; import lombok.RequiredArgsConstructor; -import lombok.SneakyThrows; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; @@ -35,24 +34,37 @@ public class VmsServiceImpl implements VmsService { return codeGeneratorService.generateCode(dto); } - @SneakyThrows + /** + * 获取vms文件路径 + * + * @return vms下的文件路径 + */ @Override public Map> vmsResourcePathList() { - List vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms"); + List vmsRelativeFiles; + Map> listMap; - return vmsRelativeFiles.stream() - .map(vmFile -> { - String[] filepathList = vmFile.split("/"); - String filename = filepathList[filepathList.length - 1].replace(".vm", ""); + try { + vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms"); + listMap = vmsRelativeFiles.stream() + .map(vmFile -> { + String[] filepathList = vmFile.split("/"); + String filename = filepathList[filepathList.length - 1].replace(".vm", ""); - return VmsPathVo.builder() - .id(VmsUtil.generateDivId()) - .name(vmFile) - .label(filename) - .type(filepathList[0]) - .build(); - }) - .collect(Collectors.groupingBy(VmsPathVo::getType)); + return VmsPathVo.builder() + .id(VmsUtil.generateDivId()) + .name(vmFile) + .label(filename) + .type(filepathList[0]) + .build(); + }) + .collect(Collectors.groupingBy(VmsPathVo::getType)); + } catch (Exception e) { + throw new RuntimeException("Get error of VMS path:" + e.getMessage()); + } + + + return listMap; } @Override