package cn.iocoder.yudao.module.qcreport.service.aiimport.document;
|
|
import org.springframework.core.annotation.Order;
|
import org.springframework.stereotype.Component;
|
|
import java.util.List;
|
import java.util.Set;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.qcreport.enums.ErrorCodeConstants.AI_IMPORT_FILE_EMPTY;
|
|
/**
|
* 图片（照片 / 截图）抽取器。
|
* <p>
|
* 三种适配器里唯一「什么都不用做」的一个：字节原样交给多模态模型，不解码、不缩放、不转码。
|
* 刻意不在这里用 {@code ImageIO} 读一遍——那不仅多一次内存拷贝，还会在遇到 webp 这类
|
* JDK 原生不支持的格式时白白失败，而模型那边其实完全能认。
|
* <p>
|
* 排在第一位：它只认图片扩展名，与另外两个的匹配集不相交，放哪都不会引起歧义，
|
* 但放在最前面能让「一张照片」这条最常见的路径以最短的路走到头。
|
*/
|
@Component
|
@Order(1)
|
public class ImageImportAdapter implements QcReportImportAdapter {
|
|
private static final Set<String> EXTENSIONS = Set.of("png", "jpg", "jpeg", "bmp", "gif", "webp");
|
|
@Override
|
public boolean supports(String extension, String contentType) {
|
return EXTENSIONS.contains(extension);
|
}
|
|
@Override
|
public QcReportDocumentExtract extract(byte[] content, String originalFilename) {
|
if (content == null || content.length == 0) {
|
throw exception(AI_IMPORT_FILE_EMPTY, originalFilename, 1);
|
}
|
String mimeType = mimeTypeOf(extensionOf(originalFilename));
|
return QcReportDocumentExtract.ofImages(
|
List.of(new QcReportDocumentExtract.ImagePart(content, mimeType, "第 1 页")),
|
1, "raw-image", List.of());
|
}
|
|
/**
|
* PDF 适配器需要同一套映射来标注渲染出来的 PNG，所以公开出来复用。
|
*/
|
static String mimeTypeOf(String extension) {
|
return switch (extension) {
|
case "jpg", "jpeg" -> "image/jpeg";
|
case "gif" -> "image/gif";
|
case "bmp" -> "image/bmp";
|
case "webp" -> "image/webp";
|
default -> "image/png";
|
};
|
}
|
|
private static String extensionOf(String filename) {
|
int dot = filename == null ? -1 : filename.lastIndexOf('.');
|
return dot < 0 ? "" : filename.substring(dot + 1).toLowerCase();
|
}
|
|
}
|
