package com.ruoyi.web.controller.common;
|
|
import com.ruoyi.common.config.RuoYiConfig;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.core.page.TableDataInfo;
|
import com.ruoyi.common.utils.file.FileUtils;
|
import com.ruoyi.manage.domain.SysFileManage;
|
import com.ruoyi.manage.domain.vo.SysFileManageVO;
|
import com.ruoyi.manage.service.SysFileService;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import jakarta.servlet.http.HttpServletRequest;
|
import jakarta.servlet.http.HttpServletResponse;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.core.io.FileSystemResource;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.File;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* <p>
|
* 文件管理表 前端控制器
|
* </p>
|
*
|
* @author sunjiawei
|
* @since 2025-05-21
|
*/
|
@RestController
|
@RequestMapping("/dp/sysFile")
|
@Tag(name = "大屏--文件管理")
|
public class SysFileController {
|
@Autowired
|
SysFileService sysFileService;
|
@Autowired
|
RuoYiConfig ruoYiConfig;
|
|
/**
|
* 查询文件管理列表
|
*/
|
@GetMapping("/listAll")
|
@Operation(summary = "查询文件管理列表")
|
public AjaxResult listAll()
|
{
|
return AjaxResult.success(sysFileService.listAll());
|
}
|
|
/**
|
* 单个上传文件
|
*/
|
@PostMapping("/uploadFile")
|
@Operation(summary = "单个上传文件")
|
public AjaxResult uploadFile(MultipartFile file,String name,String type,
|
@RequestParam(value = "remark",required = false) String remark) {
|
SysFileManage sysFileManage = new SysFileManage();
|
sysFileManage.setName(name);
|
sysFileManage.setType(type);
|
sysFileManage.setRemark(remark);
|
try {
|
int res = sysFileService.saveFile(file,sysFileManage);
|
return AjaxResult.success(res);
|
}
|
catch (Exception e)
|
{
|
return AjaxResult.error(e.getMessage());
|
}
|
}
|
|
/**
|
* 批量删除文件
|
*/
|
@PostMapping("/deleteBatch")
|
@Operation(summary = "批量删除文件")
|
public AjaxResult deleteBatch(@RequestBody List<String> ids)
|
{
|
List<String> res = sysFileService.deleteBatch(ids);
|
if(res.size()==0){
|
return AjaxResult.success("删除成功");
|
}
|
return AjaxResult.success(res);
|
}
|
|
@PostMapping("/download")
|
@Operation(summary = "下载文件")
|
public ResponseEntity<FileSystemResource> resourceDownload(String id){
|
try {
|
SysFileManage sysFileManage = sysFileService.getById(id);
|
String filePath = sysFileManage.getFilePath();
|
String name = sysFileManage.getName();
|
String realFilePath = filePath.replaceFirst("/profile",RuoYiConfig.getProfile());
|
String ext = sysFileManage.getExt();
|
if(ext!=null && !ext.isEmpty()){
|
if(ext.equals("json")){
|
realFilePath = filePath.replaceFirst("/profile",ruoYiConfig.getJson());
|
}
|
}
|
String realFileName = name + realFilePath.substring(realFilePath.indexOf("_") + 1);
|
|
File file = new File(realFilePath);
|
if(!file.exists()){
|
return ResponseEntity.notFound().build();
|
}
|
FileSystemResource resource = new FileSystemResource(file);
|
|
return ResponseEntity.ok()
|
.header("Content-Disposition","attachment;filename=" + realFileName)
|
.body(resource);
|
|
}catch (Exception e){
|
e.printStackTrace();
|
return ResponseEntity.status(500).build();
|
}
|
}
|
|
@GetMapping("/getCount")
|
@Operation(summary = "查询总条数")
|
public AjaxResult getCount(){
|
return AjaxResult.success(sysFileService.queryCount());
|
}
|
|
|
/**
|
* 分页查询文件管理列表
|
*/
|
@PostMapping("/getPageList")
|
@Operation(summary = "分页查询文件管理列表")
|
public TableDataInfo getPageList(@Validated @RequestBody SysFileManageVO vo) {
|
List<SysFileManage> list = sysFileService.queryData(vo);
|
return paginate(list, vo.getPageNum(), vo.getPageSize());
|
}
|
|
//对列表进行分页
|
private TableDataInfo paginate(List list, int pageNum, int pageSize) {
|
if (list == null || list.isEmpty()) {
|
return new TableDataInfo(new ArrayList<>(), 0);
|
}
|
int total = list.size();
|
int formIndex = (pageNum - 1) * pageSize;
|
int toIndex = Math.min(formIndex + pageSize, total);
|
if (formIndex >= total) {
|
return new TableDataInfo(new ArrayList<>(), total);
|
}
|
List subList = list.subList(formIndex, toIndex);
|
return new TableDataInfo(subList, total);
|
}
|
|
}
|
