xiaohuo
8 months ago
parent
df48131704
commit
33d3f19c4c
@ -0,0 +1,189 @@ |
|||||||
|
package com.alops.web.controller.LlmKnowledgeController; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.io.IOException; |
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import javax.servlet.http.HttpServletResponse; |
||||||
|
|
||||||
|
import com.alops.chat.entity.UploadedFile; |
||||||
|
import com.alops.chat.service.UploadedFileService; |
||||||
|
import com.alops.common.config.RuoYiConfig; |
||||||
|
import com.alops.common.core.controller.BaseController; |
||||||
|
import com.alops.common.utils.file.FileUtils; |
||||||
|
import com.alops.common.utils.file.FileTypeUtils; |
||||||
|
import com.alops.common.utils.StringUtils; |
||||||
|
import org.slf4j.Logger; |
||||||
|
import org.slf4j.LoggerFactory; |
||||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||||
|
import org.springframework.http.MediaType; |
||||||
|
import org.springframework.security.access.prepost.PreAuthorize; |
||||||
|
import org.springframework.web.bind.annotation.GetMapping; |
||||||
|
import org.springframework.web.bind.annotation.PathVariable; |
||||||
|
import org.springframework.web.bind.annotation.RequestMapping; |
||||||
|
import org.springframework.web.bind.annotation.RestController; |
||||||
|
|
||||||
|
import java.net.URLEncoder; |
||||||
|
import java.nio.charset.StandardCharsets; |
||||||
|
|
||||||
|
/** |
||||||
|
* 文件下载和预览控制器 |
||||||
|
* |
||||||
|
* @author lingma |
||||||
|
*/ |
||||||
|
@RestController |
||||||
|
@RequestMapping("/QandA/knowledge/file") |
||||||
|
public class FileController extends BaseController { |
||||||
|
private static final Logger log = LoggerFactory.getLogger(FileController.class); |
||||||
|
|
||||||
|
@Autowired |
||||||
|
private UploadedFileService uploadedFileService; |
||||||
|
|
||||||
|
/** |
||||||
|
* 下载文件 |
||||||
|
* |
||||||
|
* @param fileId 文件ID |
||||||
|
* @param response 响应对象 |
||||||
|
*/ |
||||||
|
@PreAuthorize("@ss.hasPermi('QandA:knowledge:filedownload')") |
||||||
|
@GetMapping("/download/{fileId}") |
||||||
|
public void download(@PathVariable Long fileId, HttpServletResponse response, HttpServletRequest request) { |
||||||
|
try { |
||||||
|
UploadedFile uploadedFile = uploadedFileService.findById(fileId); |
||||||
|
if (uploadedFile == null) { |
||||||
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "文件未找到"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 获取文件路径
|
||||||
|
String filePath = uploadedFile.getFilePath(); |
||||||
|
// 获取实际文件路径（处理配置文件前缀）
|
||||||
|
String realPath = RuoYiConfig.getProfile() + filePath.replaceFirst("/profile", ""); |
||||||
|
|
||||||
|
// 检查文件是否存在
|
||||||
|
File file = new File(realPath); |
||||||
|
if (!file.exists()) { |
||||||
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "文件未找到"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 检查文件是否允许下载
|
||||||
|
if (!FileUtils.checkAllowDownload(realPath)) { |
||||||
|
throw new Exception(StringUtils.format("文件({})非法，不允许下载。", realPath)); |
||||||
|
} |
||||||
|
|
||||||
|
// 设置下载文件名（从文件路径中提取真实文件名）
|
||||||
|
String fileName = FileUtils.getName(realPath); |
||||||
|
if (StringUtils.isEmpty(fileName)) { |
||||||
|
fileName = uploadedFile.getSource(); |
||||||
|
// 如果source中也不包含扩展名，则尝试添加扩展名
|
||||||
|
String extension = uploadedFile.getFormat(); |
||||||
|
if (StringUtils.isNotEmpty(extension) && !fileName.endsWith("." + extension)) { |
||||||
|
fileName = fileName + "." + extension; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 对文件名进行编码以支持中文
|
||||||
|
String encodedFileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.toString()) |
||||||
|
.replaceAll("\\+", "%20"); |
||||||
|
|
||||||
|
response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); |
||||||
|
response.setHeader("Content-Disposition", "attachment; filename=" + encodedFileName); |
||||||
|
FileUtils.writeBytes(realPath, response.getOutputStream()); |
||||||
|
} catch (Exception e) { |
||||||
|
log.error("下载文件失败", e); |
||||||
|
try { |
||||||
|
response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, "下载文件失败: " + e.getMessage()); |
||||||
|
} catch (IOException ioException) { |
||||||
|
log.error("设置错误响应失败", ioException); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 预览文件 |
||||||
|
* |
||||||
|
* @param fileId 文件ID |
||||||
|
* @param response 响应对象 |
||||||
|
*/ |
||||||
|
@PreAuthorize("@ss.hasPermi('QandA:knowledge:filepreview')") |
||||||
|
@GetMapping("/preview/{fileId}") |
||||||
|
public void preview(@PathVariable Long fileId, HttpServletResponse response, HttpServletRequest request) { |
||||||
|
try { |
||||||
|
UploadedFile uploadedFile = uploadedFileService.findById(fileId); |
||||||
|
if (uploadedFile == null) { |
||||||
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "文件未找到"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 获取文件路径
|
||||||
|
String filePath = uploadedFile.getFilePath(); |
||||||
|
// 获取实际文件路径（处理配置文件前缀）
|
||||||
|
String realPath = RuoYiConfig.getProfile() + filePath.replaceFirst("/profile", ""); |
||||||
|
|
||||||
|
// 检查文件是否存在
|
||||||
|
File file = new File(realPath); |
||||||
|
if (!file.exists()) { |
||||||
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "文件未找到"); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 检查文件是否允许预览
|
||||||
|
if (!FileUtils.checkAllowDownload(realPath)) { |
||||||
|
throw new Exception(StringUtils.format("文件({})非法，不允许预览。", realPath)); |
||||||
|
} |
||||||
|
|
||||||
|
// 设置Content-Type以便浏览器预览
|
||||||
|
String fileExtension = FileTypeUtils.getFileType(uploadedFile.getSource()); |
||||||
|
if (StringUtils.isEmpty(fileExtension)) { |
||||||
|
fileExtension = uploadedFile.getFormat(); |
||||||
|
} |
||||||
|
String contentType = getContentTypeByExtension(fileExtension); |
||||||
|
if (contentType != null) { |
||||||
|
response.setContentType(contentType); |
||||||
|
} else { |
||||||
|
response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); |
||||||
|
} |
||||||
|
|
||||||
|
FileUtils.writeBytes(realPath, response.getOutputStream()); |
||||||
|
} catch (Exception e) { |
||||||
|
log.error("预览文件失败", e); |
||||||
|
try { |
||||||
|
response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, "预览文件失败: " + e.getMessage()); |
||||||
|
} catch (IOException ioException) { |
||||||
|
log.error("设置错误响应失败", ioException); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 根据文件扩展名获取对应的Content-Type |
||||||
|
* |
||||||
|
* @param extension 文件扩展名 |
||||||
|
* @return Content-Type字符串 |
||||||
|
*/ |
||||||
|
private String getContentTypeByExtension(String extension) { |
||||||
|
if (StringUtils.isEmpty(extension)) { |
||||||
|
return MediaType.APPLICATION_OCTET_STREAM_VALUE; |
||||||
|
} |
||||||
|
|
||||||
|
switch (extension.toLowerCase()) { |
||||||
|
case "txt": |
||||||
|
return "text/plain;charset=UTF-8"; |
||||||
|
case "doc": |
||||||
|
return "application/msword"; |
||||||
|
case "docx": |
||||||
|
return "application/vnd.openxmlformats-officedocument.wordprocessingml.document"; |
||||||
|
case "pdf": |
||||||
|
return "application/pdf"; |
||||||
|
case "jpg": |
||||||
|
case "jpeg": |
||||||
|
return "image/jpeg"; |
||||||
|
case "png": |
||||||
|
return "image/png"; |
||||||
|
case "gif": |
||||||
|
return "image/gif"; |
||||||
|
default: |
||||||
|
return MediaType.APPLICATION_OCTET_STREAM_VALUE; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue