调用dify,解决时间问题,源语料为空的问题

This commit is contained in:
spllzh
2025-09-08 00:00:08 +08:00
parent a25d80ae28
commit 206275317f
9 changed files with 1552 additions and 0 deletions

View File

@@ -0,0 +1,302 @@
package com.rj.service.impl;
import com.rj.entity.AudioManagement;
import com.rj.service.IFileUploadService;
import com.rj.service.IAudioManagementService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
/**
* 文件上传服务实现类
*
* @author 李中华 ,spllzh
* @since 2025-08-07
*/
@Slf4j
@Service
public class FileUploadServiceImpl implements IFileUploadService {
@Autowired
private IAudioManagementService audioManagementService;
@Value("${app.audio.upload.path:uploads/audio}")
private String uploadPath;
@Value("${app.audio.access.url:/api/audio/}")
private String accessUrl;
@Value("${app.audio.max.size:100MB}")
private String maxFileSize;
// 支持的音频格式
private static final String[] SUPPORTED_AUDIO_FORMATS = {
"mp3", "wav", "m4a", "aac", "ogg", "flac", "wma"
};
@Override
public Map<String, Object> uploadAudioFile(MultipartFile file, String audioId) throws IOException {
Map<String, Object> result = new HashMap<>();
try {
// 验证文件
if (file == null || file.isEmpty()) {
result.put("success", false);
result.put("message", "文件不能为空");
return result;
}
// 验证文件大小
long maxSize = parseFileSize(maxFileSize);
if (file.getSize() > maxSize) {
result.put("success", false);
result.put("message", "文件大小超过限制,最大允许 " + maxFileSize);
return result;
}
// 验证文件格式
String originalFilename = file.getOriginalFilename();
if (!isValidAudioFormat(originalFilename)) {
result.put("success", false);
result.put("message", "不支持的音频格式,支持的格式:" + String.join(", ", SUPPORTED_AUDIO_FORMATS));
return result;
}
// 创建上传目录
Path uploadDir = Paths.get(uploadPath);
if (!Files.exists(uploadDir)) {
Files.createDirectories(uploadDir);
}
// 生成文件名
String fileExtension = getFileExtension(originalFilename);
String fileName = audioId+"_"+UUID.randomUUID() + "." + fileExtension;
Path filePath = uploadDir.resolve(fileName);
// 保存文件
Files.copy(file.getInputStream(), filePath);
// 生成访问URL
String fileUrl = accessUrl + fileName;
result.put("success", true);
result.put("message", "文件上传成功");
result.put("fileName", fileName);
result.put("fileUrl", fileUrl);
result.put("fileSize", file.getSize());
result.put("originalName", originalFilename);
log.info("音频文件上传成功audioId={}, fileName={}, size={}", audioId, fileName, file.getSize());
} catch (Exception e) {
log.error("音频文件上传失败audioId={}, error={}", audioId, e.getMessage(), e);
result.put("success", false);
result.put("message", "文件上传失败:" + e.getMessage());
}
return result;
}
@Override
public Map<String, Object> uploadAudioFileAndUpdateRecord(MultipartFile file, String audioId) throws IOException {
Map<String, Object> result = new HashMap<>();
try {
// 先上传文件
Map<String, Object> uploadResult = uploadAudioFile(file, audioId);
if (!(Boolean) uploadResult.get("success")) {
return uploadResult;
}
// 获取录音记录
AudioManagement audio = audioManagementService.getById(audioId);
if (audio == null) {
result.put("success", false);
result.put("message", "录音记录不存在");
return result;
}
// 更新录音记录的文件信息
String fileName = (String) uploadResult.get("fileName");
String fileUrl = (String) uploadResult.get("fileUrl");
Long fileSize = (Long) uploadResult.get("fileSize");
String originalName = (String) uploadResult.get("originalName");
String fileExtension = getFileExtension(originalName);
audio.setAudioFilePath(uploadPath + "/" + fileName);
audio.setAudioFileUrl(fileUrl);
audio.setAudioFileSize(fileSize);
audio.setAudioFileOriginalName(originalName);
audio.setAudioFileExtension(fileExtension);
audio.setUploadTime(java.time.LocalDateTime.now());
audio.setUpdateTime(java.time.LocalDateTime.now());
// 保存更新
boolean updated = audioManagementService.updateById(audio);
if (updated) {
result.put("success", true);
result.put("message", "音频文件上传成功并更新录音记录");
result.put("data", uploadResult);
result.put("audioRecord", audio);
log.info("音频文件上传并更新录音记录成功audioId={}, fileName={}", audioId, fileName);
} else {
result.put("success", false);
result.put("message", "文件上传成功但更新录音记录失败");
result.put("data", uploadResult);
}
} catch (Exception e) {
log.error("音频文件上传并更新录音记录失败audioId={}, error={}", audioId, e.getMessage(), e);
result.put("success", false);
result.put("message", "文件上传并更新录音记录失败:" + e.getMessage());
}
return result;
}
@Override
public String getAudioFileUrl(String audioId) {
if (StringUtils.isEmpty(audioId)) {
return null;
}
Path uploadDir = Paths.get(uploadPath);
File dir = uploadDir.toFile();
if (!dir.exists() || !dir.isDirectory()) {
return null;
}
// 查找对应的音频文件
File[] files = dir.listFiles((d, name) -> name.startsWith(audioId + "."));
if (files != null && files.length > 0) {
String fileName = files[0].getName();
return accessUrl + fileName;
}
return null;
}
@Override
public boolean deleteAudioFile(String audioId) {
if (StringUtils.isEmpty(audioId)) {
return false;
}
try {
Path uploadDir = Paths.get(uploadPath);
File dir = uploadDir.toFile();
if (!dir.exists() || !dir.isDirectory()) {
return false;
}
// 查找对应的音频文件
File[] files = dir.listFiles((d, name) -> name.startsWith(audioId + "."));
if (files != null && files.length > 0) {
boolean deleted = files[0].delete();
if (deleted) {
log.info("音频文件删除成功audioId={}, fileName={}", audioId, files[0].getName());
}
return deleted;
}
return false;
} catch (Exception e) {
log.error("音频文件删除失败audioId={}, error={}", audioId, e.getMessage(), e);
return false;
}
}
@Override
public boolean audioFileExists(String audioId) {
if (StringUtils.isEmpty(audioId)) {
return false;
}
Path uploadDir = Paths.get(uploadPath);
File dir = uploadDir.toFile();
if (!dir.exists() || !dir.isDirectory()) {
return false;
}
File[] files = dir.listFiles((d, name) -> name.startsWith(audioId + "."));
return files != null && files.length > 0;
}
/**
* 验证音频文件格式
*/
private boolean isValidAudioFormat(String filename) {
if (StringUtils.isEmpty(filename)) {
return false;
}
String extension = getFileExtension(filename).toLowerCase();
for (String format : SUPPORTED_AUDIO_FORMATS) {
if (format.equals(extension)) {
return true;
}
}
return false;
}
/**
* 获取文件扩展名
*/
private String getFileExtension(String filename) {
if (StringUtils.isEmpty(filename)) {
return "";
}
int lastDotIndex = filename.lastIndexOf('.');
if (lastDotIndex > 0 && lastDotIndex < filename.length() - 1) {
return filename.substring(lastDotIndex + 1);
}
return "";
}
/**
* 解析文件大小字符串
*/
private long parseFileSize(String sizeStr) {
if (StringUtils.isEmpty(sizeStr)) {
return 100 * 1024 * 1024; // 默认100MB
}
sizeStr = sizeStr.trim().toUpperCase();
long multiplier = 1;
if (sizeStr.endsWith("KB")) {
multiplier = 1024;
sizeStr = sizeStr.substring(0, sizeStr.length() - 2);
} else if (sizeStr.endsWith("MB")) {
multiplier = 1024 * 1024;
sizeStr = sizeStr.substring(0, sizeStr.length() - 2);
} else if (sizeStr.endsWith("GB")) {
multiplier = 1024 * 1024 * 1024;
sizeStr = sizeStr.substring(0, sizeStr.length() - 2);
}
try {
return Long.parseLong(sizeStr) * multiplier;
} catch (NumberFormatException e) {
return 100 * 1024 * 1024; // 默认100MB
}
}
}