feat(resource): 集成OnlyOffice文档编辑功能

linwei

2026-07-10 41bc55bc6ab5e70a12736ae174a248b5d37ec5bd

 drone-ops/drone-resource/src/main/java/org/sxkj/resource/controller/AttachController.java

@@ -547,40 +547,63 @@
      return baseUrl.toString() + "/attach/onlyoffice-download?attachId=" + attachId;
   }

   /**
    * OnlyOffice代理下载接口，用于代理下载远程文件
    * <p>
    * 当OnlyOffice Document Server无法直接访问远程文件URL时，
    * 通过本接口代理下载文件并流式返回给Document Server。
    * </p>
    *
    * @param request  HTTP请求对象
    * @param response HTTP响应对象
    * @param attachId 附件ID，用于从缓存中查找对应的远程文件URL
    * @throws IOException 当下载或写入文件流时发生I/O异常
    */
   @GetMapping("/onlyoffice-download")
   @ApiOperation(value = "OnlyOffice代理下载", notes = "代理下载远程文件供OnlyOffice Document Server访问")
   public void onlyofficeDownload(
         HttpServletRequest request,
         HttpServletResponse response,
         @ApiParam(value = "附件ID", required = true) @RequestParam Long attachId) throws IOException {
      HttpServletRequest request,
      HttpServletResponse response,
      @ApiParam(value = "附件ID", required = true) @RequestParam Long attachId) throws IOException {

      // 1. 从缓存中获取远程文件URL
      String remoteUrl = REMOTE_URL_CACHE.get(attachId);
      if (remoteUrl == null || remoteUrl.isEmpty()) {
         response.sendError(HttpServletResponse.SC_NOT_FOUND, "未找到对应的远程文件URL");
      // 处理 OPTIONS 预检请求
      if ("OPTIONS".equalsIgnoreCase(request.getMethod())) {
         response.setHeader("Access-Control-Allow-Origin", "*");
         response.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
         response.setHeader("Access-Control-Allow-Headers", "*");
         response.setStatus(HttpServletResponse.SC_OK);
         return;
      }

      // 2. 根据附件ID查询附件信息，获取文件名
      Attach attach = attachService.getOne(Wrappers.lambdaQuery(Attach.class).eq(Attach::getId, attachId));
      String fileName = (attach != null && attach.getName() != null) ? attach.getName() : "document.docx";
      log.info("ONLYOFFICE 下载请求，attachId: {}", attachId);

      // 3. 从远程URL下载文件
      // 1. 根据附件ID查询附件信息
      Attach attach = attachService.getOne(Wrappers.lambdaQuery(Attach.class).eq(Attach::getId, attachId));
      if (attach == null) {
         response.sendError(HttpServletResponse.SC_NOT_FOUND, "附件不存在，attachId: " + attachId);
         return;
      }

      // 2. 优先从缓存中获取远程文件URL
      String remoteUrl = REMOTE_URL_CACHE.get(attachId);
      if (remoteUrl == null || remoteUrl.isEmpty()) {
         remoteUrl = attach.getLink();
         log.info("从数据库获取远程URL，attachId: {}, link: {}", attachId, remoteUrl);
      }

      // 3. 验证远程URL
      if (remoteUrl == null || remoteUrl.isEmpty()) {
         response.sendError(HttpServletResponse.SC_NOT_FOUND, "附件缺少远程文件URL，attachId: " + attachId);
         return;
      }

      // 4. 获取纯文件名（不含路径）
      String fileName = attach.getName();
      if (fileName == null || fileName.isEmpty()) {
         fileName = "document.docx";
      }
      // 移除路径部分
      if (fileName.contains("/")) {
         fileName = fileName.substring(fileName.lastIndexOf("/") + 1);
      }
      if (fileName.contains("\\")) {
         fileName = fileName.substring(fileName.lastIndexOf("\\") + 1);
      }
      log.info("下载文件名: {}", fileName);

      // 5. 下载远程文件
      HttpURLConnection connection = null;
      InputStream inputStream = null;
      OutputStream outputStream = null;

      try {
         URL url = new URL(remoteUrl);
         connection = (HttpURLConnection) url.openConnection();
@@ -588,42 +611,78 @@
         connection.setConnectTimeout(CONNECT_TIMEOUT);
         connection.setReadTimeout(READ_TIMEOUT);

         // 4. 检查响应状态码
         // 复制请求头（如果需要）
         String userAgent = request.getHeader("User-Agent");
         if (userAgent != null) {
            connection.setRequestProperty("User-Agent", userAgent);
         }

         int responseCode = connection.getResponseCode();
         if (responseCode != HttpURLConnection.HTTP_OK) {
            log.error("远程文件下载失败，状态码: {}, url: {}", responseCode, remoteUrl);
            response.sendError(HttpServletResponse.SC_BAD_GATEWAY, "远程文件下载失败，状态码: " + responseCode);
            return;
         }

         // 5. 获取远程文件的Content-Type
         // 6. 设置响应头
         String contentType = connection.getContentType();
         if (contentType == null || contentType.isEmpty()) {
            contentType = "application/octet-stream";
            contentType = getContentTypeByFileName(fileName);
         }
         response.setContentType(contentType);

         // 设置 Content-Length
         int contentLength = connection.getContentLength();
         if (contentLength > 0) {
            response.setContentLength(contentLength);
         }

         // 6. 设置响应头
         response.setContentType(contentType);
         response.setHeader("Content-Disposition", "attachment; filename=\""
               + URLEncoder.encode(fileName, StandardCharsets.UTF_8.name()) + "\"");
         // 7. 正确设置 Content-Disposition
         String encodedFileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name())
            .replaceAll("\\+", "%20");

         // 7. 流式传输文件内容
         // ONLYOFFICE 需要 inline 方式访问
         response.setHeader("Content-Disposition",
            "inline; filename=\"" + encodedFileName + "\"; " +
               "filename*=UTF-8''" + encodedFileName);

         // CORS 头
         response.setHeader("Access-Control-Allow-Origin", "*");
         response.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
         response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");

         // 缓存控制
         response.setHeader("Cache-Control", "no-cache, no-store, must-revalidate");
         response.setHeader("Pragma", "no-cache");
         response.setHeader("Expires", "0");

         // 8. 流式传输
         inputStream = connection.getInputStream();
         outputStream = response.getOutputStream();
         byte[] buffer = new byte[8192];
         int bytesRead;
         long totalBytes = 0;
         while ((bytesRead = inputStream.read(buffer)) != -1) {
            outputStream.write(buffer, 0, bytesRead);
            totalBytes += bytesRead;
         }
         outputStream.flush();

         log.info("下载完成，文件大小: {} bytes", totalBytes);

      } catch (Exception e) {
         log.error("代理下载远程文件失败，attachId: {}, remoteUrl: {}, 错误: {}", attachId, remoteUrl, e.getMessage(), e);
         log.error("代理下载远程文件失败，attachId: {}, remoteUrl: {}, 错误: {}",
            attachId, remoteUrl, e.getMessage(), e);
         if (!response.isCommitted()) {
            response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, "代理下载远程文件失败: " + e.getMessage());
            response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR,
               "代理下载远程文件失败: " + e.getMessage());
         }
      } finally {
         // 8. 关闭资源
         if (inputStream != null) {
            try { inputStream.close(); } catch (IOException ignored) {}
         }
         if (outputStream != null) {
            try { outputStream.close(); } catch (IOException ignored) {}
         }
         if (connection != null) {
            connection.disconnect();
@@ -631,4 +690,42 @@
      }
   }

   /**
    * 根据文件名获取 Content-Type
    */
   private String getContentTypeByFileName(String fileName) {
      if (fileName == null) return "application/octet-stream";

      String extension = "";
      int lastDot = fileName.lastIndexOf(".");
      if (lastDot > 0) {
         extension = fileName.substring(lastDot + 1).toLowerCase();
      }

      switch (extension) {
         case "docx":
            return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
         case "doc":
            return "application/msword";
         case "xlsx":
            return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
         case "xls":
            return "application/vnd.ms-excel";
         case "pptx":
            return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
         case "ppt":
            return "application/vnd.ms-powerpoint";
         case "pdf":
            return "application/pdf";
         case "txt":
            return "text/plain";
         case "csv":
            return "text/csv";
         case "rtf":
            return "application/rtf";
         default:
            return "application/octet-stream";
      }
   }

}