Java+Springboot搭建一個在線網(wǎng)盤文件分享系統(tǒng)
前言
springboot+freemark+jpa+MySQL實現(xiàn)的在線網(wǎng)盤文件分享系統(tǒng),其功能跟百度網(wǎng)盤非常類似,普通用戶可以注冊登錄,注冊后默認分配1G的空間大小,登錄進去后可以新建文件夾、上傳各種類型的文件、文件移動、復(fù)制、下載、刪除、分享,分享分為私密分享和公開分享,還可以設(shè)置分享過期時間,打開分享鏈接后可以對文件進行查看、下載、保存到自己網(wǎng)盤等。超級管理員登錄后可以設(shè)置普通用戶的空間大小、角色類型、權(quán)限等。
本系統(tǒng)主要涉及的特色功能有:
1:系統(tǒng)在上傳文件時自動計算文件的md5值,并且檢查該值是否存在,若存在則文件不再上傳到服務(wù)器,直接進行引用原有的文件。
2:定時器定時檢查限時分享的文件是否過期。
3:網(wǎng)盤目錄結(jié)構(gòu)可無限制層級創(chuàng)建,保存移動復(fù)制文件時遞歸所選文件夾層級等等。
使用技術(shù):java+springboot+freemark+jpa+MySQL+maven
效果圖
?
?
主要代碼
管理員控制器:
/** * @ClassName: AdminController * @Description: 管理員控制器 **/ @Controller public class AdminController extends BaseController { private Logger logger = LogUtils.getInstance(AdminController.class); /** * @Description 前往用戶管理頁面 **/ @GetMapping("/manages-users") public String manageUsers(Map<String,Object> map,Integer cur){ if (loginUser.getRole() == 1){ //用于無訪問權(quán)限 logger.error("當前登錄用戶:"+loginUser.getUserName()+"無管理員權(quán)限!"); return "redirect:/error401Page"; } //獲取全部的用戶 Integer usersCount = userService.getUsersCount(); //獲取當前查詢的頁數(shù),如果為空,默認為0 cur = (cur == null || cur<0)?0:cur; //獲得統(tǒng)計信息 FileStoreStatistics statistics = myFileService.getCountStatistics(loginUser.getFileStoreId()); //分頁獲得20個用戶信息 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)限和倉庫大小成功!"); return "200"; }else { //更新失敗,返回500狀態(tài)碼 logger.error("修改用戶"+userService.queryById(uId).getUserName()+":的權(quán)限和倉庫大小失?。?); 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()); //刪除該用戶倉庫根目錄下的所有文件 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())); //刪除文件表對應(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){ //獲得當前文件夾下的所有子文件夾 List<FileFolder> folders = fileFolderService.getFileFolderByParentFolderId(folder.getFileFolderId()); //刪除當前文件夾的所有的文件 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()); } }
文件倉庫控制器:
/** * @ClassName: FileStoreController * @Description: 文件倉庫控制器 **/ @Controller public class FileStoreController extends BaseController { private Logger logger = LogUtils.getInstance(FileStoreController.class); /** * @Description 上傳臨時文件 **/ @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("臨時文件上傳失敗!文件名不符合規(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("臨時文件上傳成功!"+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()){ //文件不存在,開始生成二維碼并保存文件 OutputStream os = new FileOutputStream(f); QRCodeUtil.encode(url, "/static/img/logo.png", os, true); os.close(); } //異步刪除臨時文件 tempFileService.deleteById(tempFile.getFileId()); session.setAttribute("imgPath","user_img/"+id+".jpg"); session.setAttribute("url",url); session.setAttribute("msg","上傳成功,掃碼/訪問鏈接 即可下載!"); return "redirect:/temp-file"; } catch (Exception e) { e.printStackTrace(); } }else { logger.info("臨時文件數(shù)據(jù)庫寫入失敗!"+name); session.setAttribute("url","error"); session.setAttribute("msg", "服務(wù)器出錯了,臨時文件上傳失敗!"); } }else{ //上傳失敗 logger.info("臨時文件上傳失敗!"+name); session.setAttribute("url","error"); session.setAttribute("msg", "服務(wù)器出錯了,上傳失敗!"); } } catch (IOException e) { e.printStackTrace(); } return "redirect:/temp-file"; } /** * @Description 網(wǎng)盤的文件上傳 * @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("用戶沒有上傳文件的權(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(" ",""); //獲取當前目錄下的所有文件,用來判斷是否已經(jīng)存在 List<MyFile> myFiles = null; if (folderId == 0){ //當前目錄為根目錄 myFiles = myFileService.getRootFilesByFileStoreId(loginUser.getFileStoreId()); }else { //當前目錄為其他目錄 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("當前文件已存在!上傳失敗..."); 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); //是否倉庫放不下該文件 if(store.getCurrentSize()+sizeInt > store.getMaxSize()){ logger.error("上傳失敗!倉庫已滿。"); 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ù)庫文件表寫入數(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()); //更新倉庫表的當前大小 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)盤的文件下載 **/ @GetMapping("/downloadFile") public String downloadFile(@RequestParam Integer fId){ if (fileStoreService.getFileStoreByUserId(loginUser.getUserId()).getPermission() == 2){ logger.error("用戶沒有下載文件的權(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)碼一下,不然會出現(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())); //刪除文件表對應(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); //強制刪除 deleteFolderF(folder); return folder.getParentFolderId() == 0?"redirect:/files":"redirect:/files?fId="+folder.getParentFolderId(); } /** * @Description 迭代刪除文件夾里面的所有文件和子文件夾 **/ public void deleteFolderF(FileFolder folder){ //獲得當前文件夾下的所有子文件夾 List<FileFolder> folders = fileFolderService.getFileFolderByParentFolderId(folder.getFileFolderId()); //刪除當前文件夾的所有的文件 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()); //獲得當前目錄下的所有文件夾,檢查當前文件夾是否已經(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ù)庫寫入數(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ù)庫信息 FileFolder fileFolder = fileFolderService.getFileFolderByFileFolderId(folder.getFileFolderId()); fileFolder.setFileFolderName(folder.getFileFolderName()); //獲得當前目錄下的所有文件夾,用于檢查文件夾是否已經(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ù)庫寫入數(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()){ //文件不存在,開始生成二維碼并保存文件 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ù)錯誤"); 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 { //解決下載文件時 中文文件名亂碼問題 boolean isMSIE = isMSBrowser(request); if (isMSIE) { //IE瀏覽器的亂碼問題解決 fileNameTemp = URLEncoder.encode(fileNameTemp, "UTF-8"); } else { //萬能亂碼問題解決 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)碼一下,不然會出現(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ù)文件的后綴名獲得對應(yīng)的類型 * @return int 1:文本類型 2:圖像類型 3:視頻類型 4:音樂類型 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 正則驗證文件名是否合法 [漢字,字符,數(shù)字,下劃線,英文句號,橫線] **/ 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 判斷當前瀏覽器是否為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 免登陸用戶入口,用于本地開發(fā)測試,上線運營為了安全請刪除此方法 * @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"; } /** * 用于注冊流程,用戶名,密碼,郵箱等校驗工作由前端來完成 */ @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", "驗證碼錯誤"); 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("注冊用戶成功!當前注冊用戶" + user); logger.info("注冊倉庫成功!當前注冊倉庫" + store); } else { map.put("errorMsg", "服務(wù)器發(fā)生錯誤,注冊失敗"); 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 ? "該郵箱尚未注冊" : "密碼錯誤"; logger.info("登錄失??!請確認郵箱和密碼是否正確!"); //登錄失敗,將失敗信息返回前端渲染 map.put("errorMsg", errorMsg); return "index"; } } /** * @return void * @Description 向注冊郵箱發(fā)送驗證碼, 并驗證郵箱是否已使用 **/ @ResponseBody @RequestMapping("/sendCode") public String sendCode(String userName, String email, String password) { User userByEmail = userService.getUserByEmail(email); if (userByEmail != null) { logger.error("發(fā)送驗證碼失?。∴]箱已被注冊!"); return "exitEmail"; } logger.info("開始發(fā)送郵件.../n" + "獲取的到郵件發(fā)送對象為:" + mailSender); mailUtils = new MailUtils(mailSender); String code = "123456"; session.setAttribute(email + "_code", code); return "success"; } /** * @Description 請求QQ登錄 **/ @GetMapping("/loginByQQ") public void login() { response.setContentType("text/html;charset=utf-8"); try { response.sendRedirect(new Oauth().getAuthorizeURL(request)); logger.info("請求QQ登錄,開始跳轉(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("登錄失敗:沒有獲取到響應(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 去獲取當前用的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("注冊用戶成功!當前注冊用戶" + user); FileStore store = FileStore.builder().userId(user.getUserId()).build(); if (fileStoreService.addFileStore(store) == 1){ user.setFileStoreId(store.getFileStoreId()); userService.update(user); logger.info("注冊倉庫成功!當前注冊倉庫" + store); } } else { logger.error("注冊用戶失?。?); } }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("很抱歉,我們沒能正確獲取到您的信息,原因是: " + userInfoBean.getMsg()); } } } catch (QQConnectException e) { } finally { logger.error("登錄成功!"); } return "登錄失敗!請查看日志信息..."; } /** * @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登錄賬號 */ 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); //登錄, 如果采用默認端口,可以使用ftp.connect(host)的方式直接連接FTP服務(wù)器 ftp.login(USERNAME, PASSWORD); ftp.setBufferSize(10240); //設(shè)置傳輸超時時間為60秒 ftp.setDataTimeout(600000); //連接超時為60秒 ftp.setConnectTimeout(600000); //FTP以二進制形式傳輸 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è)置上傳文件的類型為二進制類型 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ù)器上的相對路徑 * @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搭建一個在線網(wǎng)盤文件分享系統(tǒng)的詳細內(nèi)容,更多關(guān)于Java Springboot 的資料請關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
java使用Apache工具集實現(xiàn)ftp文件傳輸代碼詳解
這篇文章主要介紹了java使用Apache工具集實現(xiàn)ftp文件傳輸代碼詳解,分享了詳細連接ftp server和上傳文件,下載文件的代碼,以及結(jié)果展示,具有一定借鑒價值,需要的朋友可以參考下。2017-12-12WMTS中TileMatrix與ScaleDenominator淺析
這篇文章主要為大家介紹了WMTS中TileMatrix與ScaleDenominator淺析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-03-03詳解Java設(shè)計模式編程中的Flyweight享元模式的開發(fā)結(jié)構(gòu)
這篇文章主要介紹了Java設(shè)計模式編程中的Flyweight享元模式的開發(fā)結(jié)構(gòu),享元模式能夠最大限度地重用現(xiàn)有的同類對象,需要的朋友可以參考下2016-04-04Springboot集成RabbitMQ死信隊列的實現(xiàn)
在大多數(shù)的MQ中間件中,都有死信隊列的概念。本文主要介紹了Springboot集成RabbitMQ死信隊列的實現(xiàn),具有一定的參考價值,感興趣的小伙伴們可以參考一下2021-09-09HttpClient的RedirectStrategy重定向處理核心機制
這篇文章主要為大家介紹了HttpClient的RedirectStrategy重定向處理核心機制源碼解讀,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-10-10java常用數(shù)據(jù)流應(yīng)用實例解析
這篇文章主要介紹了java常用數(shù)據(jù)流應(yīng)用實例解析,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-12-12