package com.ard.web.controller.sample.media;
|
|
import com.ard.common.config.OssConfiguration;
|
import com.ard.oss.model.CredentialsDTO;
|
import com.ard.oss.service.impl.OssServiceContext;
|
import com.ard.sample.utils.S3PutObjectResponse;
|
import io.minio.MinioClient;
|
import io.minio.PutObjectArgs;
|
import io.minio.credentials.StaticProvider;
|
import io.minio.errors.MinioException;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.http.MediaType;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.PutMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestHeader;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.io.ByteArrayInputStream;
|
import java.security.InvalidKeyException;
|
import java.security.NoSuchAlgorithmException;
|
import java.time.OffsetDateTime;
|
import java.time.format.DateTimeFormatter;
|
|
/**
|
* 适配 MinIO SDK 8.3.7 + 服务器 RELEASE.2024-08-03
|
* 承接大疆 PUT 上传请求，中转上传到 MinIO HTTPS 接口
|
*/
|
@Slf4j
|
@RestController
|
@RequestMapping("/minioTransfer") // 根路径，匹配 Nginx 转发的 /dock/、/kmz/ 路径
|
public class MinioProxyUploadController {
|
|
// @Autowired
|
// private MinioStsUtils minioStsUtils; // 复用你已有的 HTTPS 临时凭证工具类
|
|
@Autowired
|
private OssServiceContext ossService;
|
|
// 你的 MinIO HTTPS 原生地址（固定，与服务器版本匹配）
|
// private static final String MINIO_HTTPS_ENDPOINT = "https://192.168.1.3:9001";
|
|
/**
|
* 核心接口：中转大疆的 PUT 上传请求到 MinIO
|
* 匹配路径：/{bucketName}/{objectName}（例：/dock/xxx.png）
|
* @param bucketName 桶名（dock/kmz，与大疆上传一致）
|
* @param objectName 文件名（例：dji-test-20240124.png）
|
* @param fileContent 文件二进制内容（大疆上传的核心数据）
|
* @param contentType 文件 MIME 类型（默认二进制流，适配多种文件格式）
|
* @return S3 规范 XML 响应（返回给大疆，衔接后续流程）
|
*/
|
@PutMapping("/{bucketName}/{objectName}")
|
public ResponseEntity<S3PutObjectResponse> proxyUpload(
|
@PathVariable String bucketName,
|
@PathVariable String objectName,
|
@RequestBody byte[] fileContent,
|
@RequestHeader(value = "Content-Type", defaultValue = "application/octet-stream") String contentType
|
) {
|
try {
|
log.info("开始中转上传文件到 MinIO,收到参数：bucketName={}, objectName={}",bucketName,objectName);
|
// 步骤 1：获取 MinIO HTTPS 临时凭证（复用你的已有工具类，已验证可行）
|
CredentialsDTO credentials = ossService.getCredentials();
|
if (credentials == null || credentials.getAccessKeyId() == null || credentials.getSecurityToken() == null) {
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(null);
|
}
|
|
// 步骤 2：构建 MinioClient（适配 SDK 8.3.7，忽略私有 SSL 证书，使用临时凭证）
|
MinioClient minioClient = MinioClient.builder()
|
.endpoint(OssConfiguration.endpoint) // 指定 MinIO HTTPS 原生地址
|
.credentialsProvider(new StaticProvider(
|
credentials.getAccessKeyId(),
|
credentials.getAccessKeySecret(),
|
credentials.getSecurityToken() // SDK 8.3.7 支持临时凭证的 sessionToken 参数
|
))
|
.build();
|
|
// 步骤 3：中转上传文件到 MinIO（核心：SDK 8.3.7 兼容 PutObjectArgs 构建）
|
minioClient.putObject(
|
PutObjectArgs.builder()
|
.bucket(bucketName) // 桶名与大疆上传一致（dock/kmz）
|
.object(objectName) // 文件名与大疆上传一致
|
.stream(
|
new ByteArrayInputStream(fileContent), // 文件二进制流
|
fileContent.length, // 文件字节长度
|
-1 // 自动分块（SDK 8.3.7 支持 -1 自动处理，适配小文件上传）
|
)
|
.contentType(contentType) // 文件 MIME 类型
|
.build()
|
);
|
log.info("文件上传成功，文件名：{}",objectName);
|
|
// 步骤 4：构造符合 S3 规范的成功响应（返回给大疆，确保流程衔接）
|
S3PutObjectResponse successResponse = new S3PutObjectResponse(
|
// ETag：生成文件 MD5 值（带双引号，符合 S3 规范，依赖 commons-codec）
|
"\"" + org.apache.commons.codec.digest.DigestUtils.md5Hex(fileContent) + "\"",
|
// LastModified：当前时间（ISO 格式，MinIO 服务器兼容该格式）
|
OffsetDateTime.now().format(DateTimeFormatter.ISO_OFFSET_DATE_TIME)
|
);
|
log.info("构造回复大疆机场的响应信息-S3PutObjectResponse:{}",successResponse);
|
|
// 步骤 5：返回 200 OK + XML 格式响应（Spring Boot 自动转换实体类为 XML）
|
return ResponseEntity.status(HttpStatus.OK)
|
.contentType(MediaType.APPLICATION_XML)
|
.body(successResponse);
|
|
} catch (MinioException e) {
|
// MinIO 上传异常（SDK 8.3.7 抛出的异常类型，与高版本一致）
|
e.printStackTrace();
|
log.info("MinIO 上传异常:{}",e.getMessage());
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(null);
|
} catch (NoSuchAlgorithmException | InvalidKeyException e) {
|
// 凭证/算法异常（适配 SDK 8.3.7 的加密校验逻辑）
|
e.printStackTrace();
|
log.info("MinIO 凭证或算法异常:{}",e.getMessage());
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(null);
|
} catch (Exception e) {
|
// 其他通用异常
|
e.printStackTrace();
|
log.info("MinIO 异常:{}",e.getMessage());
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(null);
|
}
|
}
|
}
|
