Java+Springboot搭建一個(gè)在線網(wǎng)盤(pán)文件分享系統(tǒng)
更新時(shí)間：2021年11月29日 10:20:44 作者：OldWinePot
本主要介紹了通過(guò)springboot+freemark+jpa+MySQL實(shí)現(xiàn)的在線網(wǎng)盤(pán)文件分享系統(tǒng)，其功能跟百度網(wǎng)盤(pán)非常類似，可以實(shí)現(xiàn)文件的上傳、移動(dòng)、復(fù)制、下載等，需要的可以參考一下
前言
springboot+freemark+jpa+MySQL實(shí)現(xiàn)的在線網(wǎng)盤(pán)文件分享系統(tǒng)，其功能跟百度網(wǎng)盤(pán)非常類似，普通用戶可以注冊(cè)登錄，注冊(cè)后默認(rèn)分配1G的空間大小，登錄進(jìn)去后可以新建文件夾、上傳各種類型的文件、文件移動(dòng)、復(fù)制、下載、刪除、分享，分享分為私密分享和公開(kāi)分享，還可以設(shè)置分享過(guò)期時(shí)間，打開(kāi)分享鏈接后可以對(duì)文件進(jìn)行查看、下載、保存到自己網(wǎng)盤(pán)等。超級(jí)管理員登錄后可以設(shè)置普通用戶的空間大小、角色類型、權(quán)限等。
本系統(tǒng)主要涉及的特色功能有：
1：系統(tǒng)在上傳文件時(shí)自動(dòng)計(jì)算文件的md5值，并且檢查該值是否存在，若存在則文件不再上傳到服務(wù)器，直接進(jìn)行引用原有的文件。
2：定時(shí)器定時(shí)檢查限時(shí)分享的文件是否過(guò)期。
3：網(wǎng)盤(pán)目錄結(jié)構(gòu)可無(wú)限制層級(jí)創(chuàng)建，保存移動(dòng)復(fù)制文件時(shí)遞歸所選文件夾層級(jí)等等。
使用技術(shù)：java+springboot+freemark+jpa+MySQL+maven
效果圖
?
?
主要代碼
管理員控制器：
/**
* @ClassName: AdminController
* @Description: 管理員控制器
**/
@Controller
public class AdminController extends BaseController {
private Logger logger = LogUtils.getInstance(AdminController.class);
/**
* @Description 前往用戶管理頁(yè)面
**/
@GetMapping("/manages-users")
public String manageUsers(Map<String,Object> map,Integer cur){
if (loginUser.getRole() == 1){
//用于無(wú)訪問(wèn)權(quán)限
logger.error("當(dāng)前登錄用戶："+loginUser.getUserName()+"無(wú)管理員權(quán)限！");
return "redirect:/error401Page";
}
//獲取全部的用戶
Integer usersCount = userService.getUsersCount();
//獲取當(dāng)前查詢的頁(yè)數(shù)，如果為空，默認(rèn)為0
cur = (cur == null || cur<0)?0:cur;
//獲得統(tǒng)計(jì)信息
FileStoreStatistics statistics = myFileService.getCountStatistics(loginUser.getFileStoreId());
//分頁(yè)獲得20個(gè)用戶信息
Page<Object> page = PageHelper.startPage(cur, 20);
List<UserToShow> users = userService.getUsers();
map.put("statistics", statistics);
map.put("users", users);
map.put("page", page);
map.put("usersCount", usersCount);
logger.info("用戶管理域的內(nèi)容："+map);
return "admin/manage-users";
}
/**
* @Description 修改用戶的權(quán)限和最大容量
**/
@GetMapping("/updateStoreInfo")
@ResponseBody
public String updateStoreInfo(Integer uId,Integer pre,Integer size){
Integer integer = fileStoreService.updatePermission(uId, pre, size*1024);
if (integer == 1) {
//更新成功，返回200狀態(tài)碼
logger.info("修改用戶"+userService.queryById(uId).getUserName()+"：的權(quán)限和倉(cāng)庫(kù)大小成功！");
return "200";
}else {
//更新失敗，返回500狀態(tài)碼
logger.error("修改用戶"+userService.queryById(uId).getUserName()+"：的權(quán)限和倉(cāng)庫(kù)大小失?。?);
return "500";
}
}
/**
* @Description 刪除用戶
**/
@GetMapping("/deleteUser")
public String deleteUser(Integer uId,Integer cur){
cur = (cur == null || cur < 0)?1:cur;
User user = userService.queryById(uId);
FileStore fileStore = fileStoreService.getFileStoreByUserId(uId);
List<FileFolder> folders = fileFolderService.getRootFoldersByFileStoreId(fileStore.getFileStoreId());
//迭代刪除文件夾
for (FileFolder f:folders) {
deleteFolderF(f);
}
List<MyFile> files = myFileService.getRootFilesByFileStoreId(fileStore.getFileStoreId());
//刪除該用戶倉(cāng)庫(kù)根目錄下的所有文件
for (MyFile f:files) {
String remotePath = f.getMyFilePath();
String fileName = f.getMyFileName()+f.getPostfix();
//從FTP文件服務(wù)器上刪除文件
boolean b = FtpUtil.deleteFile("/"+remotePath, fileName);
if (b){
//刪除成功,返回空間
fileStoreService.subSize(f.getFileStoreId(),Integer.valueOf(f.getSize()));
//刪除文件表對(duì)應(yīng)的數(shù)據(jù)
myFileService.deleteByFileId(f.getMyFileId());
}
logger.info("刪除文件成功!"+f);
}
if (FtpUtil.deleteFolder("/" + uId)){
logger.info("清空FTP上該用戶的文件成功");
}else {
logger.error("清空FTP上該用戶的文件失敗");
}
userService.deleteById(uId);
fileStoreService.deleteById(fileStore.getFileStoreId());
return "redirect:/manages-users?cur="+cur;
}
/**
* @Description 迭代刪除文件夾里面的所有文件和子文件夾
**/
public void deleteFolderF(FileFolder folder){
//獲得當(dāng)前文件夾下的所有子文件夾
List<FileFolder> folders = fileFolderService.getFileFolderByParentFolderId(folder.getFileFolderId());
//刪除當(dāng)前文件夾的所有的文件
List<MyFile> files = myFileService.getFilesByParentFolderId(folder.getFileFolderId());
if (files.size()!=0){
for (int i = 0; i < files.size(); i++) {
Integer fileId = files.get(i).getMyFileId();
boolean b = FtpUtil.deleteFile("/"+files.get(i).getMyFilePath(), files.get(i).getMyFileName() + files.get(i).getPostfix());
if (b){
myFileService.deleteByFileId(fileId);
fileStoreService.subSize(folder.getFileStoreId(),Integer.valueOf(files.get(i).getSize()));
}
}
}
if (folders.size()!=0){
for (int i = 0; i < folders.size(); i++) {
deleteFolderF(folders.get(i));
}
}
fileFolderService.deleteFileFolderById(folder.getFileFolderId());
}
}
文件倉(cāng)庫(kù)控制器：
/**
* @ClassName: FileStoreController
* @Description: 文件倉(cāng)庫(kù)控制器
**/
@Controller
public class FileStoreController extends BaseController {
private Logger logger = LogUtils.getInstance(FileStoreController.class);
/**
* @Description 上傳臨時(shí)文件
**/
@PostMapping("/uploadTempFile")
public String uploadTempFile(@RequestParam("file") MultipartFile file,String url) {
session.setAttribute("imgPath","https://ss3.bdstatic.com/70cFv8Sh_Q1YnxGkpoWK1HF6hhy/it/u=2654852821,3851565636&fm=26&gp=0.jpg");
String name = file.getOriginalFilename().replaceAll(" ","");
if (!checkTarget(name)){
logger.error("臨時(shí)文件上傳失敗!文件名不符合規(guī)范...");
session.setAttribute("msg", "上傳失敗!文件名不符合規(guī)范");
return "redirect:/temp-file";
}
SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
String dateStr = format.format(new Date());
String path = "temp/"+dateStr +"/"+UUID.randomUUID();
try {
if (FtpUtil.uploadFile("/"+path, name, file.getInputStream())){
//上傳成功
logger.info("臨時(shí)文件上傳成功!"+name);
String size = String.valueOf(file.getSize());
TempFile tempFile = TempFile.builder().fileName(name).filePath(path).size(size).uploadTime(new Date()).build();
if (tempFileService.insert(tempFile)) {
try {
String id = UUID.randomUUID().toString();
String p = request.getSession().getServletContext().getRealPath("/user_img/");
Long t = tempFile.getUploadTime().getTime();
url = url+"/file/share?t="+ UUID.randomUUID().toString().substring(0,10) +"&f="+tempFile.getFileId()+"&p="+size+"&flag=2";
File targetFile = new File(p, "");
if (!targetFile.exists()) {
targetFile.mkdirs();
}
File f = new File(p, id + ".jpg");
if (!f.exists()){
//文件不存在,開(kāi)始生成二維碼并保存文件
OutputStream os = new FileOutputStream(f);
QRCodeUtil.encode(url, "/static/img/logo.png", os, true);
os.close();
}
//異步刪除臨時(shí)文件
tempFileService.deleteById(tempFile.getFileId());
session.setAttribute("imgPath","user_img/"+id+".jpg");
session.setAttribute("url",url);
session.setAttribute("msg","上傳成功，掃碼/訪問(wèn)鏈接 即可下載！");
return "redirect:/temp-file";
} catch (Exception e) {
e.printStackTrace();
}
}else {
logger.info("臨時(shí)文件數(shù)據(jù)庫(kù)寫(xiě)入失敗!"+name);
session.setAttribute("url","error");
session.setAttribute("msg", "服務(wù)器出錯(cuò)了，臨時(shí)文件上傳失敗!");
}
}else{
//上傳失敗
logger.info("臨時(shí)文件上傳失敗!"+name);
session.setAttribute("url","error");
session.setAttribute("msg", "服務(wù)器出錯(cuò)了，上傳失敗!");
}
} catch (IOException e) {
e.printStackTrace();
}
return "redirect:/temp-file";
}
/**
* @Description 網(wǎng)盤(pán)的文件上傳
* @Author xw
* @Date 23:10 2020/2/10
* @Param [files]
* @return java.util.Map<java.lang.String,java.lang.Object>
**/
@PostMapping("/uploadFile")
@ResponseBody
public Map<String, Object> uploadFile(@RequestParam("file") MultipartFile files) {
Map<String, Object> map = new HashMap<>();
if (fileStoreService.getFileStoreByUserId(loginUser.getUserId()).getPermission() != 0){
logger.error("用戶沒(méi)有上傳文件的權(quán)限!上傳失敗...");
map.put("code", 499);
return map;
}
FileStore store = fileStoreService.getFileStoreByUserId(loginUser.getUserId());
Integer folderId = Integer.valueOf(request.getHeader("id"));
String name = files.getOriginalFilename().replaceAll(" ","");
//獲取當(dāng)前目錄下的所有文件，用來(lái)判斷是否已經(jīng)存在
List<MyFile> myFiles = null;
if (folderId == 0){
//當(dāng)前目錄為根目錄
myFiles = myFileService.getRootFilesByFileStoreId(loginUser.getFileStoreId());
}else {
//當(dāng)前目錄為其他目錄
myFiles = myFileService.getFilesByParentFolderId(folderId);
}
for (int i = 0; i < myFiles.size(); i++) {
if ((myFiles.get(i).getMyFileName()+myFiles.get(i).getPostfix()).equals(name)){
logger.error("當(dāng)前文件已存在!上傳失敗...");
map.put("code", 501);
return map;
}
}
SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
String dateStr = format.format(new Date());
String path = loginUser.getUserId()+"/"+dateStr +"/"+folderId;
if (!checkTarget(name)){
logger.error("上傳失敗!文件名不符合規(guī)范...");
map.put("code", 502);
return map;
}
Integer sizeInt = Math.toIntExact(files.getSize() / 1024);
//是否倉(cāng)庫(kù)放不下該文件
if(store.getCurrentSize()+sizeInt > store.getMaxSize()){
logger.error("上傳失敗!倉(cāng)庫(kù)已滿。");
map.put("code", 503);
return map;
}
//處理文件大小
String size = String.valueOf(files.getSize()/1024.0);
int indexDot = size.lastIndexOf(".");
size = size.substring(0,indexDot);
int index = name.lastIndexOf(".");
String tempName = name;
String postfix = "";
int type = 4;
if (index!=-1){
tempName = name.substring(index);
name = name.substring(0,index);
//獲得文件類型
type = getType(tempName.toLowerCase());
postfix = tempName.toLowerCase();
}
try {
//提交到FTP服務(wù)器
boolean b = FtpUtil.uploadFile("/"+path, name + postfix, files.getInputStream());
if (b){
//上傳成功
logger.info("文件上傳成功!"+files.getOriginalFilename());
//向數(shù)據(jù)庫(kù)文件表寫(xiě)入數(shù)據(jù)
myFileService.addFileByFileStoreId(
MyFile.builder()
.myFileName(name).fileStoreId(loginUser.getFileStoreId()).myFilePath(path)
.downloadTime(0).uploadTime(new Date()).parentFolderId(folderId).
size(Integer.valueOf(size)).type(type).postfix(postfix).build());
//更新倉(cāng)庫(kù)表的當(dāng)前大小
fileStoreService.addSize(store.getFileStoreId(),Integer.valueOf(size));
try {
Thread.sleep(5000);
map.put("code", 200);
} catch (InterruptedException e) {
e.printStackTrace();
}
}else{
logger.error("文件上傳失敗!"+files.getOriginalFilename());
map.put("code", 504);
}
} catch (IOException e) {
e.printStackTrace();
}
return map;
}
/**
* @Description 網(wǎng)盤(pán)的文件下載
**/
@GetMapping("/downloadFile")
public String downloadFile(@RequestParam Integer fId){
if (fileStoreService.getFileStoreByUserId(loginUser.getUserId()).getPermission() == 2){
logger.error("用戶沒(méi)有下載文件的權(quán)限!下載失敗...");
return "redirect:/error401Page";
}
//獲取文件信息
MyFile myFile = myFileService.getFileByFileId(fId);
String remotePath = myFile.getMyFilePath();
String fileName = myFile.getMyFileName()+myFile.getPostfix();
try {
//去FTP上拉取
OutputStream os = new BufferedOutputStream(response.getOutputStream());
response.setCharacterEncoding("utf-8");
// 設(shè)置返回類型
response.setContentType("multipart/form-data");
// 文件名轉(zhuǎn)碼一下，不然會(huì)出現(xiàn)中文亂碼
response.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(fileName, "UTF-8"));
boolean flag = FtpUtil.downloadFile("/" + remotePath, fileName, os);
if (flag) {
myFileService.updateFile(
MyFile.builder().myFileId(myFile.getMyFileId()).downloadTime(myFile.getDownloadTime() + 1).build());
os.flush();
os.close();
logger.info("文件下載成功!" + myFile);
}
} catch (Exception e) {
e.printStackTrace();
}
return "success";
}
/**
* @Description 刪除文件
**/
@GetMapping("/deleteFile")
public String deleteFile(@RequestParam Integer fId,Integer folder){
//獲得文件信息
MyFile myFile = myFileService.getFileByFileId(fId);
String remotePath = myFile.getMyFilePath();
String fileName = myFile.getMyFileName()+myFile.getPostfix();
//從FTP文件服務(wù)器上刪除文件
boolean b = FtpUtil.deleteFile("/"+remotePath, fileName);
if (b){
//刪除成功,返回空間
fileStoreService.subSize(myFile.getFileStoreId(),Integer.valueOf(myFile.getSize()));
//刪除文件表對(duì)應(yīng)的數(shù)據(jù)
myFileService.deleteByFileId(fId);
}
logger.info("刪除文件成功!"+myFile);
return "redirect:/files?fId="+folder;
}
/**
* @Description 刪除文件夾并清空文件
**/
@GetMapping("/deleteFolder")
public String deleteFolder(@RequestParam Integer fId){
FileFolder folder = fileFolderService.getFileFolderByFileFolderId(fId);
//強(qiáng)制刪除
deleteFolderF(folder);
return folder.getParentFolderId() == 0?"redirect:/files":"redirect:/files?fId="+folder.getParentFolderId();
}
/**
* @Description 迭代刪除文件夾里面的所有文件和子文件夾
**/
public void deleteFolderF(FileFolder folder){
//獲得當(dāng)前文件夾下的所有子文件夾
List<FileFolder> folders = fileFolderService.getFileFolderByParentFolderId(folder.getFileFolderId());
//刪除當(dāng)前文件夾的所有的文件
List<MyFile> files = myFileService.getFilesByParentFolderId(folder.getFileFolderId());
if (files.size()!=0){
for (int i = 0; i < files.size(); i++) {
Integer fileId = files.get(i).getMyFileId();
boolean b = FtpUtil.deleteFile("/"+files.get(i).getMyFilePath(), files.get(i).getMyFileName() + files.get(i).getPostfix());
if (b){
myFileService.deleteByFileId(fileId);
fileStoreService.subSize(folder.getFileStoreId(),Integer.valueOf(files.get(i).getSize()));
}
}
}
if (folders.size()!=0){
for (int i = 0; i < folders.size(); i++) {
deleteFolderF(folders.get(i));
}
}
fileFolderService.deleteFileFolderById(folder.getFileFolderId());
}
/**
* @Description 添加文件夾
**/
@PostMapping("/addFolder")
public String addFolder(FileFolder folder,Map<String, Object> map) {
//設(shè)置文件夾信息
folder.setFileStoreId(loginUser.getFileStoreId());
folder.setTime(new Date());
//獲得當(dāng)前目錄下的所有文件夾,檢查當(dāng)前文件夾是否已經(jīng)存在
List<FileFolder> fileFolders = null;
if (folder.getParentFolderId() == 0){
//向用戶根目錄添加文件夾
fileFolders = fileFolderService.getRootFoldersByFileStoreId(loginUser.getFileStoreId());
}else{
//向用戶的其他目錄添加文件夾
fileFolders = fileFolderService.getFileFolderByParentFolderId(folder.getParentFolderId());
}
for (int i = 0; i < fileFolders.size(); i++) {
FileFolder fileFolder = fileFolders.get(i);
if (fileFolder.getFileFolderName().equals(folder.getFileFolderName())){
logger.info("添加文件夾失敗!文件夾已存在...");
return "redirect:/files?error=1&fId="+folder.getParentFolderId();
}
}
//向數(shù)據(jù)庫(kù)寫(xiě)入數(shù)據(jù)
Integer integer = fileFolderService.addFileFolder(folder);
logger.info("添加文件夾成功!"+folder);
return "redirect:/files?fId="+folder.getParentFolderId();
}
/**
* @Description 重命名文件夾
**/
@PostMapping("/updateFolder")
public String updateFolder(FileFolder folder,Map<String, Object> map) {
//獲得文件夾的數(shù)據(jù)庫(kù)信息
FileFolder fileFolder = fileFolderService.getFileFolderByFileFolderId(folder.getFileFolderId());
fileFolder.setFileFolderName(folder.getFileFolderName());
//獲得當(dāng)前目錄下的所有文件夾,用于檢查文件夾是否已經(jīng)存在
List<FileFolder> fileFolders = fileFolderService.getFileFolderByParentFolderId(fileFolder.getParentFolderId());
for (int i = 0; i < fileFolders.size(); i++) {
FileFolder folder1 = fileFolders.get(i);
if (folder1.getFileFolderName().equals(folder.getFileFolderName()) && folder1.getFileFolderId() != folder.getFileFolderId()){
logger.info("重命名文件夾失敗!文件夾已存在...");
return "redirect:/files?error=2&fId="+fileFolder.getParentFolderId();
}
}
//向數(shù)據(jù)庫(kù)寫(xiě)入數(shù)據(jù)
Integer integer = fileFolderService.updateFileFolderById(fileFolder);
logger.info("重命名文件夾成功!"+folder);
return "redirect:/files?fId="+fileFolder.getParentFolderId();
}
/**
* @Description 重命名文件
**/
@PostMapping("/updateFileName")
public String updateFileName(MyFile file,Map<String, Object> map) {
MyFile myFile = myFileService.getFileByFileId(file.getMyFileId());
if (myFile != null){
String oldName = myFile.getMyFileName();
String newName = file.getMyFileName();
if (!oldName.equals(newName)){
boolean b = FtpUtil.reNameFile(myFile.getMyFilePath() + "/" + oldName+myFile.getPostfix(), myFile.getMyFilePath() + "/" + newName+myFile.getPostfix());
if (b){
Integer integer = myFileService.updateFile(
MyFile.builder().myFileId(myFile.getMyFileId()).myFileName(newName).build());
if (integer == 1){
logger.info("修改文件名成功!原文件名:"+oldName+" 新文件名:"+newName);
}else{
logger.error("修改文件名失敗!原文件名:"+oldName+" 新文件名:"+newName);
}
}
}
}
return "redirect:/files?fId="+myFile.getParentFolderId();
}
/**
* @Description 獲得二維碼
**/
@GetMapping("getQrCode")
@ResponseBody
public Map<String,Object> getQrCode(@RequestParam Integer id,@RequestParam String url){
Map<String,Object> map = new HashMap<>();
map.put("imgPath","https://ss3.bdstatic.com/70cFv8Sh_Q1YnxGkpoWK1HF6hhy/it/u=2654852821,3851565636&fm=26&gp=0.jpg");
if (id != null){
MyFile file = myFileService.getFileByFileId(id);
if (file != null){
try {
String path = request.getSession().getServletContext().getRealPath("/user_img/");
url = url+"/file/share?t="+ UUID.randomUUID().toString().substring(0,10) +"&f="+file.getMyFileId()+"&p="+file.getUploadTime().getTime()+""+file.getSize()+"&flag=1";
File targetFile = new File(path, "");
if (!targetFile.exists()) {
targetFile.mkdirs();
}
File f = new File(path, id + ".jpg");
if (!f.exists()){
//文件不存在,開(kāi)始生成二維碼并保存文件
OutputStream os = new FileOutputStream(f);
QRCodeUtil.encode(url, "/static/img/logo.png", os, true);
os.close();
}
map.put("imgPath","user_img/"+id+".jpg");
map.put("url",url);
} catch (Exception e) {
e.printStackTrace();
}
}
}
return map;
}
/**
* @Description 分享文件
**/
@GetMapping("/file/share")
public String shareFile(Integer f,String p,String t,Integer flag){
String fileNameTemp = "";
String remotePath = "";
String fileName = "";
Integer times = 0;
if (flag == null || f == null || p == null || t == null){
logger.info("下載分享文件失敗，參數(shù)錯(cuò)誤");
return "redirect:/error400Page";
}
if(flag == 1){
//獲取文件信息
MyFile myFile = myFileService.getFileByFileId(f);
if (myFile == null){
return "redirect:/error404Page";
}
String pwd = myFile.getUploadTime().getTime()+""+myFile.getSize();
if (!pwd.equals(p)){
return "redirect:/error400Page";
}
remotePath = myFile.getMyFilePath();
fileName = myFile.getMyFileName()+myFile.getPostfix();
}else if(flag == 2){
TempFile tempFile = tempFileService.queryById(f);
if (tempFile == null){
return "redirect:/error404Page";
}
Long test = tempFile.getUploadTime().getTime();
String pwd = tempFile.getSize();
if (!pwd.equals(p)){
return "redirect:/error400Page";
}
remotePath = tempFile.getFilePath();
fileName = tempFile.getFileName();
}else {
return "redirect:/error400Page";
}
fileNameTemp = fileName;
try {
//解決下載文件時(shí) 中文文件名亂碼問(wèn)題
boolean isMSIE = isMSBrowser(request);
if (isMSIE) {
//IE瀏覽器的亂碼問(wèn)題解決
fileNameTemp = URLEncoder.encode(fileNameTemp, "UTF-8");
} else {
//萬(wàn)能亂碼問(wèn)題解決
fileNameTemp = new String(fileNameTemp.getBytes("UTF-8"), "ISO-8859-1");
}
//去FTP上拉取
OutputStream os = new BufferedOutputStream(response.getOutputStream());
response.setCharacterEncoding("utf-8");
// 設(shè)置返回類型
response.setContentType("multipart/form-data");
// 文件名轉(zhuǎn)碼一下，不然會(huì)出現(xiàn)中文亂碼
response.setHeader("Content-Disposition", "attachment;fileName=" + fileNameTemp);
if (FtpUtil.downloadFile("/" + remotePath, fileName, os)) {
myFileService.updateFile(
MyFile.builder().myFileId(f).downloadTime(times + 1).build());
os.flush();
os.close();
logger.info("文件下載成功!");
}
} catch (Exception e) {
e.printStackTrace();
}
return "success";
}
/**
* @Description 根據(jù)文件的后綴名獲得對(duì)應(yīng)的類型
* @return int 1:文本類型 2:圖像類型 3:視頻類型 4:音樂(lè)類型 5:其他類型
**/
public int getType(String type){
if (".chm".equals(type)||".txt".equals(type)||".xmind".equals(type)||".xlsx".equals(type)||".md".equals(type)
||".doc".equals(type)||".docx".equals(type)||".pptx".equals(type)
||".wps".equals(type)||".word".equals(type)||".html".equals(type)||".pdf".equals(type)){
return 1;
}else if (".bmp".equals(type)||".gif".equals(type)||".jpg".equals(type)||".ico".equals(type)||".vsd".equals(type)
||".pic".equals(type)||".png".equals(type)||".jepg".equals(type)||".jpeg".equals(type)||".webp".equals(type)
||".svg".equals(type)){
return 2;
} else if (".avi".equals(type)||".mov".equals(type)||".qt".equals(type)
||".asf".equals(type)||".rm".equals(type)||".navi".equals(type)||".wav".equals(type)
||".mp4".equals(type)||".mkv".equals(type)||".webm".equals(type)){
return 3;
} else if (".mp3".equals(type)||".wma".equals(type)){
return 4;
} else {
return 5;
}
}
/**
* @Description 正則驗(yàn)證文件名是否合法 [漢字,字符,數(shù)字,下劃線,英文句號(hào),橫線]
**/
public boolean checkTarget(String target) {
final String format = "[^\\u4E00-\\u9FA5\\uF900-\\uFA2D\\w-_.]";
Pattern pattern = Pattern.compile(format);
Matcher matcher = pattern.matcher(target);
return !matcher.find();
}
/**
* @Description 判斷當(dāng)前瀏覽器是否為ie
**/
public static boolean isMSBrowser(HttpServletRequest request) {
String[] IEBrowserSignals = {"MSIE", "Trident", "Edge"};
String userAgent = request.getHeader("User-Agent");
for (String signal : IEBrowserSignals) {
if (userAgent.contains(signal)){
return true;
}
}
return false;
}
}
登錄控制器：
/**
* @Description 登錄控制器
**/
@Controller
public class LoginController extends BaseController {
private Logger logger = LogUtils.getInstance(LoginController.class);
/**
* @Description 免登陸用戶入口，用于本地開(kāi)發(fā)測(cè)試，上線運(yùn)營(yíng)為了安全請(qǐng)刪除此方法
* @Author xw
* @Date 15:17 2020/2/26
* @Param []
* @return java.lang.String
**/
@GetMapping("/admin")
public String adminLogin(){
User user = userService.getUserByOpenId("123456");
logger.info("使用免登陸方式登錄成功！"+user);
session.setAttribute("loginUser", user);
return "redirect:/index";
}
/**
* 用于注冊(cè)流程,用戶名，密碼，郵箱等校驗(yàn)工作由前端來(lái)完成
*/
@PostMapping("/register")
public String register(User user, String code, Map<String, Object> map) {
String uCode = (String) session.getAttribute(user.getEmail() + "_code");
if (!code.equals(uCode)) {
map.put("errorMsg", "驗(yàn)證碼錯(cuò)誤");
return "index";
}
// 用戶名去空格
user.setUserName(user.getUserName().trim());
user.setImagePath("https://p.qpic.cn/qqconnect/0/app_101851241_1582451550/100?max-age=2592000&t=0");
user.setRegisterTime(new Date());
user.setRole(1);
if (userService.insert(user)) {
FileStore store = FileStore.builder().userId(user.getUserId()).currentSize(0).build();
fileStoreService.addFileStore(store);
user.setFileStoreId(store.getFileStoreId());
userService.update(user);
logger.info("注冊(cè)用戶成功！當(dāng)前注冊(cè)用戶" + user);
logger.info("注冊(cè)倉(cāng)庫(kù)成功！當(dāng)前注冊(cè)倉(cāng)庫(kù)" + store);
} else {
map.put("errorMsg", "服務(wù)器發(fā)生錯(cuò)誤，注冊(cè)失敗");
return "index";
}
session.removeAttribute(user.getEmail() + "_code");
session.setAttribute("loginUser", user);
return "redirect:/index";
}
/**
* 用戶登錄
*/
@PostMapping("/login")
public String login(User user, Map<String, Object> map) {
User userByEmail = userService.getUserByEmail(user.getEmail());
if (userByEmail != null && userByEmail.getPassword().equals(user.getPassword())) {
session.setAttribute("loginUser", userByEmail);
logger.info("登錄成功！"+userByEmail);
return "redirect:/index";
}else{
User user1 = userService.getUserByEmail(user.getEmail());
String errorMsg = user1 == null ? "該郵箱尚未注冊(cè)" : "密碼錯(cuò)誤";
logger.info("登錄失?。≌?qǐng)確認(rèn)郵箱和密碼是否正確！");
//登錄失敗，將失敗信息返回前端渲染
map.put("errorMsg", errorMsg);
return "index";
}
}
/**
* @return void
* @Description 向注冊(cè)郵箱發(fā)送驗(yàn)證碼, 并驗(yàn)證郵箱是否已使用
**/
@ResponseBody
@RequestMapping("/sendCode")
public String sendCode(String userName, String email, String password) {
User userByEmail = userService.getUserByEmail(email);
if (userByEmail != null) {
logger.error("發(fā)送驗(yàn)證碼失??！郵箱已被注冊(cè)！");
return "exitEmail";
}
logger.info("開(kāi)始發(fā)送郵件.../n" + "獲取的到郵件發(fā)送對(duì)象為:" + mailSender);
mailUtils = new MailUtils(mailSender);
String code = "123456";
session.setAttribute(email + "_code", code);
return "success";
}
/**
* @Description 請(qǐng)求QQ登錄
**/
@GetMapping("/loginByQQ")
public void login() {
response.setContentType("text/html;charset=utf-8");
try {
response.sendRedirect(new Oauth().getAuthorizeURL(request));
logger.info("請(qǐng)求QQ登錄,開(kāi)始跳轉(zhuǎn)...");
} catch (QQConnectException | IOException e) {
e.printStackTrace();
}
}
/**
* @Description QQ登錄回調(diào)地址
**/
@GetMapping("/connection")
public String connection() {
try {
AccessToken accessTokenObj = (new Oauth()).getAccessTokenByRequest(request);
String accessToken = null, openID = null;
long tokenExpireIn = 0L;
if ("".equals(accessTokenObj.getAccessToken())) {
logger.error("登錄失敗:沒(méi)有獲取到響應(yīng)參數(shù)");
return "accessTokenObj=>" + accessTokenObj + "; accessToken" + accessTokenObj.getAccessToken();
} else {
accessToken = accessTokenObj.getAccessToken();
tokenExpireIn = accessTokenObj.getExpireIn();
logger.error("accessToken" + accessToken);
request.getSession().setAttribute("demo_access_token", accessToken);
request.getSession().setAttribute("demo_token_expirein", String.valueOf(tokenExpireIn));
// 利用獲取到的accessToken 去獲取當(dāng)前用的openid -------- start
OpenID openIDObj = new OpenID(accessToken);
openID = openIDObj.getUserOpenID();
UserInfo qzoneUserInfo = new UserInfo(accessToken, openID);
UserInfoBean userInfoBean = qzoneUserInfo.getUserInfo();
if (userInfoBean.getRet() == 0) {
logger.info("用戶的OPEN_ID: " + openID);
logger.info("用戶的昵稱: " + removeNonBmpUnicode(userInfoBean.getNickname()));
logger.info("用戶的頭像URI: " + userInfoBean.getAvatar().getAvatarURL100());
//設(shè)置用戶信息
User user = userService.getUserByOpenId(openID);
if (user == null){
user = User.builder()
.openId(openID).userName(removeNonBmpUnicode(userInfoBean.getNickname()))
.imagePath(userInfoBean.getAvatar().getAvatarURL100()).
registerTime(new Date()).build();
if (userService.insert(user)){
logger.info("注冊(cè)用戶成功！當(dāng)前注冊(cè)用戶" + user);
FileStore store = FileStore.builder().userId(user.getUserId()).build();
if (fileStoreService.addFileStore(store) == 1){
user.setFileStoreId(store.getFileStoreId());
userService.update(user);
logger.info("注冊(cè)倉(cāng)庫(kù)成功！當(dāng)前注冊(cè)倉(cāng)庫(kù)" + store);
}
} else {
logger.error("注冊(cè)用戶失敗！");
}
}else {
user.setUserName(removeNonBmpUnicode(userInfoBean.getNickname()));
user.setImagePath(userInfoBean.getAvatar().getAvatarURL100());
userService.update(user);
}
logger.info("QQ用戶登錄成功！"+user);
session.setAttribute("loginUser", user);
return "redirect:/index";
} else {
logger.error("很抱歉，我們沒(méi)能正確獲取到您的信息，原因是： " + userInfoBean.getMsg());
}
}
} catch (QQConnectException e) {
} finally {
logger.error("登錄成功!");
}
return "登錄失敗!請(qǐng)查看日志信息...";
}
/**
* @Description 處理掉QQ網(wǎng)名中的特殊表情
**/
public String removeNonBmpUnicode(String str) {
if (str == null) {
return null;
}
str = str.replaceAll("[^\\u0000-\\uFFFF]", "");
if ("".equals(str)) {
str = "($ _ $)";
}
return str;
}
/**
* @Description 退出登錄，清空session
**/
@GetMapping("/logout")
public String logout() {
logger.info("用戶退出登錄！");
session.invalidate();
return "redirect:/";
}
}
FTP工具類：?
/**
* @ClassName: FtpUtil
* @Description: FTP工具類
**/
public class FtpUtil {
/**
* FTP服務(wù)器hostname
*/
private static String HOST = "192.168.1.106";
/**
* FTP服務(wù)器端口
*/
private static int PORT = 21;
/**
* FTP登錄賬號(hào)
*/
private static String USERNAME = "chen";
/**
* FTP登錄密碼
*/
private static String PASSWORD = "chenfuning";
/**
* FTP服務(wù)器基礎(chǔ)目錄
*/
private static String BASEPATH = "";
/**
* FTP客戶端
*/
private static FTPClient ftp;
/**
* @Description 初始化FTP客戶端
**/
public static boolean initFtpClient(){
ftp = new FTPClient();
int reply;
try {
// 連接FTP服務(wù)器
ftp.connect(HOST, PORT);
//登錄, 如果采用默認(rèn)端口，可以使用ftp.connect(host)的方式直接連接FTP服務(wù)器
ftp.login(USERNAME, PASSWORD);
ftp.setBufferSize(10240);
//設(shè)置傳輸超時(shí)時(shí)間為60秒
ftp.setDataTimeout(600000);
//連接超時(shí)為60秒
ftp.setConnectTimeout(600000);
//FTP以二進(jìn)制形式傳輸
ftp.setFileType(FTP.BINARY_FILE_TYPE);
reply = ftp.getReplyCode();
if (!FTPReply.isPositiveCompletion(reply)) {
ftp.disconnect();
return false;
}
} catch (IOException e) {
e.printStackTrace();
}
return true;
}
/**
* Description: 向FTP服務(wù)器上傳文件
* @param filePath FTP服務(wù)器文件存放路徑。例如分日期存放：/2015/01/01。文件的路徑為basePath+filePath
* @param filename 上傳到FTP服務(wù)器上的文件名
* @param input 本地要上傳的文件的 輸入流
* @return 成功返回true，否則返回false
*/
public static boolean uploadFile(String filePath, String filename, InputStream input) {
boolean result = false;
try {
filePath = new String(filePath.getBytes("GBK"),"iso-8859-1");
filename = new String(filename.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return result;
};
//切換到上傳目錄
ftp.enterLocalPassiveMode();
if (!ftp.changeWorkingDirectory(BASEPATH+filePath)) {
//如果目錄不存在創(chuàng)建目錄
String[] dirs = filePath.split("/");
String tempPath = BASEPATH;
for (String dir : dirs) {
if (null == dir || "".equals(dir)){
continue;
}
tempPath += "/" + dir;
if (!ftp.changeWorkingDirectory(tempPath)) {
if (!ftp.makeDirectory(tempPath)) {
return result;
} else {
ftp.changeWorkingDirectory(tempPath);
}
}
}
}
//設(shè)置上傳文件的類型為二進(jìn)制類型
ftp.setFileType(FTP.BINARY_FILE_TYPE);
//上傳文件
ftp.enterLocalPassiveMode();
if (!ftp.storeFile(filename, input)) {
return result;
}
input.close();
ftp.logout();
result = true;
}
catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return result;
}
/**
* Description: 從FTP服務(wù)器下載文件
* @param remotePath FTP服務(wù)器上的相對(duì)路徑
* @param fileName 要下載的文件名
* @return
*/
public static boolean downloadFile( String remotePath,String fileName,String localPath) {
boolean result = false;
try {
remotePath = new String(remotePath.getBytes("GBK"),"iso-8859-1");
fileName = new String(fileName.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return result;
};
// 轉(zhuǎn)移到FTP服務(wù)器目錄
ftp.changeWorkingDirectory(remotePath);
ftp.enterLocalPassiveMode();
FTPFile[] fs = ftp.listFiles();
for (FTPFile ff : fs) {
if (ff.getName().equals(fileName)) {
ftp.enterLocalPassiveMode();
FileOutputStream outputStream = new FileOutputStream(new File(localPath));
ftp.retrieveFile(remotePath+"/"+fileName,outputStream);
result = true;
outputStream.close();
}
}
ftp.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return result;
}
/**
* @Description 從ftp服務(wù)器下載文件到指定輸出流
* @Param [remotePath, fileName, outputStream]
* @return boolean
**/
public static boolean downloadFile(String remotePath, String fileName, OutputStream outputStream) {
boolean result = false;
try {
remotePath = new String(remotePath.getBytes("GBK"),"iso-8859-1");
fileName = new String(fileName.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return result;
};
// 轉(zhuǎn)移到FTP服務(wù)器目錄
ftp.changeWorkingDirectory(remotePath);
ftp.enterLocalPassiveMode();
FTPFile[] fs = ftp.listFiles();
for (FTPFile ff : fs) {
if (ff.getName().equals(fileName)) {
ftp.enterLocalPassiveMode();
ftp.retrieveFile(remotePath+"/"+fileName,outputStream);
result = true;
}
}
ftp.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return result;
}
/**
* @Description 刪除文件
* @Param [remotePath, fileName]
* @return void
**/
public static boolean deleteFile( String remotePath,String fileName){
boolean flag = false;
try {
remotePath = new String(remotePath.getBytes("GBK"),"iso-8859-1");
fileName = new String(fileName.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return flag;
};
// 轉(zhuǎn)移到FTP服務(wù)器目錄
ftp.changeWorkingDirectory(remotePath);
ftp.enterLocalPassiveMode();
FTPFile[] fs = ftp.listFiles();
for (FTPFile ff : fs) {
if ("".equals(fileName)){
return flag;
}
if (ff.getName().equals(fileName)){
String filePath = remotePath + "/" +fileName;
ftp.deleteFile(filePath);
flag = true;
}
}
ftp.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return flag;
}
/**
* @Description 刪除文件夾
* @Param [remotePath, fileName]
* @return void
**/
public static boolean deleteFolder( String remotePath){
boolean flag = false;
try {
remotePath = new String(remotePath.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return flag;
};
// 轉(zhuǎn)移到FTP服務(wù)器目錄
ftp.changeWorkingDirectory(remotePath);
ftp.enterLocalPassiveMode();
FTPFile[] fs = ftp.listFiles();
if (fs.length==0){
ftp.removeDirectory(remotePath);
flag = true;
}
ftp.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return flag;
}
/**
* @Description 修改文件名稱或者文件夾名
* @Param [oldAllName, newAllName]
* @return boolean
**/
public static boolean reNameFile( String oldAllName,String newAllName){
boolean flag = false;
try {
oldAllName = new String(oldAllName.getBytes("GBK"),"iso-8859-1");
newAllName = new String(newAllName.getBytes("GBK"),"iso-8859-1");
if (!initFtpClient()){
return flag;
};
ftp.enterLocalPassiveMode();
ftp.rename(oldAllName,newAllName);
flag = true;
ftp.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (ftp.isConnected()) {
try {
ftp.disconnect();
} catch (IOException ioe) {
}
}
}
return flag;
}
}
以上就是Java+Springboot搭建一個(gè)在線網(wǎng)盤(pán)文件分享系統(tǒng)的詳細(xì)內(nèi)容，更多關(guān)于Java Springboot 的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
log4j2 自動(dòng)刪除過(guò)期日志文件的配置及實(shí)現(xiàn)原理
這篇文章主要介紹了log4j2 自動(dòng)刪除過(guò)期日志文件配置及實(shí)現(xiàn)原理解析,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-07-07
這篇文章主要為大家詳細(xì)為大家詳細(xì)介紹了JDBC如何獲取數(shù)據(jù)庫(kù)連接，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-08-08
這篇文章主要介紹了Java中的@PreAuthorize注解源碼解析,@PreAuthorize注解會(huì)在方法執(zhí)行前進(jìn)行權(quán)限驗(yàn)證,支持Spring?EL表達(dá)式,它是基于方法注解的權(quán)限解決方案,需要的朋友可以參考下2023-10-10
這篇文章主要介紹了JedisPool資源池優(yōu)化方法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-03-03
Java File類的簡(jiǎn)單使用教程（創(chuàng)建、刪除、遍歷與判斷是否存在等）
這篇文章主要給大家介紹了關(guān)于Java File類的簡(jiǎn)單使用（創(chuàng)建、刪除、遍歷與判斷是否存在等）的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
Java http請(qǐng)求封裝工具類代碼實(shí)例
這篇文章主要介紹了Java http請(qǐng)求封裝工具類代碼實(shí)例,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
零基礎(chǔ)搭建boot+MybatisPlus的詳細(xì)教程
這篇文章主要介紹了零基礎(chǔ)搭建boot+MybatisPlus,首先需要?jiǎng)?chuàng)建數(shù)據(jù)庫(kù)表和創(chuàng)建boot項(xiàng)目使用mybatisplus操作數(shù)據(jù)庫(kù)，本文通過(guò)示例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下2022-03-03
MyBatis使用@MapKey注解進(jìn)行連表查詢,返回一個(gè)Map集合,Map的key為每條記錄的主鍵或指定字段,value為該記錄的字段名稱和字段值2024-12-12
SpringBoot+Mybatis-Plus實(shí)現(xiàn)mysql讀寫(xiě)分離方案的示例代碼
這篇文章主要介紹了SpringBoot+Mybatis-Plus實(shí)現(xiàn)mysql讀寫(xiě)分離方案的示例代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
springboot 實(shí)現(xiàn)mqtt物聯(lián)網(wǎng)的示例代碼
這篇文章主要介紹了springboot 實(shí)現(xiàn)mqtt物聯(lián)網(wǎng),本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-03-03