package com.ruoyi.web.controller.tool; import java.util.ArrayList; import java.util.List; import javax.servlet.http.HttpServletResponse; import com.ruoyi.common.config.RuoYiConfig; import com.ruoyi.common.utils.file.FileUploadUtils; import com.ruoyi.common.utils.file.FileUtils; import com.ruoyi.framework.config.ServerConfig; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PutMapping; import org.springframework.web.bind.annotation.DeleteMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.ruoyi.common.annotation.Log; import com.ruoyi.common.core.controller.BaseController; import com.ruoyi.common.core.domain.AjaxResult; import com.ruoyi.common.enums.BusinessType; import com.ruoyi.tool.domain.TUnifyFile; import com.ruoyi.tool.service.ITUnifyFileService; import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.common.core.page.TableDataInfo; import org.springframework.web.multipart.MultipartFile; /** * 附件Controller * * @author ruoyi * @date 2024-02-23 */ @RestController @RequestMapping("/system/file") public class TUnifyFileController extends BaseController { @Autowired private ITUnifyFileService tUnifyFileService; @Autowired private ServerConfig serverConfig; /** * 查询附件列表 */ @PreAuthorize("@ss.hasPermi('system:file:list')") @GetMapping("/list") public TableDataInfo list(TUnifyFile tUnifyFile) { startPage(); List list = tUnifyFileService.selectTUnifyFileList(tUnifyFile); return getDataTable(list); } /** * 导出附件列表 */ @PreAuthorize("@ss.hasPermi('system:file:export')") @Log(title = "附件", businessType = BusinessType.EXPORT) @PostMapping("/export") public void export(HttpServletResponse response, TUnifyFile tUnifyFile) { List list = tUnifyFileService.selectTUnifyFileList(tUnifyFile); ExcelUtil util = new ExcelUtil(TUnifyFile.class); util.exportExcel(response, list, "附件数据"); } /** * 获取附件详细信息 */ @PreAuthorize("@ss.hasPermi('system:file:query')") @GetMapping(value = "/{id}") public AjaxResult getInfo(@PathVariable("id") String id) { return success(tUnifyFileService.selectTUnifyFileById(id)); } /** * 新增附件 */ @PreAuthorize("@ss.hasPermi('system:file:add')") @Log(title = "附件", businessType = BusinessType.INSERT) @PostMapping public AjaxResult add(@RequestBody TUnifyFile tUnifyFile) { return toAjax(tUnifyFileService.insertTUnifyFile(tUnifyFile)); } /** * 修改附件 */ @PreAuthorize("@ss.hasPermi('system:file:edit')") @Log(title = "附件", businessType = BusinessType.UPDATE) @PutMapping public AjaxResult edit(@RequestBody TUnifyFile tUnifyFile) { return toAjax(tUnifyFileService.updateTUnifyFile(tUnifyFile)); } /** * 删除附件 */ @PreAuthorize("@ss.hasPermi('system:file:remove')") @Log(title = "附件", businessType = BusinessType.DELETE) @DeleteMapping("/{ids}") public AjaxResult remove(@PathVariable String[] ids) { return toAjax(tUnifyFileService.updateTUnifyFileByIds(ids)); } /** * 上传附件 */ @PreAuthorize("@ss.hasPermi('system:file:add')") @Log(title = "附件", businessType = BusinessType.INSERT) @PostMapping("/uploads") public AjaxResult uploadFiles(List files) { try { // 上传文件路径 String filePath = RuoYiConfig.getUploadPath(); for (MultipartFile file : files) { // 上传并返回新文件名称 String fileName = FileUploadUtils.upload(filePath, file); String url = serverConfig.getUrl() + fileName; TUnifyFile tUnifyFile = new TUnifyFile(); tUnifyFile.setUploadPath(url); tUnifyFile.setUploadName(fileName); tUnifyFile.setNewUploadName(FileUtils.getName(fileName)); tUnifyFile.setUploadFormat(file.getOriginalFilename()); tUnifyFileService.insertTUnifyFile(tUnifyFile); } return AjaxResult.success(); } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } }