288 lines
11 KiB
Java
288 lines
11 KiB
Java
package com.fjy.spring.controller;
|
|
|
|
import com.fjy.spring.constant.GlobalConstant;
|
|
import com.fjy.spring.domain.Homework;
|
|
import com.fjy.spring.domain.TbFile;
|
|
import com.fjy.spring.domain.TbLog;
|
|
import com.fjy.spring.domain.TbUser;
|
|
import com.fjy.spring.properties.ServerProperties;
|
|
import com.fjy.spring.service.FileService;
|
|
import com.fjy.spring.service.HomeworkService;
|
|
import com.fjy.spring.service.LogService;
|
|
import com.fjy.spring.untils.FormatFileSizeUtil;
|
|
import com.fjy.spring.untils.GetIPAddrUtil;
|
|
import com.fjy.spring.untils.LogUtil;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.apache.commons.io.FileUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
|
|
import javax.annotation.Resource;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.text.SimpleDateFormat;
|
|
import java.util.ArrayList;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import static com.fjy.spring.constant.GlobalConstant.USER_SESSION_KEY;
|
|
|
|
@Controller
|
|
@Slf4j
|
|
public class UpLoadController {
|
|
|
|
/**
|
|
* 服务器配置信息
|
|
*/
|
|
@Autowired
|
|
private ServerProperties serverProperties;
|
|
|
|
/**
|
|
* 文件相关数据库操作
|
|
*/
|
|
@Autowired
|
|
private FileService fileService;
|
|
|
|
@Autowired
|
|
private HomeworkService homeworkService;
|
|
|
|
@Autowired
|
|
private LogService logService;
|
|
|
|
@Resource
|
|
HttpServletRequest httpServletRequest;
|
|
|
|
/**
|
|
* 测试文件上传路径地址
|
|
*
|
|
* @param imageFile
|
|
* @param request
|
|
* @return
|
|
*/
|
|
@RequestMapping("/test")
|
|
@ResponseBody
|
|
public String testURL(@RequestParam("imageFile") MultipartFile imageFile, HttpServletRequest request) {
|
|
String uploadUrl = request.getSession().getServletContext().getRealPath("/") + "upload/";
|
|
return uploadUrl;
|
|
}
|
|
|
|
/**
|
|
* 单文件上传
|
|
*
|
|
* @param imageFile
|
|
* @param request
|
|
* @return
|
|
*/
|
|
@RequestMapping(value = "/oneUpload")
|
|
public String oneUpload(@RequestParam("imageFile") MultipartFile imageFile, HttpServletRequest request) {
|
|
|
|
TbUser user = (TbUser) request.getSession().getAttribute(GlobalConstant.USER_SESSION_KEY);
|
|
//String uploadUrl = request.getSession().getServletContext().getRealPath("/") + "upload/";
|
|
String uploadUrl = serverProperties.getFilePath() + "upload/";
|
|
String filename = imageFile.getOriginalFilename();
|
|
File dir = new File(uploadUrl);
|
|
//判断目录是否存在，否则自动创建
|
|
if (!dir.exists()) {
|
|
dir.mkdirs();
|
|
}
|
|
/**
|
|
* 存储文件信息
|
|
*/
|
|
TbFile file = new TbFile();
|
|
file.setColfilesize(FormatFileSizeUtil.GetFileSize(imageFile.getSize()));
|
|
file.setColfilename(filename);
|
|
file.setColfilepath(uploadUrl + filename);
|
|
file.setColip(request.getRemoteAddr());
|
|
file.setColuserid(user.getColuserid());
|
|
if (fileService.addFile(file)) {
|
|
log.info("记录写入数据库成功");
|
|
|
|
// 记录上传日志
|
|
addVisitLog("上传文件" + file.getColrealname() + " " + file.getColfilename());
|
|
|
|
}
|
|
//System.out.println("记录写入数据库成功");
|
|
else {
|
|
log.error("记录写入数据库失败");
|
|
}
|
|
//System.out.println("记录写入数据库失败");
|
|
|
|
log.info("文件上传到: " + uploadUrl + filename);
|
|
log.info("文件大小: " + FormatFileSizeUtil.GetFileSize(imageFile.getSize()));
|
|
log.info("文件名: " + filename);
|
|
|
|
File targetFile = new File(uploadUrl + filename);
|
|
if (!targetFile.exists()) {
|
|
try {
|
|
targetFile.createNewFile();
|
|
} catch (IOException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
try {
|
|
imageFile.transferTo(targetFile);
|
|
} catch (IllegalStateException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
} catch (IOException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
}
|
|
return "redirect:" + request.getScheme() + "://" + request.getServerName() + ":"
|
|
+ serverProperties.getPortNum() + request.getContextPath() + "/upload/" + filename;
|
|
//return "redirect:http://localhost:8080/cms/upload/" + filename;
|
|
// return "index";
|
|
}
|
|
|
|
/**
|
|
* 多文件上传
|
|
*
|
|
* @param request
|
|
* @return
|
|
*/
|
|
@RequestMapping("/home/moreUpload")
|
|
public void moreUpload(HttpServletRequest request,
|
|
@RequestParam(value = "courseName", required = false) String courseName,
|
|
@RequestParam(value = "folder", required = false) String folder,
|
|
@RequestParam(value = "workid", required = false) Integer workId,
|
|
@RequestParam(value = "rename", required = true) boolean rename) {
|
|
|
|
MultipartHttpServletRequest multipartHttpServletRequest = (MultipartHttpServletRequest) request;
|
|
Map<String, MultipartFile> files = multipartHttpServletRequest.getFileMap();
|
|
|
|
TbUser user = (TbUser) request.getSession().getAttribute(GlobalConstant.USER_SESSION_KEY);
|
|
Date date = new Date();
|
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
|
|
SimpleDateFormat sdf2 = new SimpleDateFormat("yyyy_MM_dd HH_mm_ss");
|
|
String dateNowStr = sdf.format(date);
|
|
String dateNowStr2 = sdf2.format(date);
|
|
String uploadUrl;
|
|
//String uploadUrl = request.getSession().getServletContext().getRealPath("/") + "upload/";
|
|
if (rename) {
|
|
uploadUrl = serverProperties.getFilePath() + "upload/" + courseName + "/" + folder + "/";
|
|
} else {
|
|
uploadUrl = serverProperties.getFilePath() + "upload/";
|
|
}
|
|
|
|
|
|
File dir = new File(uploadUrl);
|
|
if (!dir.exists()) {
|
|
dir.mkdirs();
|
|
}
|
|
|
|
List<String> fileList = new ArrayList<String>();
|
|
|
|
for (MultipartFile file : files.values()) {
|
|
Homework homework = new Homework();
|
|
//管理员上传不需要传workId
|
|
if (workId != null) {
|
|
homework = homeworkService.findById(workId);
|
|
}
|
|
String filePrefix = homework.getFilePrefix();
|
|
String fileSuffix = homework.getFileSuffix();
|
|
String filename = file.getOriginalFilename();
|
|
|
|
//获取文件后缀
|
|
String suffix = "." + filename.substring(filename.lastIndexOf(".") + 1);
|
|
TbFile tbFile = new TbFile();
|
|
String pathname;
|
|
|
|
|
|
//文件重命名
|
|
if (rename && workId != null) {
|
|
pathname = uploadUrl + filePrefix + user.getColstudentno() + user.getColrealname() + fileSuffix + suffix;
|
|
tbFile.setColfilename(filePrefix + user.getColstudentno() + user.getColrealname() + fileSuffix + suffix);
|
|
} else {
|
|
pathname = uploadUrl + filename;
|
|
tbFile.setColfilename(filename);
|
|
}
|
|
|
|
File targetFile = new File(pathname);
|
|
//若文件已存在则自动重命名
|
|
if (targetFile.exists()) {
|
|
String bakpathname;
|
|
if (rename) {
|
|
bakpathname = uploadUrl + "bak/" + user.getColstudentno() + user.getColrealname() + suffix;
|
|
} else {
|
|
bakpathname = uploadUrl + "bak/" + filename;
|
|
}
|
|
log.info("源文件路径:" + pathname);
|
|
TbFile file1 = fileService.findByFilepath(pathname);
|
|
file1.setColfilepath(bakpathname + "." + dateNowStr2 + ".bak");
|
|
file1.setColfilename(file1.getColfilename() + "." + dateNowStr2 + ".bak");
|
|
if (fileService.addFile(file1)) {
|
|
log.info("重命名文件数据库更新成功");
|
|
} else {
|
|
log.error("重命名文件数据库更新失败");
|
|
}
|
|
File mvfile = new File(bakpathname + "." + dateNowStr2 + ".bak");
|
|
try {
|
|
FileUtils.moveFile(targetFile, mvfile);
|
|
log.info("源文件：" + targetFile.getName() + "已重命名为：" + mvfile.getName());
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
log.info("文件上传到: " + uploadUrl + filename);
|
|
log.info("文件大小: " + FormatFileSizeUtil.GetFileSize(file.getSize()));
|
|
log.info("文件名: " + filename);
|
|
|
|
tbFile.setColfilesize(FormatFileSizeUtil.GetFileSize(file.getSize()));
|
|
|
|
tbFile.setColtime(dateNowStr);
|
|
tbFile.setColrealname(filename);
|
|
//文件自动学号+姓名命名
|
|
tbFile.setColfilepath(pathname);
|
|
tbFile.setColip(GetIPAddrUtil.getIpAddr(request));
|
|
tbFile.setColuserid(user.getColuserid());
|
|
tbFile.setCourseName(courseName);
|
|
tbFile.setWorkFolder(folder);
|
|
|
|
if (fileService.addFile(tbFile)) {
|
|
log.info("记录写入数据库成功");
|
|
|
|
// 记录上传日志
|
|
addVisitLog("上传了" + tbFile.getColrealname() + "->" + tbFile.getColfilename());
|
|
} else {
|
|
log.error("记录写入数据库失败");
|
|
}
|
|
|
|
if (!targetFile.exists()) {
|
|
try {
|
|
targetFile.createNewFile();
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
try {
|
|
file.transferTo(targetFile);
|
|
fileList.add(
|
|
request.getScheme() + "://" + request.getServerName() + ":"
|
|
+ serverProperties.getPortNum() + request.getContextPath() + "/upload/"
|
|
+ file.getOriginalFilename());
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 登陆后的访问日志记录
|
|
* @param content
|
|
*/
|
|
private void addVisitLog(String content) {
|
|
TbUser user = (TbUser) httpServletRequest.getSession().getAttribute(USER_SESSION_KEY);
|
|
TbLog log = LogUtil.addLog(user, content, httpServletRequest);
|
|
logService.addLogRec(log);
|
|
}
|
|
}
|