目錄
- 5.8 匯入其他介面代碼
- 5.8.1 介面匯入-分頁查詢附件
- 5.8.2 介面匯入-根據業務型別/業務id查詢附件
- 5.9 匯入網盤服務介面
- 5.9.1 匯入FileController
- 5.9.2 匯入StatisticsController
- 5.9.3 匯入FileRestManager
- 5.9.4 匯入FileService
- 5.9.5 匯入FileServiceImpl
- 5.9.6 擴展FileMapper介面方法
5.8 匯入其他介面代碼
第2-1-2章 傳統方式安裝FastDFS-附FastDFS常用命令
第2-1-3章 docker-compose安裝FastDFS,實作檔案存盤服務
第2-1-5章 docker安裝MinIO實作檔案存盤服務-springboot整合minio-minio全網最全的資料
全套代碼及資料全部完整提供，點此處下載
5.8.1 介面匯入-分頁查詢附件
介面檔案：
AttachmentController代碼：
/**
* 分頁查詢附件
*
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/分頁查詢附件", notes = "分頁查詢附件")
@ApiImplicitParams({
@ApiImplicitParam(name = "current", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/當前頁", dataType = "long", paramType = "query", defaultValue = "https://www.cnblogs.com/gitBook/archive/2022/11/20/1"),
@ApiImplicitParam(name = "size", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/每頁顯示幾條", dataType = "long", paramType = "query", defaultValue = "https://www.cnblogs.com/gitBook/archive/2022/11/20/10"),
})
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/page")
public R<IPage<Attachment>> page(FilePageReqDTO data) {
Page<Attachment> page = getPage();
attachmentService.page(page, data);
return success(page);
}
AttachmentService介面：
/**
* 查詢附件分頁資料
*
* @param page
* @param data
* @return
*/
IPage<Attachment> page(Page<Attachment> page, FilePageReqDTO data);
AttachmentServiceImpl類：
/**
* 查詢附件分頁資料
*
* @param page
* @param data
* @return
*/
public IPage<Attachment> page(Page<Attachment> page, FilePageReqDTO data) {
Attachment attachment = dozer.map(data, Attachment.class);
// ${ew.customSqlSegment} 語法一定要手動eq like 等 不能用lbQ!
LbqWrapper<Attachment> wrapper = Wraps.<Attachment>lbQ()
.like(Attachment::getSubmittedFileName, attachment.getSubmittedFileName())
.like(Attachment::getBizType, attachment.getBizType())
.like(Attachment::getBizId, attachment.getBizId())
.eq(Attachment::getDataType, attachment.getDataType())
.orderByDesc(Attachment::getId);
return baseMapper.page(page, wrapper);
}
5.8.2 介面匯入-根據業務型別/業務id查詢附件
介面檔案：
AttachmentController代碼：
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/查詢附件", notes = "查詢附件")
@ApiResponses(
@ApiResponse(code = 60103, message = "檔案id為空")
)
@GetMapping
public R<List<AttachmentResultDTO>> findAttachment(@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/bizTypes", required = false) String[] bizTypes,
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/bizIds", required = false) String[] bizIds) {
//不能同時為空
BizAssert.isTrue(!(ArrayUtils.isEmpty(bizTypes) && ArrayUtils.isEmpty(bizIds)), BASE_VALID_PARAM.build("業務型別不能為空"));
return success(attachmentService.find(bizTypes, bizIds));
}
AttachmentService介面：
/**
* 根據業務型別和業務id查詢附件
*
* @param bizTypes
* @param bizIds
* @return
*/
List<AttachmentResultDTO> find(String[] bizTypes, String[] bizIds);
AttachmentServiceImpl類：
/**
* 根據業務型別和業務id查詢附件
*
* @param bizTypes
* @param bizIds
* @return
*/
public List<AttachmentResultDTO> find(String[] bizTypes, String[] bizIds) {
return baseMapper.find(bizTypes, bizIds);
}
5.9 匯入網盤服務介面
前面我們已經完成了檔案服務中的附件服務相關介面的開發，附件服務最終是將上傳的檔案資訊保存在pd_attachment表中，
本小節要完成的是檔案服務中的網盤服務功能，此功能最終是將上傳的檔案資訊保存在pd_file表中，網盤服務和附件服務非常類似，只是多了一個檔案夾的概念，
5.9.1 匯入FileController
package com.itheima.pinda.file.controller;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.itheima.pinda.base.BaseController;
import com.itheima.pinda.base.R;
import com.itheima.pinda.dozer.DozerUtils;
import com.itheima.pinda.file.dto.FilePageReqDTO;
import com.itheima.pinda.file.dto.FileUpdateDTO;
import com.itheima.pinda.file.dto.FolderDTO;
import com.itheima.pinda.file.dto.FolderSaveDTO;
import com.itheima.pinda.file.entity.File;
import com.itheima.pinda.file.manager.FileRestManager;
import com.itheima.pinda.file.service.FileService;
import com.itheima.pinda.log.annotation.SysLog;
import io.swagger.annotations.*;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import javax.validation.constraints.NotNull;
/**
* 檔案前端控制器
*/
@Validated
@RestController
@RequestMapping("/file")
@Slf4j
@Api(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/檔案表", tags = "檔案表")
public class FileController extends BaseController {
@Autowired
private FileService fileService;
@Autowired
private FileRestManager fileRestManager;
@Autowired
private DozerUtils dozerUtils;
/**
* 查詢單個檔案資訊
*
* @param id
* @return
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/查詢檔案", notes = "查詢檔案")
@GetMapping
public R<File> get(@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/id") Long id) {
File file = fileService.getById(id);
if (file != null && file.getIsDelete()) {
return success(null);
}
return success(file);
}
/**
* 獲取檔案分頁
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/分頁查詢檔案", notes = "獲取檔案分頁")
@ApiImplicitParams({
@ApiImplicitParam(name = "current", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/當前頁", dataType = "long", paramType = "query", defaultValue = "https://www.cnblogs.com/gitBook/archive/2022/11/20/1"),
@ApiImplicitParam(name = "size", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/每頁顯示幾條", dataType = "long", paramType = "query", defaultValue = "https://www.cnblogs.com/gitBook/archive/2022/11/20/10"),
})
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/page")
public R<IPage<File>> page(FilePageReqDTO data) {
return success(fileRestManager.page(getPage(), data));
}
/**
* 上傳檔案
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/上傳檔案", notes = "上傳檔案 ")
@ApiResponses({
@ApiResponse(code = 60102, message = "檔案夾為空"),
})
@ApiImplicitParams({
@ApiImplicitParam(name = "source", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/檔案來源", dataType = "String", paramType = "query"),
@ApiImplicitParam(name = "userId", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/用戶id", dataType = "long", paramType = "query"),
@ApiImplicitParam(name = "folderId", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/檔案夾id", dataType = "long", paramType = "query"),
@ApiImplicitParam(name = "file", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/附件", dataType = "MultipartFile", allowMultiple = true, required = true),
})
@RequestMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/upload", method = RequestMethod.POST)
public R<File> upload(
@NotNull(message = "檔案夾不能為空")
@RequestParam(name = "source", defaultValue = "https://www.cnblogs.com/gitBook/archive/2022/11/20/inner") String source,
@RequestParam(name = "userId", required = false) Long userId,
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/folderId") Long folderId,
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/file") MultipartFile simpleFile) {
//1，先將檔案存在本地,并且生成檔案名
log.info("contentType={}, name={} , sfname={}", simpleFile.getContentType(), simpleFile.getName(), simpleFile.getOriginalFilename());
// 忽略路徑欄位,只處理檔案型別
if (simpleFile.getContentType() == null) {
return fail("檔案為空");
}
File file = fileService.upload(simpleFile, folderId, source, userId);
return success(file);
}
/**
* 保存檔案夾
*/
@ApiResponses({
@ApiResponse(code = 60000, message = "檔案夾為空"),
@ApiResponse(code = 60001, message = "檔案夾名稱為空"),
@ApiResponse(code = 60002, message = "父檔案夾為空"),
})
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "https://www.cnblogs.com/gitBook/archive/2022/11/20/新增檔案夾", notes = "新增檔案夾")
@RequestMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "", method = RequestMethod.POST)
public R<FolderDTO> saveFolder(@Valid @RequestBody FolderSaveDTO folderSaveDto) {
//2，獲取身份
FolderDTO folder = fileService.saveFolder(folderSaveDto);
return success(folder);
}
/**
* 修改檔案、檔案夾資訊
*
* @param fileUpdateDTO
* @return
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "修改檔案/檔案夾名稱", notes = "修改檔案/檔案夾名稱")
@ApiResponses({
@ApiResponse(code = 60100, message = "檔案為空"),
})
@RequestMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "", method = RequestMethod.PUT)
public R<Boolean> update(@Valid @RequestBody FileUpdateDTO fileUpdateDTO) {
// 判斷檔案名是否有 后綴
if (StringUtils.isNotEmpty(fileUpdateDTO.getSubmittedFileName())) {
File oldFile = fileService.getById(fileUpdateDTO.getId());
if (oldFile.getExt() != null && !fileUpdateDTO.getSubmittedFileName().endsWith(oldFile.getExt())) {
fileUpdateDTO.setSubmittedFileName(fileUpdateDTO.getSubmittedFileName() + "." + oldFile.getExt());
}
}
File file = dozerUtils.map2(fileUpdateDTO, File.class);
fileService.updateById(file);
return success(true);
}
/**
* 根據Ids進行檔案洗掉
*
* @param ids
* @return
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "根據Ids進行檔案洗掉", notes = "根據Ids進行檔案洗掉 ")
@DeleteMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "/ids")
public R<Boolean> removeList(@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "ids[]") Long[] ids) {
Long userId = getUserId();
return success(fileService.removeList(userId, ids));
}
/**
* 下載一個檔案或多個檔案打包下載
*
* @param ids
* @param response
* @throws Exception
*/
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "下載一個檔案或多個檔案打包下載", notes = "下載一個檔案或多個檔案打包下載")
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "/download", produces = "application/octet-stream")
public void download(
@ApiParam(name = "ids[]", valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "檔案id 陣列")
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "ids[]") Long[] ids,
HttpServletRequest request, HttpServletResponse response) throws Exception {
fileRestManager.download(request, response, ids, null);
}
}
5.9.2 匯入StatisticsController
package com.itheima.pinda.file.controller;
import com.itheima.pinda.base.BaseController;
import com.itheima.pinda.base.R;
import com.itheima.pinda.file.domain.FileStatisticsDO;
import com.itheima.pinda.file.dto.FileOverviewDTO;
import com.itheima.pinda.file.dto.FileStatisticsAllDTO;
import com.itheima.pinda.file.service.FileService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.time.LocalDateTime;
import java.util.List;
/**
* 檔案統計介面
*/
@Slf4j
@RestController
@RequestMapping("/statistics")
@Api(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "Statistics", tags = "統計介面")
public class StatisticsController extends BaseController {
@Autowired
private FileService fileService;
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "云盤首頁資料概覽", notes = "云盤首頁資料概覽")
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "/overview")
public R<FileOverviewDTO> overview(@RequestParam(name = "userId", required = false) Long userId) {
return success(fileService.findOverview(userId, null, null));
}
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "按照型別，統計各種型別的 大小和數量", notes = "按照型別，統計當前登錄人各種型別的大小和數量")
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "/type")
public R<List<FileStatisticsDO>> findAllByDataType(@RequestParam(name = "userId", required = false) Long userId) {
return success(fileService.findAllByDataType(userId));
}
@ApiOperation(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "按照時間統計各種型別的檔案的數量和大小", notes = "按照時間統計各種型別的檔案的數量和大小 不指定時間，默認查詢一個月")
@GetMapping(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "")
public R<FileStatisticsAllDTO> findNumAndSizeToTypeByDate(@RequestParam(name = "userId", required = false) Long userId,
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "startTime", required = false) LocalDateTime startTime,
@RequestParam(valuehttps://www.cnblogs.com/gitBook/archive/2022/11/20/= "endTime", required = false) LocalDateTime endTime) {
return success(fileService.findNumAndSizeToTypeByDate(userId, startTime, endTime));
}
}
5.9.3 匯入FileRestManager
package com.itheima.pinda.file.manager;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.itheima.pinda.context.BaseContextHandler;
import com.itheima.pinda.file.constant.FileConstants;
import com.itheima.pinda.file.dto.FilePageReqDTO;
import com.itheima.pinda.file.entity.File;
import com.itheima.pinda.file.service.FileService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import static com.itheima.pinda.utils.StrPool.DEF_PARENT_ID;
/**
* 檔案 公共代碼 管理類
*/
@Component
public class FileRestManager {
@Autowired
private FileService fileService;
public IPage<File> page(IPage<File> page, FilePageReqDTO filePageReq) {
//型別和檔案夾id同時為null時， 表示查詢 全部檔案
if (filePageReq.getFolderId() == null && filePageReq.getDataType() == null) {
filePageReq.setFolderId(DEF_PARENT_ID);
}
QueryWrapper<File> query = new QueryWrapper<>();
LambdaQueryWrapper<File> lambdaQuery = query.lambda()
.eq(File::getIsDelete, false)
.eq(filePageReq.getDataType() != null, File::getDataType, filePageReq.getDataType())
.eq(filePageReq.getFolderId() != null, File::getFolderId, filePageReq.getFolderId())
.like(StringUtils.isNotEmpty(filePageReq.getSubmittedFileName()), File::getSubmittedFileName, filePageReq.getSubmittedFileName());
query.orderByDesc(String.format("case when %s='DIR' THEN 1 else 0 end", FileConstants.DATA_TYPE));
lambdaQuery.orderByDesc(File::getCreateTime);
fileService.page(page, lambdaQuery);
return page;
}
public void download(HttpServletRequest request, HttpServletResponse response, Long[] ids, Long userId) throws Exception {
userId = userId == null || userId <= 0 ? BaseContextHandler.getUserId() : userId;
fileService.download(request, response, ids, userId);
}
}
5.9.4 匯入FileService
package com.itheima.pinda.file.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.itheima.pinda.file.domain.FileAttrDO;
import com.itheima.pinda.file.domain.FileStatisticsDO;
import com.itheima.pinda.file.dto.FileOverviewDTO;
import com.itheima.pinda.file.dto.FileStatisticsAllDTO;
import com.itheima.pinda.file.dto.FolderDTO;
import com.itheima.pinda.file.dto.FolderSaveDTO;
import com.itheima.pinda.file.entity.File;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.time.LocalDateTime;
import java.util.List;
/**
* 業務介面
* 檔案表
*
*/
public interface FileService extends IService<File> {
/**
* 保存檔案夾
*
* @param folderSaveDto 檔案夾
* @return
*/
FolderDTO saveFolder(FolderSaveDTO folderSaveDto);
/**
* 根據檔案id下載檔案，并統計下載次數
*
* @param request 請求
* @param response 回應
* @param ids 檔案id集合
* @param userId 用戶id
* @throws Exception
*/
void download(HttpServletRequest request, HttpServletResponse response,
Long[] ids, Long userId) throws Exception;
/**
* 根據檔案id和用戶id 洗掉檔案或者檔案夾
*
* @param userId 用戶id
* @param ids 檔案id集合
* @return
*/
Boolean removeList(Long userId, Long[] ids);
/**
* 根據檔案夾id查詢
*
* @param folderId
* @return
*/
FileAttrDO getFileAttrDo(Long folderId);
/**
* 檔案上傳
*
* @param simpleFile 檔案
* @param folderId 檔案夾id
* @return
*/
File upload(MultipartFile simpleFile, Long folderId);
/**
* 檔案上傳
*
* @param simpleFile
* @param folderId
* @param source
* @param userId
* @return
*/
File upload(MultipartFile simpleFile, Long folderId, String source, Long userId);
/**
* 首頁概覽
*
* @param userId
* @param startTime
* @param endTime
* @return
*/
FileOverviewDTO findOverview(Long userId, LocalDateTime startTime, LocalDateTime endTime);
/**
* 首頁個人檔案發展概覽
*
* @param userId
* @param startTime
* @param endTime
* @return
*/
FileStatisticsAllDTO findAllByDate(Long userId, LocalDateTime startTime, LocalDateTime endTime);
/**
* 按照 資料型別分類查詢 當前人的所有檔案的數量和大小
*
* @param userId
* @return
*/
List<FileStatisticsDO> findAllByDataType(Long userId);
/**
* 查詢下載排行前20的檔案
*
* @param userId
* @return
*/
List<FileStatisticsDO> downTop20(Long userId);
/**
* 根據日期查詢，特定型別的數量和大小
*
* @param userId
* @param startTime
* @param endTime
* @return
*/
FileStatisticsAllDTO findNumAndSizeToTypeByDate(Long userId, LocalDateTime startTime, LocalDateTime endTime);
/**
* 根據日期查詢下載大小
*
* @param userId
* @param startTime
* @param endTime
* @return
*/
FileStatisticsAllDTO findDownSizeByDate(Long userId, LocalDateTime startTime,
LocalDateTime endTime);
}
5.9.5 匯入FileServiceImpl
package com.itheima.pinda.file.service.impl;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.pinda.database.mybatis.conditions.Wraps;
import com.itheima.pinda.database.mybatis.conditions.update.LbuWrapper;
import com.itheima.pinda.dozer.DozerUtils;
import com.itheima.pinda.file.biz.FileBiz;
import com.itheima.pinda.file.dao.FileMapper;
import com.itheima.pinda.file.domain.FileAttrDO;
import com.itheima.pinda.file.domain.FileDO;
import com.itheima.pinda.file.domain.FileDeleteDO;
import com.itheima.pinda.file.domain.FileStatisticsDO;
import com.itheima.pinda.file.dto.FileOverviewDTO;
import com.itheima.pinda.file.dto.FileStatisticsAllDTO;
import com.itheima.pinda.file.dto.FolderDTO;
import com.itheima.pinda.file.dto.FolderSaveDTO;
import com.itheima.pinda.file.entity.File;
import com.itheima.pinda.file.enumeration.DataType;
import com.itheima.pinda.file.enumeration.IconType;
import com.itheima.pinda.file.service.FileService;
import com.itheima.pinda.file.strategy.FileStrategy;
import com.itheima.pinda.utils.BizAssert;
import com.itheima.pinda.utils.DateUtils;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static com.itheima.pinda.exception.code.ExceptionCode.BASE_VALID_PARAM;
import static com.itheima.pinda.utils.StrPool.DEF_PARENT_ID;
import static com.itheima.pinda.utils.StrPool.DEF_ROOT_PATH;
import static java.util.stream.Collectors.groupingBy;
/**
* 業務實作類
* 檔案表
*/
@Slf4j
@Service
public class FileServiceImpl extends ServiceImpl<FileMapper, File> implements FileService {
@Autowired
private DozerUtils dozerUtils;
@Autowired
private FileBiz fileBiz;
@Resource
private FileStrategy fileStrategy;
@Override
public File upload(MultipartFile simpleFile, Long folderId) {
FileAttrDO fileAttrDO = this.getFileAttrDo(folderId);
String treePath = fileAttrDO.getTreePath();
String folderName = fileAttrDO.getFolderName();
Integer grade = fileAttrDO.getGrade();
File file = fileStrategy.upload(simpleFile);
file.setFolderId(folderId);
file.setFolderName(folderName);
file.setGrade(grade);
file.setTreePath(treePath);
super.save(file);
return file;
}
@Override
public File upload(MultipartFile simpleFile, Long folderId, String source, Long userId) {
FileAttrDO fileAttrDO = this.getFileAttrDo(folderId);
String treePath = fileAttrDO.getTreePath();
String folderName = fileAttrDO.getFolderName();
Integer grade = fileAttrDO.getGrade();
File file = fileStrategy.upload(simpleFile);
file.setFolderId(folderId);
file.setFolderName(folderName);
file.setGrade(grade);
file.setTreePath(treePath);
file.setSource(source);
if (userId != null) {
file.setCreateUser(userId);
}
super.save(file);
return file;
}
@Override
public FileAttrDO getFileAttrDo(Long folderId) {
String treePath = DEF_ROOT_PATH;
String folderName = "";
Integer grade = 1;
if (folderId == null || folderId <= 0) {
return new FileAttrDO(treePath, grade, folderName, DEF_PARENT_ID);
}
File folder = this.getById(folderId);
if (folder != null && !folder.getIsDelete() && DataType.DIR.eq(folder.getDataType())) {
folderName = folder.getSubmittedFileName();
treePath = StringUtils.join(folder.getTreePath(), folder.getId(), DEF_ROOT_PATH);
grade = folder.getGrade() + 1;
}
BizAssert.isTrue(grade <= 10, BASE_VALID_PARAM.build("檔案夾層級不能超過10層"));
return new FileAttrDO(treePath, grade, folderName, folderId);
}
@Override
public FolderDTO saveFolder(FolderSaveDTO folderSaveDto) {
File folder = dozerUtils.map2(folderSaveDto, File.class);
if (folderSaveDto.getFolderId() == null || folderSaveDto.getFolderId() <= 0) {
folder.setFolderId(DEF_PARENT_ID);
folder.setTreePath(DEF_ROOT_PATH);
folder.setGrade(1);
} else {
File parent = super.getById(folderSaveDto.getFolderId());
BizAssert.notNull(parent, BASE_VALID_PARAM.build("父檔案夾不能為空"));
BizAssert.isFalse(parent.getIsDelete(), BASE_VALID_PARAM.build("父檔案夾已經被洗掉"));
BizAssert.equals(DataType.DIR.name(), parent.getDataType().name(), BASE_VALID_PARAM.build("父檔案夾不存在"));
BizAssert.isTrue(parent.getGrade() < 10, BASE_VALID_PARAM.build("檔案夾層級不能超過10層"));
folder.setFolderName(parent.getSubmittedFileName());
folder.setTreePath(StringUtils.join(parent.getTreePath(), parent.getId(), DEF_ROOT_PATH));
folder.setGrade(parent.getGrade() + 1);
}
if (folderSaveDto.getOrderNum() == null) {
folderSaveDto.setOrderNum(0);
}
folder.setIsDelete(false);
folder.setDataType(DataType.DIR);
folder.setIcon(IconType.DIR.getIcon());
setDate(folder);
super.save(folder);
return dozerUtils.map2(folder, FolderDTO.class);
}
private void setDate(File file) {
LocalDateTime now = LocalDateTime.now();
file.setCreateMonth(DateUtils.formatAsYearMonthEn(now))
.setCreateWeek(DateUtils.formatAsYearWeekEn(now))
.setCreateDay(DateUtils.formatAsDateEn(now));
}
public boolean removeFile(Long[] ids, Long userId) {
LbuWrapper<File> lambdaUpdate =
Wraps.<File>lbU()
.in(File::getId, ids)
.eq(File::getCreateUser, userId);
File file = File.builder().isDelete(Boolean.TRUE).build();
return super.update(file, lambdaUpdate);
}
@Override
public Boolean removeList(Long userId, Long[] ids) {
if (ArrayUtils.isEmpty(ids)) {
return Boolean.TRUE;
}
List<File> list = super.list(Wrappers.<File>lambdaQuery().in(File::getId, ids));
if (list.isEmpty()) {
return true;
}
super.removeByIds(Arrays.asList(ids));
fileStrategy.delete(list.stream().map((fi) -> FileDeleteDO.builder()
.relativePath(fi.getRelativePath())
.fileName(fi.getFilename())
.group(fi.getGroup())
.path(fi.getPath())
.file(false)
.build())
.collect(Collectors.toList()));
return true;
}
@Override
public void download(HttpServletRequest request, HttpServletResponse response, Long[] ids, Long userId) throws Exception {
if (ids == null || ids.length == 0) {
return;
}
List<File> list = (List<File>) super.listByIds(Arrays.asList(ids));
if (list == null || list.size() == 0) {
return;
}
List<FileDO> listDo = list.stream().map((file) ->
FileDO.builder()
.dataType(file.getDataType())
.size(file.getSize())
.submittedFileName(file.getSubmittedFileName())
.url(file.getUrl())
.build())
.collect(Collectors.toList());
fileBiz.down(listDo, request, response);
}
@Override
public FileOverviewDTO findOverview(Long userId, LocalDateTime startTime, LocalDateTime endTime) {
InnerQueryDate innerQueryDate = new InnerQueryDate(userId, startTime, endTime).invoke();
startTime = innerQueryDate.getStartTime();
endTime = innerQueryDate.getEndTime();
List<FileStatisticsDO> list = baseMapper.findNumAndSizeByUserId(userId, null, "ALL", startTime, endTime);
FileOverviewDTO.FileOverviewDTOBuilder builder = FileOverviewDTO.myBuilder();
long allSize = 0L;
int allNum = 0;
for (FileStatisticsDO fs : list) {
allSize += fs.getSize();
allNum += fs.getNum();
switch (fs.getDataType()) {
case DIR:
builder.dirNum(fs.getNum());
break;
case IMAGE:
builder.imgNum(fs.getNum());
break;
case VIDEO:
builder.videoNum(fs.getNum());
break;
case DOC:
builder.docNum(fs.getNum());
break;
case AUDIO:
builder.audioNum(fs.getNum());
break;
case OTHER:
builder.otherNum(fs.getNum());
break;
default:
break;
}
}
builder.allFileNum(allNum).allFileSize(allSize);
return builder.build();
}
@Override
public FileStatisticsAllDTO findAllByDate(Long userId, LocalDateTime startTime, LocalDateTime endTime) {
InnerQueryDate innerQueryDate = new InnerQueryDate(userId, startTime, endTime).invoke();
startTime = innerQueryDate.getStartTime();
endTime = innerQueryDate.getEndTime();
List<String> dateList = innerQueryDate.getDateList();
String dateType = innerQueryDate.getDateType();
//不完整的資料
List<FileStatisticsDO> list = baseMapper.findNumAndSizeByUserId(userId, dateType, null, startTime, endTime);
//按月份分類
Map<String, List<FileStatisticsDO>> map = list.stream().collect(groupingBy(FileStatisticsDO::getDateType));
List<Long> sizeList = new ArrayList<>();
List<Integer> numList = new ArrayList<>();
dateList.forEach((date) -> {
if (map.containsKey(date)) {
List<FileStatisticsDO> subList = map.get(date);
Long size = subList.stream().mapToLong(FileStatisticsDO::getSize).sum();
Integer num = subList.stream().filter((fs) -> !DataType.DIR.eq(fs.getDataType()))
.mapToInt(FileStatisticsDO::getNum).sum();
sizeList.add(size);
numList.add(num);
} else {
sizeList.add(0L);
numList.add(0);
}
});
return FileStatisticsAllDTO.builder().dateList(dateList).numList(numList).sizeList(sizeList).build();
}
@Override
public List<FileStatisticsDO> findAllByDataType(Long userId) {
List<DataType> dataTypes = Arrays.asList(DataType.values());
List<FileStatisticsDO> list = baseMapper.findNumAndSizeByUserId(userId, null, "ALL", null, null);
Map<DataType, List<FileStatisticsDO>> map = list.stream().collect(groupingBy(FileStatisticsDO::getDataType));
return dataTypes.stream().map((type) -> {
FileStatisticsDO fs = null;
if (map.containsKey(type)) {
fs = map.get(type).get(0);
} else {
fs = FileStatisticsDO.builder().dataType(type).size(0L).num(0).build();
}
return fs;
}).collect(Collectors.toList());
}
@Override
public List<FileStatisticsDO> downTop20(Long userId) {
return baseMapper.findDownTop20(userId);
}
@Override
public FileStatisticsAllDTO findNumAndSizeToTypeByDate(Long userId, LocalDateTime startTime, LocalDateTime endTime) {
return common(userId, startTime, endTime,
(qd) -> baseMapper.findNumAndSizeByUserId(qd.getUserId(), qd.getDateType(), "ALL", qd.getStartTime(), qd.getEndTime()));
}
@Override
public FileStatisticsAllDTO findDownSizeByDate(Long userId, LocalDateTime startTime,
LocalDateTime endTime) {
return common(userId, startTime, endTime,
(qd) -> baseMapper.findDownSizeByDate(qd.getUserId(), qd.getDateType(), qd.getStartTime(), qd.getEndTime()));
}
/**
* 抽取公共查詢公共代碼
*
* @param userId 用戶id
* @param startTime 開始時間
* @param endTime 結束時間
* @param function 回呼函式
* @return
*/
private FileStatisticsAllDTO common(Long userId, LocalDateTime startTime, LocalDateTime endTime, Function<InnerQueryDate, List<FileStatisticsDO>> function) {
InnerQueryDate innerQueryDate = new InnerQueryDate(userId, startTime, endTime).invoke();
List<String> dateList = innerQueryDate.getDateList();
List<FileStatisticsDO> list = function.apply(innerQueryDate);
//按月份分類
Map<String, List<FileStatisticsDO>> map = list.stream().collect(groupingBy(FileStatisticsDO::getDateType));
List<Long> sizeList = new ArrayList<>(dateList.size());
List<Integer> numList = new ArrayList<>(dateList.size());
List<Integer> dirNumList = new ArrayList<>(dateList.size());
List<Long> imgSizeList = new ArrayList<>(dateList.size());
List<Integer> imgNumList = new ArrayList<>(dateList.size());
List<Long> videoSizeList = new ArrayList<>(dateList.size());
List<Integer> videoNumList = new ArrayList<>(dateList.size());
List<Long> audioSizeList = new ArrayList<>(dateList.size());
List<Integer> audioNumList = new ArrayList<>(dateList.size());
List<Long> docSizeList = new ArrayList<>(dateList.size());
List<Integer> docNumList = new ArrayList<>(dateList.size());
List<Long> otherSizeList = new ArrayList<>(dateList.size());
List<Integer> otherNumList = new ArrayList<>(dateList.size());
dateList.forEach((date) -> {
if (map.containsKey(date)) {
List<FileStatisticsDO> subList = map.get(date);
Function<DataType, Stream<FileStatisticsDO>> stream = (dataType) -> subList.stream().filter((fs) -> !dataType.eq(fs.getDataType()));
Long size = stream.apply(DataType.DIR).mapToLong(FileStatisticsDO::getSize).sum();
Integer num = stream.apply(DataType.DIR).mapToInt(FileStatisticsDO::getNum).sum();
sizeList.add(size);
numList.add(num);
Integer dirNum = subList.stream().filter((fs) -> DataType.DIR.eq(fs.getDataType()))
.mapToInt(FileStatisticsDO::getNum).sum();
dirNumList.add(dirNum);
add(imgSizeList, imgNumList, subList, DataType.IMAGE);
add(videoSizeList, videoNumList, subList, DataType.VIDEO);
add(audioSizeList, audioNumList, subList, DataType.AUDIO);
add(docSizeList, docNumList, subList, DataType.DOC);
add(otherSizeList, otherNumList, subList, DataType.OTHER);
} else {
sizeList.add(0L);
numList.add(0);
dirNumList.add(0);
imgSizeList.add(0L);
imgNumList.add(0);
videoSizeList.add(0L);
videoNumList.add(0);
audioSizeList.add(0L);
audioNumList.add(0);
docSizeList.add(0L);
docNumList.add(0);
otherSizeList.add(0L);
otherNumList.add(0);
}
});
return FileStatisticsAllDTO.builder()
.dateList(dateList)
.numList(numList).sizeList(sizeList)
.dirNumList(dirNumList)
.imgNumList(imgNumList).imgSizeList(imgSizeList)
.videoNumList(videoNumList).videoSizeList(videoSizeList)
.audioNumList(audioNumList).audioSizeList(audioSizeList)
.docNumList(docNumList).docSizeList(docSizeList)
.otherNumList(otherNumList).otherSizeList(otherSizeList)
.build();
}
private void add(List<Long> sizeList, List<Integer> numList, List<FileStatisticsDO> subList, DataType dt) {
Function<DataType, Stream<FileStatisticsDO>> stream =
dataType -> subList.stream().filter(fs -> dataType.eq(fs.getDataType()));
Long size = stream.apply(dt).mapToLong(FileStatisticsDO::getSize).sum();
Integer num = stream.apply(dt).mapToInt(FileStatisticsDO::getNum).sum();
sizeList.add(size);
numList.add(num);
}
@Getter
private static class InnerQueryDate {
private LocalDateTime startTime;
private LocalDateTime endTime;
private List<String> dateList;
private String dateType;
private Long userId;
public InnerQueryDate(Long userId, LocalDateTime startTime, LocalDateTime endTime) {
this.userId = userId;
this.startTime = startTime;
this.endTime = endTime;
}
public InnerQueryDate invoke() {
if (startTime == null) {
startTime = LocalDateTime.now().plusDays(-9);
}
if (endTime == null) {
endTime = LocalDateTime.now();
}
endTime = LocalDateTime.of(endTime.toLocalDate(), LocalTime.MAX);
dateList = new ArrayList<>();
dateType = DateUtils.calculationEn(startTime, endTime, dateList);
return this;
}
}
}
5.9.6 擴展FileMapper介面方法
/**
* 查詢下次次數前20的檔案
*
* @param userId
* @return
*/
List<FileStatisticsDO> findDownTop20(@Param("userId") Long userId);
/**
* 統計時間區間內檔案的下次次數和大小
*
* @param userId
* @param dateType 日期型別 {MONTH:按月;WEEK:按周;DAY:按日} 來統計
* @param startTime
* @param endTime
* @return
*/
List<FileStatisticsDO> findDownSizeByDate(@Param("userId") Long userId,
@Param("dateType") String dateType,
@Param("startTime") LocalDateTime startTime,
@Param("endTime") LocalDateTime endTime);
第2-1-2章 傳統方式安裝FastDFS-附FastDFS常用命令
第2-1-3章 docker-compose安裝FastDFS,實作檔案存盤服務
第2-1-5章 docker安裝MinIO實作檔案存盤服務-springboot整合minio-minio全網最全的資料
全套代碼及資料全部完整提供，點此處下載
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/536213.html
標籤：其他
上一篇：python中的函式（1）