源码获取：博客首页 "资源" 里下载！
很多同学都有自己的网盘，方便存储一些java学习教程。该毕业设计实现了一个简易的网盘，包含文件上传和文件分享等功能。
后端技术采用了spring，spring mvc，JPA，前端采用了thymeleaf模板语言。整体业务功能相对不是太复杂，亮点是文件分享功能。
分享操作业务：
@Controller
public class ShareController {
private Logger logger = LoggerFactory.getLogger(this.getClass());
@Autowired
private LinkSecretServiceImpl linkSecretService;
@Autowired
private FileServiceImpl fileService;
private SaveServiceImpl saveService;
/**
* 安卓上查看分享记录的接口
*
* @return
*/
@RequestMapping(value = "/shareRecord", produces = "application/json; charset=utf-8")
@ResponseBody
public List<ShareMessage> shareRecord(HttpServletRequest request) {
User user = (User) request.getSession().getAttribute("user");
String userName = user.getUserName();
List<ShareMessage> shareMessageList = new ArrayList<>();
List<LinkSecret> linkSecretList = linkSecretService.findLinkSecretsByUserName(userName);
for (int i = 0; i < linkSecretList.size(); i++) {
LinkSecret linkSecret = linkSecretList.get(i);
ShareMessage shareMessage = new ShareMessage();
shareMessage.setDownloadNum(linkSecret.getDownloadNum());
shareMessage.setExpireDate(linkSecret.getExpireDate());
shareMessage.setFileName(linkSecret.getLocalLink().substring(linkSecret.getLocalLink().lastIndexOf("/")));
shareMessage.setDownloadName(null);
shareMessageList.add(shareMessage);
}
return shareMessageList;
}
//保存到网盘-----link是加密的链接 downloadLink解密后的链接/data/share/zc2/Fuck.java, path：保存路径--是用户名后面的路径
@RequestMapping(value = "/shareToMyPan")
@ResponseBody
public ResponseMsg shareToMyPan(HttpServletRequest request, String path, String link) {
String downloadLink = "";
if (link.contains("/data/share")) {
downloadLink = link;
} else {
downloadLink = fileService.fileShareCodeDecode(link);
}
logger.warn("downloadLink:" + downloadLink);
boolean b = false;
ResponseMsg responseMsg = new ResponseMsg();
User user = (User) request.getSession().getAttribute("user");
if (user == null) {
responseMsg.setSuccess(false);
responseMsg.setMsg("未登录");
} else {
String userName = user.getUserName();
if (path == null) {
path = "/";
}
logger.warn(userName + " " + downloadLink + " " + path);
b = fileService.copyFileToMyPan(userName, downloadLink, path);
responseMsg.setSuccess(b);
if (b == false) {
responseMsg.setMsg("保存失败");
} else {
responseMsg.setMsg("保存成功");
}
}
return responseMsg;
}
/**
* 下载客户端的apk
* filename:下载apk的名字
* downloadPath:下载的文件夹，放在/root/pan/share目录中
*/
@RequestMapping(value = "/downloadApk", produces = {"application/json; charset=UTF-8"})
@ResponseBody
public ResponseMsg shareToMyPan(HttpServletRequest request, HttpServletResponse response, String filename,
String downloadPath) throws FileNotFoundException, UnsupportedEncodingException {
// 读到流中
// String filePath="F:/"+downloadPath+"/"+filename;//window上测试的路径
ResponseMsg responseMsg = new ResponseMsg();
String filePath = fileRootPath + downloadPath + "/" + filename;
InputStream inStream = new FileInputStream(filePath);
if (!new File(filePath).exists()) {
responseMsg.setMsg("找不到文件");
}
// 设置输出的格式
String agent = request.getHeader("user-agent");
String fileName4 = "";
if (agent.contains("Firefox")) {
fileName4 = new String(filename.getBytes(), "iso-8859-1");
} else {
//空格，（，），；，@，#，&，逗号在谷歌浏览器中出现乱码
fileName4 = URLEncoder.encode(filename, "UTF-8").replaceAll("\\+", "%20").
replaceAll("%28", "\\(").
replaceAll("%29", "\\)").
replaceAll("%3B", ";").
replaceAll("%40", "@").
replaceAll("%23", "\\#").
replaceAll("%26", "\\&").
replaceAll("%2C", "\\,");
}
response.addHeader("Content-Disposition", "attachment; filename=\"" + fileName4 + "\"");
// 循环取出流中的数据
byte[] b = new byte[100];
int len;
try {
while ((len = inStream.read(b)) > 0) {
response.getOutputStream().write(b, 0, len);
}
inStream.close();
responseMsg.setSuccess(true);
} catch (IOException e) {
logger.error("Exception:", e);
responseMsg.setSuccess(false);
}
return responseMsg;
}
}
登录业务控制器：
/**
* 登录
* 管理员admin 密码123 权限0(最高)
*
*/
@Controller
public class LoginController {
private Logger logger = LoggerFactory.getLogger(this.getClass());
@Autowired
private IUserService userService;
@Autowired
private IVerifyCodeService iVerifyCodeService;
// 登录
@RequestMapping(value = "/login", method = RequestMethod.POST)
@ResponseBody
public Map<String, Object> login(HttpServletRequest request, HttpServletResponse response) {
Map<String, Object> map = new HashMap<>();
String userName = request.getParameter("userName");
String password = request.getParameter("password");
String encryptedPwd = "";
try {
//加密的用户密码
encryptedPwd = Md5SaltTool.getEncryptedPwd(password);
logger.warn("encryptedPwd:" + encryptedPwd);
} catch (NoSuchAlgorithmException e) {
logger.error("Exception:", e);
} catch (UnsupportedEncodingException e) {
logger.error("Exception:", e);
}
User dataBaseUser = userService.queryUserByUsername(userName);
if (dataBaseUser != null && encryptedPwd.equals(dataBaseUser.getPassWord())) {
User user = new User(userName, encryptedPwd, dataBaseUser.getLevelType(), dataBaseUser.getEmail(),
dataBaseUser.getPhone(), dataBaseUser.getAlias());
request.getSession().setAttribute("user", user);
logger.info("用户登录成功！");
map.put("result", "1");
map.put("userName", userName);
} else if (dataBaseUser != null && !encryptedPwd.equals(dataBaseUser.getPassWord())) {
logger.info("密码错误！");
map.put("result", "2");
} else {
logger.info("用户不存在！");
map.put("result", "0");
}
return map;
}
// 退出登录
@RequestMapping(value = "/quit", method = RequestMethod.GET)
public String loginOut(HttpServletRequest request) {
if (!SystemUtil.isWindows()) {
// 非windows环境下要删除用户文件
FileUtil.deleteDir(fileRootPath + WebUtil.getUserNameByRequest(request));
}
// 清除session
request.getSession().invalidate();
logger.info("退出登录成功！");
return "login";
}
// 注册
@RequestMapping(value = "/signin", method = RequestMethod.POST)
@ResponseBody
public Map<String, Object> signin(@RequestParam(required = false) String alias,
@RequestParam(required = true) String userName, @RequestParam(required = true) String password,
@RequestParam(required = false) String regcode, @RequestParam(required = false) String email,
@RequestParam(required = false) String phone, HttpServletRequest request, HttpServletResponse response) {
Map<String, Object> map = new HashMap<>();
String encryptedPwd = "";
try {
//加密的用户密码
encryptedPwd = Md5SaltTool.getEncryptedPwd(password);
} catch (NoSuchAlgorithmException e) {
logger.error("NoSuchAlgorithmException:", e);
} catch (UnsupportedEncodingException e) {
logger.error("UnsupportedEncodingException:", e);
}
User dataBaseUser = userService.queryUserByUsername(encryptedPwd);
//regcode的重新写入
if (!iVerifyCodeService.isValid(regcode)) {
logger.info("注册失败，激活码失效或不正确！");
map.put("result", "2");
return map;
} else {
if (dataBaseUser == null) {
User user = new User(userName, encryptedPwd, "0", email, phone, alias);
userService.add(user);
logger.info("账号注册成功！");
map.put("result", "1");
} else {
logger.info("用户已经存在，请登录或换一个用户名！");
map.put("result", "0");
}
return map;
}
}
@RequestMapping(value = "/username", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg fileRename(HttpServletRequest request) {
ResponseMsg j = new ResponseMsg();
// 获取用户名
User user = (User) request.getSession().getAttribute("user");
String userName = user.getUserName();
if (userName == null) {
userName = "null";
}
j.setMsg(userName);
j.setSuccess(true);
return j;
}
@RequestMapping(value = "/getUserByUserName")
@ResponseBody
public User getUserByUserName(HttpServletRequest request) {
User user = (User) request.getSession().getAttribute("user");
return user;
}
//更新当前用户信息
@RequestMapping(value = "/updateUserByUserName")
@ResponseBody
public ResponseMsg updateUserByUserName(@RequestParam(required = false) String username,
@RequestParam(required = false) String alias, @RequestParam(required = false) String password,
@RequestParam(required = false) String email, @RequestParam(required = false) String phone,
HttpServletRequest request) {
ResponseMsg j = new ResponseMsg();
User user = (User) request.getSession().getAttribute("user");
if (username != null) {
user.setUserName(username);
}
if (alias != null) {
user.setAlias(alias);
}
if (password != null) {
try {
String encryptedPwd = Md5SaltTool.getEncryptedPwd(password);
user.setPassWord(encryptedPwd);
} catch (NoSuchAlgorithmException e) {
logger.error("NoSuchAlgorithmException:", e);
} catch (UnsupportedEncodingException e) {
logger.error("UnsupportedEncodingException:", e);
}
}
if (email != null) {
user.setEmail(email);
}
if (phone != null) {
user.setPhone(phone);
}
userService.update(user);
j.setMsg("更新用户信息成功！");
j.setSuccess(true);
return j;
}
//检查是否是登录状态
@RequestMapping(value = "/islogin")
@ResponseBody
public ResponseMsg isLogin(@RequestParam(required = false) String savePath, HttpServletRequest request) {
if (savePath == null) {
savePath = "/";
}
ResponseMsg j = new ResponseMsg();
// 获取用户名
User user = (User) request.getSession().getAttribute("user");
if (user == null) {
//未登录，跳转到登录界面，登录之后默认保存到网盘连接的地址<a href="wut://pan">链接到app</a>
j.setMsg("未登录");
j.setSuccess(false);
} else {
j.setMsg("已登录");
j.setSuccess(true);
}
return j;
}
}
文件管理业务：
/**
* 文件管理
*
*
*/
@Controller
public class FileController {
public static String fileRootPath;
public static String tempPath; // 分块文件临时存储地址
//MD5文件的大小
public static int size;
private static int secretLen;
// 秘钥
private static String key;
private Logger logger = LoggerFactory.getLogger(this.getClass());
@Autowired
private IFileService fileService;
@Autowired
private LinkSecretServiceImpl linkSecretService;
@Autowired
private SaveServiceImpl saveService;
@Value("${key}")
public void setKey(String key) {
FileController.key = key;
}
@Value("${tempPath}")
public void setTempPath(String tempPath) {
FileController.tempPath = tempPath;
}
@Value("${fileRootPath}")
public void setFileRootPath(String fileRootPath) {
FileController.fileRootPath = fileRootPath;
}
@Value("${secretLen}")
public void setSecretLen(int secretLen) {
FileController.secretLen = secretLen;
}
@Value("${size}")
public void setSize(int size) {
FileController.size = size;
}
// 文件上传
@RequestMapping(value = "/upload", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg upload(@RequestParam MultipartFile file, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
ResponseMsg j = new ResponseMsg();
if (file.isEmpty()) {
j.setMsg("请选择要上传的文件！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// 上传文件
boolean b = fileService.upload(file, userName, path);
// 反馈用户信息
if (b) {
j.setSuccess(true);
j.setMsg("上传成功！");
} else {
j.setMsg("上传失败！");
}
return j;
}
// 文件下载
@RequestMapping(value = "/download", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsgAdd download(@RequestParam String fileName, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
ResponseMsgAdd j = new ResponseMsgAdd();
if (fileName.isEmpty()) {
j.setMsg("文件名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// String userName ="zc";
// 下载文件，获取下载路径,这个是 个映射的路径
String link = fileService.download(fileName, userName, path);
try {
//这里校验要填真实的路经
String newLink = link.replace("/data/", fileRootPath);
String[] md5Array = FileSplit.splitBySizeSubSection(newLink, size,
fileRootPath + "/tempMd5/" + userName + "/");
j.setObj(md5Array);
} catch (Exception e) {
logger.error("Exception:", e);
j.setObj("");
}
if (!link.isEmpty()) {
j.setSuccess(true);
j.setMsg(link);
} else {
j.setMsg("");
logger.warn("下载失败");
}
return j;
}
// 搜索接口
@RequestMapping(value = "/search", produces = "application/json; charset=utf-8")
@ResponseBody
public List<FileMsg> search(@RequestParam String key, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
List<FileMsg> fileMsgList = fileService.search(key, userName, path);
// 判断文件转码情况
for (FileMsg fileMsg : fileMsgList) {
// 跳过文件夹
if (fileMsg.getSize().equals("Directory")) {
continue;
}
// 正常文件
int suffixidx = (int) StringUtil.getfilesuffix(fileMsg.getName(), true);
String suffix = fileMsg.getName().substring(suffixidx);
if (suffix.equals("mkv") || suffix.equals("rmvb") || suffix.equals("avi") || suffix.equals("wmv")
|| suffix.equals("3gp") || suffix.equals("rm")) {
// 取非文件夹的所有文件名
List<String> namelist = fileMsgList.stream()
.filter(f -> !f.getSize().equals("Directory"))
.map(FileMsg::getName)
.collect(Collectors.toList());
// 含有转码文件的情况下
if (namelist.contains(fileMsg.getName().substring(0, suffixidx) + "mp4")) {
Map<String, Object> map = FfmpegUtil.ffmpegTaskMap.get(fileMsg.getLink());
// 含有转码文件且有转码记录
if (null != map) {
String transcode = (String) map.get("flag");
fileMsg.setTranscode(transcode);
}
// 含有转码文件但没有转码记录，说明之前已完成转码
else {
fileMsg.setTranscode("complete");
}
}
// 没有转码文件说明可转码
else {
fileMsg.setTranscode("transcodable");
}
}
}
return fileMsgList;
}
// 用户根目录文件列出
@RequestMapping(value = "/userfilelist", produces = "application/json; charset=utf-8")
@ResponseBody
public List<FileMsg> userFileList(@RequestParam(required = false) String key, String path,
HttpServletRequest request) {
if (path == null) {
path = "/";
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// 列出用户文件
List<FileMsg> fileMsgList = fileService.userFileList(userName, path);
// 判断文件转码情况
for (FileMsg fileMsg : fileMsgList) {
// 跳过文件夹
if (fileMsg.getSize().equals("Directory")) {
continue;
}
// 正常文件
int suffixidx = (int) StringUtil.getfilesuffix(fileMsg.getName(), true);
String suffix = fileMsg.getName().substring(suffixidx);
if (suffix.equals("mkv") || suffix.equals("rmvb") || suffix.equals("avi") || suffix.equals("wmv")
|| suffix.equals("3gp") || suffix.equals("rm")) {
// 取非文件夹的所有文件名
List<String> namelist = fileMsgList.stream()
.filter(f -> !f.getSize().equals("Directory"))
.map(FileMsg::getName)
.collect(Collectors.toList());
// 含有转码文件的情况下
if (namelist.contains(fileMsg.getName().substring(0, suffixidx) + "mp4")) {
Map<String, Object> map = FfmpegUtil.ffmpegTaskMap.get(fileMsg.getLink());
// 含有转码文件且有转码记录
if (null != map) {
String transcode = (String) map.get("flag");
fileMsg.setTranscode(transcode);
}
// 含有转码文件但没有转码记录，说明之前已完成转码
else {
fileMsg.setTranscode("complete");
}
}
// 没有转码文件说明可转码
else {
fileMsg.setTranscode("transcodable");
}
}
}
return fileMsgList;
}
// 查看单个文件转码状态
@RequestMapping(value = "/transcodestatus", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg transCodeStatus(String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
logger.warn("transCodeStatus():" + path);
logger.warn("ffmpegTaskMap:" + JSONObject.toJSONString(FfmpegUtil.ffmpegTaskMap));
ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setMsg("noneed");
responseMsg.setSuccess(true);
// 判断文件转码情况
Map<String, Object> map = FfmpegUtil.ffmpegTaskMap.get(path);
if (null != map) {
Boolean transcode = (Boolean) map.get("flag");
if (transcode) {
responseMsg.setMsg("complete");
} else {
responseMsg.setMsg("transcoding");
}
}
return responseMsg;
}
// 文件删除
@RequestMapping(value = "/filedelete", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg fileDelete(String fileName, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
ResponseMsg j = new ResponseMsg();
if (fileName.isEmpty()) {
j.setMsg("文件名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// 删除文件
Boolean[] b = fileService.userFileDelete(fileName, userName, path);
boolean flag = true;
for (int i = 0; i < b.length; i++) {
if (b[i] == false) {
flag = false;
}
}
if (flag) {
j.setSuccess(true);
j.setMsg("删除成功！");
} else {
j.setMsg("删除失败！");
}
return j;
}
// 文件重命名 文件夹重命名时 老名字写path 新名字写newName oldName填@dir@
@RequestMapping(value = "/filerename", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg fileRename(String oldName, String newName, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
ResponseMsg j = new ResponseMsg();
if (oldName.isEmpty() || newName.isEmpty()) {
j.setMsg("文件名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// 重命名文件
boolean b = fileService.userFileRename(oldName, newName, userName, path);
String saveFilePath = fileRootPath + userName + "/" + path;
String oldNameWithPath = saveFilePath + "/" + oldName;
File file = new File(oldNameWithPath);
if (b) {
j.setSuccess(true);
j.setMsg("重命名成功！");
logger.warn(j.getMsg());
} else if (!file.exists()) {
j.setMsg("没有重命名的权限！");
logger.warn(j.getMsg());
} else {
j.setMsg("重命名失败！");
logger.warn(j.getMsg());
}
return j;
}
// 文件夹创建
@RequestMapping(value = "/dircreate", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg dirCreate(String dirName, String path, HttpServletRequest request) {
if (path == null) {
path = "/";
}
ResponseMsg j = new ResponseMsg();
if (dirName.isEmpty() || path.isEmpty()) {
j.setMsg("文件夹名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// path = /pan/userName/当前path
if (!SystemUtil.isWindows()) {
path = "/pan/" + userName + path;
} else {
path = fileRootPath + userName + path;
}
// 重命名文件
boolean b = fileService.userDirCreate(dirName, path);
if (b) {
j.setSuccess(true);
j.setMsg("文件夹创建成功！");
} else {
j.setMsg("文件夹创建失败！");
}
return j;
}
// 文件提取码->真实地址（验证提取码是否正确）
@RequestMapping(value = "/shareCallBack", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg shareCallBack(String link) {
logger.warn("执行shareCallBack接口！：" + link);
ResponseMsg j = new ResponseMsg();
if (link.isEmpty()) {
j.setMsg("提取码为空！");
return j;
}
String downloadLink = fileService.fileShareCodeDecode(link);
logger.warn("downloadLink:" + downloadLink);
if (!"null".equals(downloadLink)) {
j.setSuccess(true);
j.setMsg(downloadLink);
} else {
j.setMsg("提取码不正确！");
}
return j;
}
// 文件分享下载地址sharefile（创建链接）-----share（定位到分享页面）-shareCallBack(验证提取码是否正确)
@RequestMapping(value = "/sharefile", produces = "application/json; charset=utf-8")
@ResponseBody
public ModelAndView shareFile(String link) {
logger.warn("shareFie接口的运行！");
String downloadLink = "";
if (!link.isEmpty()) {
// downloadLink= fileService.fileShareCodeDecode(link);
downloadLink = link;
}
//在数据库查找这个链接
logger.warn("zc1+downloadLink:" + downloadLink);
LinkSecret linkSecret = linkSecretService.findLinkSecretBysecretLink(downloadLink);
if (linkSecret == null) {
ModelAndView modelAndView = new ModelAndView("errorPage");
modelAndView.addObject("message", "链接失效");
return modelAndView;
} else {
Date date1 = linkSecret.getExpireDate();
//表示链接永久有效
if (date1 == null) {
ModelAndView modelAndView = new ModelAndView("shareSecret");
modelAndView.addObject("link", link);
return modelAndView;
} else {
//得到long类型当前时间
long dataTemp = System.currentTimeMillis();
Date date2 = new Date(dataTemp);
if (date1.before(date2)) {
//不能下载
ModelAndView modelAndView = new ModelAndView("errorPage");
modelAndView.addObject("message", link + "链接失效");
//删除本地共享文件夹的内容
FileUtil.delete(downloadLink);
return modelAndView;
} else {
ModelAndView modelAndView = new ModelAndView("shareSecret");
modelAndView.addObject("link", link);
return modelAndView;
}
}
}
}
//定位到分享页面
@RequestMapping("/share")
public ModelAndView share(String link, HttpServletRequest request) {
logger.warn("zzc:" + link);
EncryptUtil des = null;
String linkDecoder = "";
try {
des = new EncryptUtil(key, "utf-8");
linkDecoder = des.decode(link);
} catch (Exception e) {
logger.error("Exception:", e);
}
ModelAndView modelAndView = new ModelAndView("share");
modelAndView.addObject("link", link);
modelAndView.addObject("linkDecoder", linkDecoder);
User user = (User) request.getSession().getAttribute("user");
if (user != null) {
modelAndView.addObject("author", user.getUserName());
}
return modelAndView;
}
@RequestMapping("/errorPage")
public ModelAndView errorPage(String message) {
logger.warn("zzc:" + message);
ModelAndView modelAndView = new ModelAndView("errorPage");
modelAndView.addObject("message", message);
return modelAndView;
}
@RequestMapping(value = "/sharefileSecret", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg shareFileSecret(String link, String secret, HttpServletRequest request,
HttpServletResponse response) {
logger.warn("执行sharefileSecret接口");
String downloadLink = "";
Map<Object, String> map = new HashMap<>();
if (!link.isEmpty()) {
// downloadLink= fileService.fileShareCodeDecode(link);
downloadLink = link;
}
//在数据库查找这个链接
LinkSecret linkSecret = linkSecretService.findLinkSecretBysecretLink(downloadLink);
String secret1 = linkSecret.getSecret();
if (secret1.toLowerCase().equals(secret.toLowerCase())) {
linkSecretService.addOneToDownloadNum(linkSecret);
logger.warn("密码正确！");
logger.warn("codeArray[1]路径：" + link);
ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setSuccess(true);
responseMsg.setMsg(link);
return responseMsg;
} else {
logger.warn("密码不正确！");
ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setSuccess(false);
responseMsg.setMsg("密码不正确！");
return responseMsg;
}
}
// 文件提取码生成,当再次分享同一个文件，只更新过期时间
@RequestMapping(value = "/generateShareLink", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg generateShareLink(@RequestParam String expireDay, String fileName, String path,
HttpServletRequest request) {
String expireDayString = expireDay;
int expireDays = 3;
if (expireDayString != null) {
if (expireDayString.equals("永久有效")) {
expireDays = -1;
} else {
expireDays = Integer.parseInt(expireDayString);
}
}
if (path == null) {
path = "/";
}
ResponseMsg j = new ResponseMsg();
if (fileName.isEmpty() || path.isEmpty()) {
j.setMsg("文件夹名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
String filePathAndName = userName + "/" + path + "/" + fileName;
filePathAndName = StringUtil.stringSlashToOne(filePathAndName);
logger.warn("filePathAndName:" + filePathAndName);
String b = fileService.fileShareCodeEncode(filePathAndName);
String secret = "";
File file = new File(fileRootPath + "/" + filePathAndName);
String localLink = "/data/share/" + filePathAndName;
//存入数据库
LinkSecret linkSecret = linkSecretService.findLinkSecretByLocalLinkAndUserName(localLink, userName);
if (linkSecret == null) {
//设置提取密码
secret = PassWordCreate.createPassWord(secretLen);
linkSecret = new LinkSecret();
linkSecret.setLocalLink(localLink);
linkSecret.setSecret(secret);
linkSecret.setUserName(userName);
linkSecret.setDownloadNum(0);
linkSecret.setFileName(fileName);
if (expireDays != -1) {
Calendar cal = Calendar.getInstance();
cal.add(Calendar.DATE, expireDays);
Date date = cal.getTime();
linkSecret.setExpireDate(date);
}
logger.warn("b:" + b);
linkSecret.setSecretLink(b);
linkSecretService.save(linkSecret);
//test
LinkSecret linkSecretTemp = linkSecretService.findLinkSecretByLocalLinkAndUserName(localLink, userName);
logger.warn(linkSecretTemp.getSecretLink());
//test
} else {
if (expireDays != -1) {
Calendar cal = Calendar.getInstance();
cal.add(Calendar.DATE, expireDays);
Date date = cal.getTime();
linkSecretService.updateExpireDay(linkSecret, date);
linkSecretService.updateShareDate(linkSecret, new Date());
secret = linkSecret.getSecret();
} else {
linkSecretService.updateExpireDay(linkSecret, null);
secret = linkSecret.getSecret();
linkSecretService.updateShareDate(linkSecret, new Date());
}
}
if (SystemUtil.isWindows()) {
b = linkSecret.getSecretLink() + "##" + secret;
} else {
b = b + "##" + secret;
}
if (!"null".equals(b)) {
j.setSuccess(true);
j.setMsg(b);
} else {
j.setMsg("提取码生成失败！");
}
return j;
}
// 文件、文件夹 移动 文件夹移动时fileName=@dir@
@RequestMapping(value = "/filemove", produces = "application/json; charset=utf-8")
@ResponseBody
public ResponseMsg fileMove(String fileName, String oldPath, String newPath, HttpServletRequest request) {
if (fileName == null) {
fileName = "@dir@";
}
ResponseMsg j = new ResponseMsg();
if (oldPath.isEmpty() || newPath.isEmpty()) {
j.setMsg("路径名字为空！");
return j;
}
// 获取用户名
String userName = WebUtil.getUserNameByRequest(request);
// 移动文件
boolean b = fileService.userFileDirMove(fileName, oldPath, newPath, userName);
if (b) {
j.setSuccess(true);
j.setMsg("移动成功！");
} else {
j.setMsg("移动失败！");
}
return j;
}
/**
* 分块上传 有断点续传的功能
*
* @param request
* @param response
* @param file
* @param path
*/
@RequestMapping(value = "/uploadsevlet", method = RequestMethod.POST)
@ResponseBody
public void uploadSevlet(HttpServletRequest request, HttpServletResponse response, MultipartFile file,
String path) {
// String fileMd5 = request.getParameter("fileMd5");
String chunk = request.getParameter("chunk");
logger.warn("chunk:" + chunk);
String fileName = file.getOriginalFilename();
String userName = WebUtil.getUserNameByRequest(request);
MultipartHttpServletRequest Murequest = (MultipartHttpServletRequest) request;
Map<String, MultipartFile> files = Murequest.getFileMap();
logger.warn("执行前---------");
if (null != files && !files.isEmpty()) {
for (MultipartFile item : files.values()) {
String tempDir = FileUtil.getTempDir(tempPath, userName, fileName);
tempDir = StringUtil.stringSlashToOne(tempDir);
logger.warn("tempDir:" + tempDir);
File dir = new File(tempDir);
if (!dir.exists()) {
dir.mkdirs();
}
// 创建分片文件并保存
File chunkFile = new File(tempDir + "/" + chunk);
logger.warn(tempDir + "/" + chunk);
try {
chunkFile.createNewFile();
item.transferTo(chunkFile);
} catch (IllegalStateException e) {
logger.warn("保存分片文件出错：" + e.getMessage());
logger.error("Exception:", e);
} catch (IOException e) {
logger.warn("保存分片文件出错：" + e.getMessage());
logger.error("Exception:", e);
}
}
}
}
/**
* 上传之前检查
*
* @param request
* @param response
* @return
*/
@RequestMapping(value = "/check", method = RequestMethod.POST)
@ResponseBody
public ResponseMsg checkChunk(HttpServletRequest request, HttpServletResponse response) {
ResponseMsg responseMsg = new ResponseMsg();
logger.warn("执行check-------------------");
String fileName = request.getParameter("fileName");
// String fileMd5 = request.getParameter("fileMd5");
String chunk = request.getParameter("chunk");
logger.warn("checkChunk+chunk:" + chunk);
String chunkSize = request.getParameter("chunkSize");
logger.warn("checkChunk+chunkSize:" + chunkSize);
String userName = WebUtil.getUserNameByRequest(request);
logger.warn(tempPath);
String tempDir = FileUtil.getTempDir(tempPath, userName, fileName);
tempDir = StringUtil.stringSlashToOne(tempDir);
File chunkFile = new File(tempDir + "/" + chunk);
boolean result = false;
// 分片文件是否存在，尺寸是否一致
if (chunkFile.exists() && chunkFile.length() == Integer.parseInt(chunkSize)) {
responseMsg.setSuccess(true);
responseMsg.setMsg(chunk);
}
return responseMsg;
}
/**
* 所有分块上传完成后合并
*
* @param request
* @param response
* @param path
*/
@RequestMapping(value = "/merge")
@ResponseBody
public void mergeChunks(HttpServletRequest request, HttpServletResponse response, String path)
throws InterruptedException {
if (path == null) {
return;
}
logger.warn("执行merge");
String fileName = request.getParameter("fileName");
logger.warn("mergeChunks+fileName:{}", fileName);
User user = (User) request.getSession().getAttribute("user");
String userName = user.getUserName();
boolean b = fileService.merge(fileName, userName, path);
logger.warn("mergeChunks() result:{}", b);
}
/**
* 调用后台转码的功能
*
* @param request
* @param filepath
*/
@PostMapping(value = "/videoconvert")
@ResponseBody
public ResponseMsg videoconvert(HttpServletRequest request, String filepath) {
ResponseMsg j = new ResponseMsg();
if (filepath.isEmpty()) {
j.setMsg("源文件路径名字为空！");
return j;
}
String ffmepgpath = fileRootPath + "/ffmpeg/bin";
Map<String, Object> retmap = ConvertVideo.convertVideo(filepath, ffmepgpath);
String retstr = (String) retmap.get("flag");
j.setMsg((String) retmap.get("path"));
// 成功true并有路径，失败false也有路径，转码中false并且没路径
if ("complete".equals(retstr)) {
j.setSuccess(true);
} else if ("transcoding".equals(retstr)) {
// 这里转码中属于bug情况
j.setMsg("");
}
return j;
}
/**
* 获取云盘服务器所在盘磁盘空间大小
*
* @param request
*/
@GetMapping(value = "/getspacesize")
@ResponseBody
public ResponseMsg getSpaceSize(HttpServletRequest request) {
// 普通用户限制80G，guest用户限制40G，
String userName = WebUtil.getUserNameByRequest(request);
Map<String, String> spaceMap = new HashMap<>();
spaceMap.put("totalSpace", "80");
double totalSpace = 80;
if ("guest".equals(userName)) {
spaceMap.put("totalSpace", "40");
totalSpace = 40;
}
long dirlength = SystemUtil.getDirSpaceSize(fileRootPath + userName);
double dirlengthDouble = dirlength / 1024.0 / 1024 / 1024;
String usedeSpace = String.format("%.2f", dirlengthDouble);
logger.warn("usedeSpace:{}", usedeSpace);
String freeSpace = String.format("%.2f", totalSpace - Double.valueOf(usedeSpace));
logger.warn("freeSpace:{}", freeSpace);
spaceMap.put("freeSpace", freeSpace);
ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setSuccess(true);
responseMsg.setMsg(JSONObject.toJSONString(spaceMap));
String aString = "ddssddsd";
int aaa = aString.lastIndexOf("d");
return responseMsg;
}
}
源码获取：博客首页 "资源" 里下载！