| | |
| | | |
| | | import cn.iocoder.yudao.framework.common.pojo.CommonResult; |
| | | import cn.iocoder.yudao.framework.common.util.object.BeanUtils; |
| | | import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils; |
| | | import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportExcelVO; |
| | | import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportReqVO; |
| | | import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportRespVO; |
| | | import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigRespVO; |
| | | import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigSaveReqVO; |
| | | import cn.iocoder.yudao.module.infra.dal.dataobject.db.DataSourceConfigDO; |
| | |
| | | import io.swagger.v3.oas.annotations.Parameter; |
| | | import io.swagger.v3.oas.annotations.tags.Tag; |
| | | import jakarta.annotation.Resource; |
| | | import jakarta.servlet.http.HttpServletResponse; |
| | | import jakarta.validation.Valid; |
| | | import org.springframework.security.access.prepost.PreAuthorize; |
| | | import org.springframework.validation.annotation.Validated; |
| | | import org.springframework.web.bind.annotation.*; |
| | | |
| | | import java.io.IOException; |
| | | import java.util.Arrays; |
| | | import java.util.List; |
| | | |
| | | import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; |
| | |
| | | @GetMapping("/get") |
| | | @Operation(summary = "获得数据源配置") |
| | | @Parameter(name = "id", description = "编号", required = true, example = "1024") |
| | | @PreAuthorize("@ss.hasPermission('infra:data-source-config:query')") |
| | | public CommonResult<DataSourceConfigRespVO> getDataSourceConfig(@RequestParam("id") Long id) { |
| | | DataSourceConfigDO config = dataSourceConfigService.getDataSourceConfig(id); |
| | | return success(BeanUtils.toBean(config, DataSourceConfigRespVO.class)); |
| | |
| | | |
| | | @GetMapping("/list") |
| | | @Operation(summary = "获得数据源配置列表") |
| | | @PreAuthorize("@ss.hasPermission('infra:data-source-config:query')") |
| | | public CommonResult<List<DataSourceConfigRespVO>> getDataSourceConfigList() { |
| | | List<DataSourceConfigDO> list = dataSourceConfigService.getDataSourceConfigList(); |
| | | return success(BeanUtils.toBean(list, DataSourceConfigRespVO.class)); |
| | | } |
| | | |
| | | @GetMapping("/get-import-template") |
| | | @Operation(summary = "获得导入数据源配置模板") |
| | | public void importTemplate(HttpServletResponse response) throws IOException { |
| | | // 手动创建导出 demo |
| | | List<DataSourceConfigImportExcelVO> list = Arrays.asList( |
| | | DataSourceConfigImportExcelVO.builder().name("示例数据源") |
| | | .url("jdbc:mysql://127.0.0.1:3306/example").username("root").password("").build() |
| | | ); |
| | | // 输出 |
| | | ExcelUtils.write(response, "数据源配置导入模板.xls", "数据源配置列表", DataSourceConfigImportExcelVO.class, list); |
| | | } |
| | | |
| | | @PostMapping("/import") |
| | | @Operation(summary = "导入数据源配置") |
| | | @PreAuthorize("@ss.hasPermission('infra:data-source-config:import')") |
| | | public CommonResult<DataSourceConfigImportRespVO> importExcel(@Valid DataSourceConfigImportReqVO importReqVO) |
| | | throws Exception { |
| | | List<DataSourceConfigImportExcelVO> list = ExcelUtils.read(importReqVO.getFile(), DataSourceConfigImportExcelVO.class); |
| | | return success(dataSourceConfigService.importDataSourceConfigList(list, importReqVO)); |
| | | } |
| | | |
| | | } |