package org.jeecg.modules.desk.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import io.swagger.annotations.Api;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.shiro.SecurityUtils;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.aspect.annotation.AutoLog;
|
import org.jeecg.common.constant.CommonConstant;
|
import org.jeecg.common.system.base.controller.JeecgController;
|
import org.jeecg.common.system.query.QueryGenerator;
|
import org.jeecg.common.system.vo.LoginUser;
|
import org.jeecg.common.util.CommonUtils;
|
import org.jeecg.common.util.DateUtils;
|
import org.jeecg.common.util.oConvertUtils;
|
import org.jeecg.modules.desk.entity.DeskFile;
|
import org.jeecg.modules.desk.service.IDeskFileService;
|
import org.jeecg.modules.desk.vo.DeskFileTree;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.AntPathMatcher;
|
import org.springframework.util.FileCopyUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
import org.springframework.web.servlet.HandlerMapping;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
|
|
@Slf4j
|
@Api(tags = "我的桌面")
|
@RestController
|
@RequestMapping("/desk/file")
|
public class DeskFileController extends JeecgController<DeskFile, IDeskFileService> {
|
@Autowired
|
private IDeskFileService deskFileService;
|
@Value(value = "${jeecg.path.upload}")
|
private String uploadpath;
|
|
/**
|
* 本地：local minio：minio 阿里：alioss
|
*/
|
@Value(value = "${jeecg.uploadType}")
|
private String uploadType;
|
|
private static int FOLDER_TYPE_2 = 2; //文件类型 1-文件 2-文件夹
|
private static String DESKTOP_FOLDER_TAG = "desk";//桌面文件标记
|
private static String DESKTOP_FOLDER_ALL_TAG = "all";//通用文件夹标记，所有人可查询此文件夹
|
|
//加载桌面文件夹
|
@RequestMapping(value = "/desktop", method = RequestMethod.GET)
|
public Result<IPage<DeskFile>> desktop(DeskFile deskFile, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
|
@RequestParam(name = "pageSize", defaultValue = "1000") Integer pageSize, HttpServletRequest req) {
|
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
String username = sysUser.getUsername();
|
Result<IPage<DeskFile>> result = new Result<IPage<DeskFile>>();
|
|
QueryWrapper<DeskFile> queryWrapper = QueryGenerator.initQueryWrapper(deskFile, req.getParameterMap());
|
//查询类型为2的数据（1-文件 2-文件夹）
|
queryWrapper.lambda().eq(DeskFile::getType, FOLDER_TYPE_2);
|
//查询tag为desk的数据（desk-桌面文件夹标记）
|
queryWrapper.lambda().and(i -> i.eq(DeskFile::getTag, DESKTOP_FOLDER_TAG));
|
|
//查询登录用户的数据
|
queryWrapper.lambda().and(i -> i.eq(DeskFile::getCreateBy, username).or().eq(DeskFile::getCreateBy,DESKTOP_FOLDER_ALL_TAG) );
|
//只查询未删除数据
|
queryWrapper.lambda().and(i -> i.isNull(true,DeskFile::getDel).or().eq(DeskFile::getDel,0));
|
|
Page<DeskFile> page = new Page<DeskFile>(pageNo, pageSize);
|
IPage<DeskFile> pageList = deskFileService.page(page, queryWrapper);
|
result.setSuccess(true);
|
result.setResult(pageList);
|
return result;
|
}
|
|
|
//通用查询
|
@RequestMapping(value = "/list", method = RequestMethod.GET)
|
public Result<IPage<DeskFile>> queryPageList(DeskFile deskFile, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
|
@RequestParam(name = "pageSize", defaultValue = "100") Integer pageSize, HttpServletRequest req) {
|
Result<IPage<DeskFile>> result = new Result<IPage<DeskFile>>();
|
if (deskFile.getPid().isEmpty()) {
|
return result.error500("上级目录不能为空!");
|
}
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
String username = sysUser.getUsername();
|
|
QueryWrapper<DeskFile> queryWrapper = QueryGenerator.initQueryWrapper(deskFile, req.getParameterMap());
|
//查询登录用户的数据
|
queryWrapper.lambda().eq(DeskFile::getCreateBy, username);
|
//只查询未删除数据
|
queryWrapper.lambda().isNull(true, DeskFile::getDel).or().eq(DeskFile::getDel, 0);
|
Page<DeskFile> page = new Page<DeskFile>(pageNo, pageSize);
|
IPage<DeskFile> pageList = deskFileService.page(page, queryWrapper);
|
//递归查询pid
|
String pid = deskFile.getPid();
|
List<Map<String, Object>> res = new ArrayList<>();
|
getParent(res, pid);
|
result.setSuccess(true);
|
result.setResult(pageList);
|
result.setResultExt(res);
|
return result;
|
}
|
|
//查询最近100条数据
|
@RequestMapping(value = "/recent", method = RequestMethod.GET)
|
public Result<IPage<DeskFile>> recent(DeskFile deskFile, @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
|
@RequestParam(name = "pageSize", defaultValue = "100") Integer pageSize, HttpServletRequest req) {
|
Result<IPage<DeskFile>> result = new Result<IPage<DeskFile>>();
|
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
String username = sysUser.getUsername();
|
|
QueryWrapper<DeskFile> queryWrapper = QueryGenerator.initQueryWrapper(deskFile, req.getParameterMap());
|
//查询登录用户的数据
|
queryWrapper.lambda().eq(DeskFile::getCreateBy, username);
|
queryWrapper.lambda().eq(DeskFile::getType, 1);
|
//只查询未删除数据
|
queryWrapper.lambda().isNull(true, DeskFile::getDel).or().eq(DeskFile::getDel, 0);
|
Page<DeskFile> page = new Page<DeskFile>(pageNo, pageSize);
|
IPage<DeskFile> pageList = deskFileService.page(page, queryWrapper);
|
result.setSuccess(true);
|
result.setResult(pageList);
|
|
return result;
|
}
|
|
private void getParent(List<Map<String, Object>> res, String pid) {
|
Map<String, Object> map = new HashMap<>();
|
DeskFile byId = deskFileService.getById(pid);
|
if (byId != null) {
|
map.put("id", byId.getId());
|
map.put("name", byId.getName());
|
res.add(map);
|
getParent(res, byId.getPid());
|
}
|
}
|
|
@RequestMapping(value = "/add", method = RequestMethod.POST)
|
@AutoLog(value = "添加文件")
|
@Transactional
|
public Result<DeskFile> add(@RequestBody DeskFile deskFile, HttpServletRequest request) {
|
Result<DeskFile> result = new Result<DeskFile>();
|
|
|
// 添加先查询是否有同名文件 TODO
|
QueryWrapper<DeskFile> queryWrapper = new QueryWrapper<>();
|
if (deskFile.getPid() != null) {
|
queryWrapper.lambda().eq(DeskFile::getPid, deskFile.getPid());
|
}
|
//添加桌面文件夹
|
if (deskFile.getTag() != null) {
|
queryWrapper.lambda().eq(DeskFile::getTag, deskFile.getTag());
|
}
|
queryWrapper.lambda().like(DeskFile::getName, deskFile.getName());
|
List<DeskFile> list = deskFileService.list(queryWrapper);
|
if (list != null && list.size() > 0) {
|
// result.error500("文件名已存在");
|
// return result;
|
deskFile.setName(deskFile.getName() + "(" + list.size() + ")");
|
}
|
|
//特殊处理批量文件添加(添加文件)
|
if (deskFile.getType() == 1) {
|
String filelist = deskFile.getFileList();
|
if (filelist != null && !StringUtils.isEmpty(filelist)) {
|
String[] split = filelist.split(",");
|
for (int i = 0; i < split.length; i++) {
|
String filePath = split[i];
|
DeskFile desk = new DeskFile();
|
desk.setFileName(filePath.substring(filePath.lastIndexOf("/") + 1, filePath.length()));
|
desk.setFilePath(filePath);
|
desk.setPid(deskFile.getPid());
|
desk.setType(deskFile.getType());
|
desk.setFileType(filePath.substring(filePath.lastIndexOf(".") + 1));
|
//截取名称
|
String name = filePath.substring(filePath.lastIndexOf("/") + 1, filePath.indexOf("_"));
|
QueryWrapper<DeskFile> wrapper = new QueryWrapper<>();
|
if (deskFile.getPid() != null) {
|
wrapper.lambda().eq(DeskFile::getPid, desk.getPid());
|
}
|
wrapper.lambda().like(DeskFile::getName, name);
|
List<DeskFile> res = deskFileService.list(wrapper);
|
if (res != null && res.size() > 0) {
|
//文件名重复
|
desk.setName(name + "(" + res.size() + ")");
|
} else {
|
//文件名不重复
|
desk.setName(name);
|
}
|
desk.setUpdateTime(new Date());
|
deskFileService.save(desk);
|
}
|
}
|
//(添加文件夹)
|
} else if (deskFile.getType() == 2) {
|
deskFile.setFileType("文件夹");
|
deskFile.setUpdateTime(new Date());
|
deskFileService.save(deskFile);
|
}
|
|
result.success("添加成功！");
|
result.setResult(deskFile);
|
return result;
|
}
|
|
|
@RequestMapping(value = "/edit", method = RequestMethod.PUT)
|
public Result<DeskFile> edit(@RequestBody DeskFile deskFile) {
|
Result<DeskFile> result = new Result<DeskFile>();
|
|
//先查询是否有同名文件 TODO
|
QueryWrapper<DeskFile> queryWrapper = new QueryWrapper<>();
|
if (deskFile.getPid() != null) {
|
queryWrapper.lambda().eq(DeskFile::getPid, deskFile.getPid());
|
}
|
queryWrapper.lambda().eq(DeskFile::getName, deskFile.getName());
|
queryWrapper.lambda().ne(DeskFile::getId, deskFile.getId());
|
List<DeskFile> list = deskFileService.list(queryWrapper);
|
if (list != null && list.size() > 0) {
|
result.error500("文件名已存在");
|
return result;
|
}
|
deskFileService.updateById(deskFile);
|
result.success("修改成功！");
|
result.setResult(deskFile);
|
return result;
|
}
|
|
|
@RequestMapping(value = "/queryTreeList", method = RequestMethod.GET)
|
public Result<List<DeskFileTree>> queryTreeList(DeskFile deskFile) {
|
|
Result<List<DeskFileTree>> result = new Result<List<DeskFileTree>>();
|
|
QueryWrapper<DeskFile> queryWrapper = new QueryWrapper<>();
|
queryWrapper.lambda().eq(DeskFile::getType, FOLDER_TYPE_2);
|
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
String username = sysUser.getUsername();
|
|
//查询登录用户的数据
|
queryWrapper.lambda().eq(DeskFile::getCreateBy, username);
|
|
List<DeskFile> list = deskFileService.list(queryWrapper);
|
List<DeskFileTree> trees = new ArrayList<>();
|
getTreeList(trees, list, null);
|
//传入用户名为最上级目录
|
getTreePath(trees, null, username);
|
result.setSuccess(true);
|
result.setResult(trees);
|
return result;
|
}
|
|
|
private void getTreeList(List<DeskFileTree> treeList, List<DeskFile> metaList, DeskFileTree temp) {
|
for (DeskFile desk : metaList) {
|
String tempPid = desk.getPid();
|
DeskFileTree tree = new DeskFileTree(desk);
|
if (temp == null && oConvertUtils.isEmpty(tempPid)) {
|
treeList.add(tree);
|
getTreeList(treeList, metaList, tree);
|
} else if (temp != null && tempPid != null && tempPid.equals(temp.getId())) {
|
temp.getChildren().add(tree);
|
getTreeList(treeList, metaList, tree);
|
}
|
|
}
|
}
|
|
//获取路径
|
public void getTreePath(List<DeskFileTree> treeList, DeskFileTree temp, String userPath) {
|
for (DeskFileTree desk : treeList) {
|
String path = desk.getPath();
|
String pid = desk.getPid();
|
if (temp != null && temp.getPid() != null && pid.equals(temp.getId())) {
|
path = temp.getPath() + '/' + path;
|
} else {
|
path = desk.getPath();
|
}
|
//加入用户名为主路径
|
desk.setPath(userPath + "/" + path);
|
if (desk.getChildren().size() < 1) {
|
} else {
|
getTreePath(desk.getChildren(), desk, userPath);
|
}
|
}
|
}
|
|
@AutoLog(value = "删除-虚拟桌面文件")
|
@RequestMapping(value = "/delete", method = RequestMethod.DELETE)
|
@Transactional
|
public Result<?> delete(@RequestParam(name = "id", required = true) String id) {
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
String username = sysUser.getUsername();
|
|
DeskFile byId = deskFileService.getById(id);
|
if(byId.getName().equals("最近上传") || byId.getName().equals(username)){
|
return Result.error("此文件夹不可删除");
|
}
|
deskFileService.removeById(byId);
|
//递归删除文件夹下文件
|
delSub(byId.getId());
|
|
|
return Result.ok("删除成功");
|
}
|
|
public void delSub(String pid) {
|
QueryWrapper<DeskFile> queryWrapper = new QueryWrapper<>();
|
queryWrapper.lambda().eq(DeskFile::getPid, pid);
|
List<DeskFile> list = deskFileService.list(queryWrapper);
|
if (list.size() > 0) {
|
deskFileService.remove(queryWrapper);
|
for (int i = 0; i < list.size(); i++) {
|
delSub(list.get(i).getId());
|
}
|
}
|
|
}
|
|
/****上传部分start****/
|
|
//由于需要直接上传功能，需要新增一个自定义上传接口
|
@PostMapping(value = "/upload")
|
public Result<?> upload(HttpServletRequest request, HttpServletResponse response) {
|
Result<?> result = new Result<>();
|
String savePath = "";
|
String bizPath = request.getParameter("biz");
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
MultipartFile file = multipartRequest.getFile("file");// 获取上传文件对象
|
//update-begin-author:lvdandan date:20200928 for:修改JEditor编辑器本地上传
|
savePath = this.uploadLocal(file, bizPath);
|
//update-begin-author:lvdandan date:20200928 for:修改JEditor编辑器本地上传
|
|
if (oConvertUtils.isNotEmpty(savePath)) {
|
result.setMessage(savePath);
|
result.setSuccess(true);
|
} else {
|
result.setMessage("上传失败！");
|
result.setSuccess(false);
|
}
|
return result;
|
}
|
|
/**
|
* 本地文件上传
|
*
|
* @param mf 文件
|
* @param bizPath 自定义路径
|
* @return
|
*/
|
private String uploadLocal(MultipartFile mf, String bizPath) {
|
try {
|
String ctxPath = uploadpath;
|
String fileName = null;
|
File file = new File(ctxPath + File.separator + bizPath + File.separator);
|
if (!file.exists()) {
|
file.mkdirs();// 创建文件根目录
|
}
|
String orgName = mf.getOriginalFilename();// 获取文件名
|
orgName = CommonUtils.getFileName(orgName);
|
if (orgName.indexOf(".") != -1) {
|
fileName = orgName.substring(0, orgName.lastIndexOf(".")) + "_" + DateUtils.date2Str(new SimpleDateFormat("yyyyMMddHHmmss")) + orgName.substring(orgName.lastIndexOf("."));
|
} else {
|
fileName = orgName + "_" + System.currentTimeMillis();
|
}
|
String savePath = file.getPath() + File.separator + fileName;
|
File savefile = new File(savePath);
|
FileCopyUtils.copy(mf.getBytes(), savefile);
|
String dbpath = null;
|
if (oConvertUtils.isNotEmpty(bizPath)) {
|
dbpath = bizPath + File.separator + fileName;
|
} else {
|
dbpath = fileName;
|
}
|
if (dbpath.contains("\\")) {
|
dbpath = dbpath.replace("\\", "/");
|
}
|
return dbpath;
|
} catch (IOException e) {
|
log.error(e.getMessage(), e);
|
}
|
return "";
|
}
|
|
/****上传部分end****/
|
|
|
@GetMapping(value = "/static/**")
|
public void view(HttpServletRequest request, HttpServletResponse response) {
|
// ISO-8859-1 ==> UTF-8 进行编码转换
|
String imgPath = extractPathFromPattern(request);
|
if(oConvertUtils.isEmpty(imgPath) || imgPath=="null"){
|
return;
|
}
|
// 其余处理略
|
InputStream inputStream = null;
|
OutputStream outputStream = null;
|
try {
|
imgPath = imgPath.replace("..", "").replace("../","");
|
if (imgPath.endsWith(",")) {
|
imgPath = imgPath.substring(0, imgPath.length() - 1);
|
}
|
String filePath = uploadpath + File.separator + imgPath;
|
File file = new File(filePath);
|
if(!file.exists()){
|
response.setStatus(404);
|
throw new RuntimeException("文件["+imgPath+"]不存在..");
|
}
|
response.setContentType("application/force-download");// 设置强制下载不打开
|
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
|
inputStream = new BufferedInputStream(new FileInputStream(filePath));
|
outputStream = response.getOutputStream();
|
byte[] buf = new byte[1024];
|
int len;
|
while ((len = inputStream.read(buf)) > 0) {
|
outputStream.write(buf, 0, len);
|
}
|
response.flushBuffer();
|
} catch (IOException e) {
|
log.error("预览文件失败" + e.getMessage());
|
response.setStatus(404);
|
e.printStackTrace();
|
} finally {
|
if (inputStream != null) {
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
log.error(e.getMessage(), e);
|
}
|
}
|
if (outputStream != null) {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
log.error(e.getMessage(), e);
|
}
|
}
|
}
|
|
}
|
/**
|
* 把指定URL后的字符串全部截断当成参数
|
* 这么做是为了防止URL中包含中文或者特殊字符（/等）时，匹配不了的问题
|
* @param request
|
* @return
|
*/
|
private static String extractPathFromPattern(final HttpServletRequest request) {
|
String path = (String) request.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE);
|
String bestMatchPattern = (String) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE);
|
return new AntPathMatcher().extractPathWithinPattern(bestMatchPattern, path);
|
}
|
|
}
|
