增加媒体库

This commit is contained in:
sdy 2026-05-27 20:25:25 +08:00
parent e96fcc42ed
commit 7d2b0edf78
7 changed files with 476 additions and 2 deletions

View File

@ -0,0 +1,137 @@
package com.multictrl.modules.business.controller;
import cn.hutool.json.JSONObject;
import com.multictrl.common.annotation.ApiOrder;
import com.multictrl.common.annotation.DataFilter;
import com.multictrl.common.annotation.LogOperation;
import com.multictrl.common.constant.Constant;
import com.multictrl.common.page.PageData;
import com.multictrl.common.utils.Result;
import com.multictrl.common.validator.AssertUtils;
import com.multictrl.modules.business.dto.FlightTaskDTO;
import com.multictrl.modules.business.dto.MediaFileDTO;
import com.multictrl.modules.business.service.MediaFileService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.RequiredArgsConstructor;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.core.io.InputStreamResource;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import java.util.List;
import java.util.Map;
/**
* 媒体图库
*
* @author Sdy
* @since 1.0.0 2026/5/27
*/
@RestController
@RequestMapping("business/media")
@Tag(name = "媒体图库")
@ApiOrder(8)
@RequiredArgsConstructor
public class MediaFileController {
private final MediaFileService mediaFileService;
@GetMapping("/page")
@Operation(summary = "分页")
@Parameters({
@Parameter(name = Constant.PAGE, description = "当前页码从1开始"),
@Parameter(name = Constant.LIMIT, description = "每页显示记录数"),
@Parameter(name = "dockSn", description = "机库sn"),
@Parameter(name = "routeId", description = "航线id"),
@Parameter(name = "startTime", description = "开始时间"),
@Parameter(name = "endTime", description = "结束时间")
})
@DataFilter(tableAlias = "t")
@RequiresPermissions("bus:media:page")
public Result<PageData<JSONObject>> page(@Parameter(hidden = true) @RequestParam Map<String, Object> params) {
PageData<JSONObject> page = mediaFileService.pageList(params);
return new Result<PageData<JSONObject>>().ok(page);
}
@GetMapping("/getAlbum")
@Operation(summary = "获取相簿")
@Parameters({
@Parameter(name = Constant.PAGE, description = "当前页码从1开始"),
@Parameter(name = Constant.LIMIT, description = "每页显示记录数"),
@Parameter(name = "dockSn", description = "机库sn")
})
@DataFilter(tableAlias = "t")
@RequiresPermissions("bus:media:album")
public Result<PageData<MediaFileDTO>> getHangarAlbum(@Parameter(hidden = true) @RequestParam Map<String, Object> params) {
PageData<MediaFileDTO> page = mediaFileService.getAlbum(params);
return new Result<PageData<MediaFileDTO>>().ok(page);
}
@DeleteMapping
@Operation(summary = "删除")
@LogOperation("删除媒体文件")
@RequiresPermissions("bus:media:delete")
public Result<Object> delete(@RequestBody Long[] ids) {
//效验数据
AssertUtils.isArrayEmpty(ids, "id");
mediaFileService.deleteMediaFile(ids);
return new Result<>();
}
@Operation(summary = "下载媒体文件")
@LogOperation("下载媒体文件")
@GetMapping("/downloadMedia/{id}")
@RequiresPermissions("bus:media:download")
public ResponseEntity<InputStreamResource> downloadMedia(@PathVariable Long id) {
return mediaFileService.downloadMedia(id);
}
@Operation(summary = "条件下载媒体文件")
@LogOperation("条件下载媒体文件")
@GetMapping("/conditionDownloadMedia")
@RequiresPermissions("bus:media:download")
@Parameters({
@Parameter(name = "hangarSn", description = "机库sn"),
@Parameter(name = "routeId", description = "航线标识"),
@Parameter(name = "startTime", description = "开始时间"),
@Parameter(name = "endTime", description = "结束时间")
})
@DataFilter(tableAlias = "t")
public ResponseEntity<InputStreamResource> conditionDownloadMedia(@Parameter(hidden = true) @RequestParam Map<String, Object> params) {
return mediaFileService.downloadMediaByZip(params);
}
@Operation(summary = "一键获取下载媒体地址")
@LogOperation("一键获取下载媒体地址")
@GetMapping("/oneClickGetDownloadPath")
@RequiresPermissions("bus:media:download")
@Parameters({
@Parameter(name = "hangarSn", description = "机库sn"),
@Parameter(name = "routeId", description = "航线标识"),
@Parameter(name = "startTime", description = "开始时间"),
@Parameter(name = "endTime", description = "结束时间")
})
@DataFilter(tableAlias = "t")
public Result<Map<String, List<String>>> oneClickGetDownloadPath(@Parameter(hidden = true) @RequestParam Map<String, Object> params) {
Map<String, List<String>> map = mediaFileService.oneClickGetDownloadPath(params);
return new Result<Map<String, List<String>>>().ok(map);
}
@DeleteMapping("/conditionDeleteMedia")
@Operation(summary = "条件删除媒体文件 -- 参数和条件下载一样形式为json")
@LogOperation("条件删除媒体文件")
@RequiresPermissions("bus:media:delete")
public Result<Object> conditionDeleteMedia(@RequestBody Map<String, Object> params) {
mediaFileService.conditionDeleteMedia(params);
return new Result<>();
}
}

View File

@ -1,8 +1,13 @@
package com.multictrl.modules.business.dao; package com.multictrl.modules.business.dao;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.multictrl.common.dao.BaseDao; import com.multictrl.common.dao.BaseDao;
import com.multictrl.modules.business.entity.MediaFileEntity; import com.multictrl.modules.business.entity.MediaFileEntity;
import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import java.util.List;
/** /**
* 媒体资源 * 媒体资源
@ -13,4 +18,12 @@ import org.apache.ibatis.annotations.Mapper;
@Mapper @Mapper
public interface MediaFileDao extends BaseDao<MediaFileEntity> { public interface MediaFileDao extends BaseDao<MediaFileEntity> {
//分页列表
IPage<MediaFileEntity> pageList(IPage<MediaFileEntity> page, @Param("ew") QueryWrapper<MediaFileEntity> queryWrapper);
//获取相簿
IPage<MediaFileEntity> getAlbum(IPage<MediaFileEntity> page, @Param("deptId") Long deptId, @Param("dockSn") String dockSn);
//列表
List<MediaFileEntity> list(@Param("ew") QueryWrapper<MediaFileEntity> queryWrapper);
} }

View File

@ -67,9 +67,15 @@ public class MediaFileDTO implements Serializable {
private Date createdTime; private Date createdTime;
@Schema(description = "事件收到时间") @Schema(description = "事件收到时间")
private Date createTime; private Date createDate;
@JsonProperty(access = JsonProperty.Access.READ_ONLY) @JsonProperty(access = JsonProperty.Access.READ_ONLY)
@Schema(description = "媒体路径") @Schema(description = "媒体路径")
private String url; private String url;
@Schema(description = "机库名称")
private String dockName;
@Schema(description = "媒体数量")
private Integer mediaNum;
} }

View File

@ -85,4 +85,14 @@ public class MediaFileEntity {
*/ */
@TableField(fill = FieldFill.INSERT) @TableField(fill = FieldFill.INSERT)
private Date createDate; private Date createDate;
/**
* 机库名称
*/
@TableField(exist = false)
private String dockName;
/**
* 媒体数量
*/
@TableField(exist = false)
private Integer mediaNum;
} }

View File

@ -1,9 +1,16 @@
package com.multictrl.modules.business.service; package com.multictrl.modules.business.service;
import cn.hutool.json.JSONObject;
import com.multictrl.common.page.PageData;
import com.multictrl.common.service.CrudService; import com.multictrl.common.service.CrudService;
import com.multictrl.modules.business.dao.MediaFileDao; import com.multictrl.modules.business.dao.MediaFileDao;
import com.multictrl.modules.business.dto.MediaFileDTO; import com.multictrl.modules.business.dto.MediaFileDTO;
import com.multictrl.modules.business.entity.MediaFileEntity; import com.multictrl.modules.business.entity.MediaFileEntity;
import org.springframework.core.io.InputStreamResource;
import org.springframework.http.ResponseEntity;
import java.util.List;
import java.util.Map;
/** /**
* 媒体资源 * 媒体资源
@ -13,6 +20,27 @@ import com.multictrl.modules.business.entity.MediaFileEntity;
*/ */
public interface MediaFileService extends CrudService<MediaFileEntity, MediaFileDTO> { public interface MediaFileService extends CrudService<MediaFileEntity, MediaFileDTO> {
//图库分页
PageData<JSONObject> pageList(Map<String, Object> params);
//获取相簿
PageData<MediaFileDTO> getAlbum(Map<String, Object> params);
//删除媒体文件
void deleteMediaFile(Long[] ids);
//下载媒体文件
ResponseEntity<InputStreamResource> downloadMedia(Long id);
//下载媒体zip
ResponseEntity<InputStreamResource> downloadMediaByZip(Map<String, Object> params);
//下载文件地址
Map<String, List<String>> oneClickGetDownloadPath(Map<String, Object> params);
//条件删除媒体
void conditionDeleteMedia(Map<String, Object> params);
//获取dao //获取dao
MediaFileDao getDao(); MediaFileDao getDao();
} }

View File

@ -1,15 +1,42 @@
package com.multictrl.modules.business.service.impl; package com.multictrl.modules.business.service.impl;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.multictrl.common.config.MinioConfig;
import com.multictrl.common.constant.BusinessConstant;
import com.multictrl.common.exception.ErrorCode;
import com.multictrl.common.exception.RenException;
import com.multictrl.common.page.PageData;
import com.multictrl.common.service.impl.CrudServiceImpl; import com.multictrl.common.service.impl.CrudServiceImpl;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import com.multictrl.common.utils.ConvertUtils;
import com.multictrl.common.utils.DateUtils;
import com.multictrl.modules.business.dao.MediaFileDao; import com.multictrl.modules.business.dao.MediaFileDao;
import com.multictrl.modules.business.dto.MediaFileDTO; import com.multictrl.modules.business.dto.MediaFileDTO;
import com.multictrl.modules.business.entity.MediaFileEntity; import com.multictrl.modules.business.entity.MediaFileEntity;
import com.multictrl.modules.business.service.MediaFileService; import com.multictrl.modules.business.service.MediaFileService;
import com.multictrl.modules.security.user.SecurityUser;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.springframework.core.io.InputStreamResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.Map; import java.io.*;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.*;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/** /**
* 媒体资源 * 媒体资源
@ -17,8 +44,11 @@ import java.util.Map;
* @author Sdy * @author Sdy
* @since 1.0.0 2026-04-29 * @since 1.0.0 2026-04-29
*/ */
@Slf4j
@Service @Service
@RequiredArgsConstructor
public class MediaFileServiceImpl extends CrudServiceImpl<MediaFileDao, MediaFileEntity, MediaFileDTO> implements MediaFileService { public class MediaFileServiceImpl extends CrudServiceImpl<MediaFileDao, MediaFileEntity, MediaFileDTO> implements MediaFileService {
private final MinioConfig minioConfig;
@Override @Override
public QueryWrapper<MediaFileEntity> getWrapper(Map<String, Object> params) { public QueryWrapper<MediaFileEntity> getWrapper(Map<String, Object> params) {
@ -33,6 +63,205 @@ public class MediaFileServiceImpl extends CrudServiceImpl<MediaFileDao, MediaFil
return wrapper; return wrapper;
} }
@Override
public PageData<JSONObject> pageList(Map<String, Object> params) {
IPage<MediaFileEntity> iPage = baseDao.pageList(getPage(params, null, false), getWrapperBy(params));
PageData<MediaFileDTO> page = getPageData(iPage, MediaFileDTO.class);
List<JSONObject> newList = new ArrayList<>();
List<MediaFileDTO> list = page.getList();
if (CollectionUtil.isNotEmpty(list)) {
Map<String, List<MediaFileDTO>> collect = list.stream()
.filter(v -> v != null && v.getCreatedTime() != null)
.peek(v -> {
String objectKey = v.getObjectKey();
v.setUrl(BusinessConstant.IMAGE_PATH + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + objectKey);
}).collect(Collectors.groupingBy(
v -> DateUtils.format(v.getCreatedTime(), "yyyy年MM月dd日"),
() -> new TreeMap<String, List<MediaFileDTO>>(Comparator.reverseOrder()),
Collectors.toList()
));
for (String key : collect.keySet()) {
JSONObject jsonObject = new JSONObject();
jsonObject.set("date", key);
jsonObject.set("list", collect.get(key));
newList.add(jsonObject);
}
}
return new PageData<>(newList, page.getTotal());
}
@Override
public PageData<MediaFileDTO> getAlbum(Map<String, Object> params) {
String dockSn = (String) params.get("dockSn");
IPage<MediaFileEntity> pageList = this.baseDao.getAlbum(getPage(params, null, false), SecurityUser.getDeptId(), dockSn);
PageData<MediaFileDTO> pageData = getPageData(pageList, MediaFileDTO.class);
for (MediaFileDTO dto : pageData.getList()) {
String objectKey = dto.getObjectKey();
dto.setUrl(BusinessConstant.IMAGE_PATH + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + objectKey);
}
return pageData;
}
@Override
public void deleteMediaFile(Long[] ids) {
for (Long id : ids) {
MediaFileDTO dto = get(id);
if (dto != null) {
String objectKey = dto.getObjectKey();
FileUtil.del(minioConfig.getOther().getDataPath() + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + objectKey);
deleteById(id);
}
}
}
@Override
public ResponseEntity<InputStreamResource> downloadMedia(Long id) {
try {
MediaFileDTO dto = get(id);
String filePath = minioConfig.getOther().getDataPath() + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + dto.getObjectKey();
String fileName = dto.getName();
File file = new File(filePath);
// 检查文件是否存在
if (!file.exists()) {
return ResponseEntity.status(HttpStatus.NOT_FOUND).body(null);
}
// 设置 Content-Disposition
String encodedFileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8).replace("+", "%20");
String contentDisposition = "attachment; filename=" + encodedFileName;
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.parseMediaType(MediaType.APPLICATION_OCTET_STREAM_VALUE));
headers.set(HttpHeaders.CONTENT_DISPOSITION, contentDisposition);
headers.setContentLength((int) Files.size(file.toPath()));
// 使用 InputStreamResource 支持大文件流式传输
InputStream inputStream = Files.newInputStream(file.toPath());
return new ResponseEntity<>(new InputStreamResource(inputStream), headers, HttpStatus.OK);
} catch (IOException e) {
log.error("媒体文件下载失败:{}", ExceptionUtils.getStackTrace(e));
throw new RenException(ErrorCode.INTERNAL_SERVER_ERROR);
}
}
@Override
public ResponseEntity<InputStreamResource> downloadMediaByZip(Map<String, Object> params) {
ZipOutputStream zipOut = null;
ByteArrayOutputStream baos = null;
try {
baos = new ByteArrayOutputStream();
zipOut = new ZipOutputStream(baos);
// 遍历所有ID将文件添加到ZIP中
List<MediaFileEntity> list = baseDao.list(getWrapperBy(params));
for (MediaFileEntity entity : list) {
try {
String filePath = minioConfig.getOther().getDataPath() + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + entity.getObjectKey();
String fileName = entity.getName();
File file = new File(filePath);
// 创建ZIP条目
ZipEntry zipEntry = new ZipEntry(fileName);
zipOut.putNextEntry(zipEntry);
// 将文件内容写入ZIP
Files.copy(file.toPath(), zipOut);
zipOut.closeEntry();
} catch (Exception e) {
// 记录错误但继续处理其他文件
log.error("处理文件ID {} 时出错: {}", entity.getId(), e.getMessage());
}
}
// 关闭ZIP流
zipOut.close();
baos.close();
// 设置响应头
String zipFileName = "media_" + System.currentTimeMillis() + ".zip";
String encodedFileName = URLEncoder.encode(zipFileName, StandardCharsets.UTF_8).replace("+", "%20");
String contentDisposition = "attachment; filename=" + encodedFileName;
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.parseMediaType("application/zip"));
headers.set(HttpHeaders.CONTENT_DISPOSITION, contentDisposition);
headers.setContentLength(baos.size());
// 创建输入流资源
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
return new ResponseEntity<>(new InputStreamResource(bais), headers, HttpStatus.OK);
} catch (IOException e) {
// 关闭流
try {
zipOut.close();
baos.close();
} catch (IOException ex) {
log.error("媒体文件(zip)下载失败:{}", ExceptionUtils.getStackTrace(e));
}
throw new RenException(ErrorCode.INTERNAL_SERVER_ERROR);
}
}
@Override
public Map<String, List<String>> oneClickGetDownloadPath(Map<String, Object> params) {
Map<String, List<String>> map = new HashMap<>();
List<String> images = new ArrayList<>();
List<String> videos = new ArrayList<>();
List<MediaFileEntity> entityList = baseDao.list(getWrapperBy(params));
List<MediaFileDTO> dtos = ConvertUtils.sourceToTarget(entityList, MediaFileDTO.class);
for (MediaFileDTO dto : dtos) {
String name = dto.getName();
if (StrUtil.containsAnyIgnoreCase(name, "mp4")) {
String url = BusinessConstant.VIDEO_PATH + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + dto.getObjectKey();
videos.add(url);
} else {
String url = BusinessConstant.IMAGE_PATH + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + dto.getObjectKey();
images.add(url);
}
}
map.put("images", images);
map.put("videos", videos);
return map;
}
@Override
public void conditionDeleteMedia(Map<String, Object> params) {
List<MediaFileEntity> list = baseDao.list(getWrapperBy(params));
if (CollectionUtil.isEmpty(list)) {
return;
}
for (MediaFileEntity entity : list) {
String path = minioConfig.getOther().getDataPath() + BusinessConstant.DOCK_MEDIA_BUCKET + "/" + entity.getObjectKey();
FileUtil.del(path);
deleteById(entity.getId());
}
}
//获取条件
private QueryWrapper<MediaFileEntity> getWrapperBy(Map<String, Object> params) {
String dockSn = (String) params.get("dockSn");
String routeId = (String) params.get("routeId");
String startTime = (String) params.get("startTime");
String endTime = (String) params.get("endTime");
QueryWrapper<MediaFileEntity> wrapper = new QueryWrapper<>();
wrapper.eq(StrUtil.isNotBlank(dockSn), "t.dock_sn", dockSn);
if (StrUtil.isNotBlank(routeId)) {
wrapper.eq("t.route_id", Long.parseLong(routeId));
}
if (StrUtil.isNotBlank(startTime) && StrUtil.isNotBlank(endTime)) {
wrapper.between("m.created_time", DateUtils.parse(startTime, DateUtils.DATE_TIME_PATTERN), DateUtils.parse(endTime, DateUtils.DATE_TIME_PATTERN));
}
wrapper.orderByDesc("m.created_time"/*, "t.create_date"*/);
// wrapper.orderByAsc("m.off_index");
return wrapper;
}
@Override @Override
public MediaFileDao getDao() { public MediaFileDao getDao() {
return baseDao; return baseDao;

View File

@ -0,0 +1,51 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.multictrl.modules.business.dao.MediaFileDao">
<select id="pageList" resultType="com.multictrl.modules.business.entity.MediaFileEntity">
select m.*
from bus_media_file m
left join bus_flight_task t on m.task_id = t.task_id
${ew.customSqlSegment}
</select>
<select id="getAlbum" resultType="com.multictrl.modules.business.entity.MediaFileEntity">
SELECT a.* FROM(
SELECT DISTINCT
ON (h.dock_name)
h.dock_name,
t.dock_sn,
m.task_id,
m.name,
m.object_key,
total_count AS media_num,h.create_date
FROM bus_flight_task t
LEFT JOIN bus_dock h ON t.dock_sn = h.dock_sn
JOIN bus_media_file m ON t.task_id = m.task_id
LEFT JOIN LATERAL (
SELECT COUNT(1) AS total_count
FROM bus_flight_task t2
JOIN bus_media_file m2 ON t2.task_id = m2.task_id
WHERE t2.dock_sn = h.dock_sn
<if test="deptId != null">
and t2.dept_id = #{deptId}
</if>
) AS subquery ON true
ORDER BY h.dock_name, t.dock_sn, m.created_time DESC
)a
<where>
<if test="dockSn != null and dockSn != ''">
a.dock_sn = #{dockSn}
</if>
</where>
ORDER BY a.create_date
</select>
<select id="list" resultType="com.multictrl.modules.business.entity.MediaFileEntity">
select m.*
from bus_media_file m
left join bus_flight_task t on m.task_id = t.task_id
${ew.customSqlSegment}
</select>
</mapper>