package org.sxkj.resource.controller;
|
|
|
import com.aliyuncs.utils.StringUtils;
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiModelProperty;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springblade.core.tool.utils.StringUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.sxkj.common.func.Streams;
|
import org.sxkj.common.model.PaginationData;
|
import org.sxkj.common.model.ResponseResult;
|
import org.sxkj.common.utils.FileTypeUtils;
|
import org.sxkj.common.utils.HeaderUtils;
|
import org.sxkj.resource.entity.MediaFileEntity;
|
import org.sxkj.resource.model.FileUploadDTO;
|
import org.sxkj.resource.model.MapKeyConst;
|
import org.sxkj.resource.model.MediaJobDTO;
|
import org.sxkj.resource.model.SearchMediaParam;
|
import org.sxkj.resource.service.IFileService;
|
import org.sxkj.resource.vo.MediaFileCountVO;
|
import org.sxkj.resource.vo.MediaFileVO;
|
import org.sxkj.resource.vo.MediaTimeFileVO;
|
import springfox.documentation.annotations.ApiIgnore;
|
|
import java.awt.*;
|
import java.io.IOException;
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* @author sean
|
* @version 0.2
|
* @date 2021/12/7
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/media/api/v1/workspaces")
|
@Api(tags = {"媒体资源相关"})
|
public class MediaController {
|
|
@Autowired
|
private IFileService fileService;
|
|
/**
|
* Check if the file has been uploaded by the fingerprint.
|
* 检查文件是否已通过指纹上传。
|
*
|
* @param workspaceId
|
* @param file
|
* @return
|
*/
|
@PostMapping("/{workspace_id}/fast-upload")
|
@ApiIgnore
|
public ResponseResult fastUpload(@PathVariable(name = "workspace_id") String workspaceId, @RequestBody FileUploadDTO file) {
|
|
boolean isExist = fileService.checkExist(workspaceId, file.getFingerprint());
|
|
return isExist ? ResponseResult.success() : ResponseResult.error(file.getFingerprint() + "不存在");
|
}
|
|
|
/**
|
* Query the files that already exist in this workspace based on the workspace id and the collection of tiny fingerprints.
|
*
|
* @param workspaceId
|
* @param tinyFingerprints There is only one tiny_fingerprint parameter in the body.
|
* But it is not recommended to use Map to receive the parameter.
|
* @return
|
*/
|
@PostMapping("/{workspace_id}/files/tiny-fingerprints")
|
@ApiIgnore
|
public ResponseResult<Map<String, List<String>>> uploadCallback(
|
@PathVariable(name = "workspace_id") String workspaceId,
|
@RequestBody Map<String, List<String>> tinyFingerprints) throws JsonProcessingException {
|
|
List<String> existingList = fileService.getExistTinyFingerprints(workspaceId, tinyFingerprints.get(MapKeyConst.TINY_FINGERPRINTS));
|
return ResponseResult.success(new ConcurrentHashMap<>(Map.of(MapKeyConst.TINY_FINGERPRINTS, existingList)));
|
}
|
|
@GetMapping("/{workspace_id}/files/media_page")
|
@ApiIgnore
|
public ResponseResult<PaginationData<MediaJobDTO>> mediaPage(
|
@PathVariable(name = "workspace_id") String workspaceId, @RequestBody SearchMediaParam param) {
|
|
PaginationData<MediaJobDTO> data = fileService.mediaPage(workspaceId, param);
|
return ResponseResult.success(data);
|
}
|
|
|
@GetMapping("/files/detail/{job_id}")
|
@ApiIgnore
|
public ResponseResult<PaginationData<MediaJobDTO>> mediaDetail(
|
@PathVariable(name = "job_id") String jobId, @RequestParam Long page, @RequestParam Long pageSize) {
|
PaginationData<MediaJobDTO> data = fileService.mediaDetail(jobId, page, pageSize);
|
return ResponseResult.success(data);
|
}
|
|
@GetMapping("/files/getMediaFileCountBy")
|
@ApiOperation(value = "首页-任务成果 ", position = 1)
|
public ResponseResult<MediaFileCountVO> getMediaFileCountByResultType(Integer resultType, String areaCode) {
|
return ResponseResult.success(fileService.getMediaFileCountByResultType(resultType, HeaderUtils.formatAreaCode(areaCode)));
|
}
|
|
/**
|
* 任务id
|
*
|
* @param jobIds 任务id
|
* @return
|
*/
|
@ApiModelProperty(name = "jobIds")
|
@GetMapping("/files/getJobsAllFiles")
|
@ApiOperation(value = "航线上事件点的数据", position = 1)
|
public ResponseResult<List<MediaFileEntity>> getJobsAllFiles(@RequestParam("jobIds") List<String> jobIds) {
|
if(CollectionUtils.isEmpty(jobIds)){
|
return ResponseResult.data(Collections.emptyList());
|
}
|
List<MediaFileEntity> list = fileService.getJobsAllFiles(jobIds);
|
list = Streams.filter(list, entity -> {
|
String objectKey = entity.getObjectKey();
|
if (!StringUtils.isEmpty(objectKey)) {
|
return FileTypeUtils.isImageFile(objectKey) || FileTypeUtils.isVideoFile(objectKey);
|
}
|
return false;
|
});
|
return ResponseResult.success(list);
|
|
}
|
|
|
}
|
