OcrService.java
2.68 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package com.xly.ocr.service;
import com.xly.ocr.util.OcrUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.util.Arrays;
import java.util.List;
@Slf4j
@Service("ocrService")
public class OcrService {
@Value("${ocr.tmpPath}")
private String tmpPath;
private static final List<String> ALLOWED_EXTENSIONS = Arrays.asList(
".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".gif"
);
private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB
/**
* 从 MultipartFile 中提取文字
*/
public String extractTextFromMultipartFile(MultipartFile file) {
// 1. 验证文件
String validationError = validateFile(file);
if (validationError != null) {
return validationError;
}
// 2. 调用 OCR 工具类识别
try {
String result = OcrUtil.ocrFile(file, tmpPath);
if (result == null || result.trim().isEmpty()) {
log.warn("未识别到文字内容，文件: {}", file.getOriginalFilename());
return "未识别到文字内容";
}
log.info("识别成功，文件: {}, 文字长度: {}",
file.getOriginalFilename(), result.length());
return result;
} catch (Exception e) {
log.error("OCR识别异常: {}", e.getMessage(), e);
return "OCR识别失败: " + e.getMessage();
}
}
/**
* 验证文件
*/
private String validateFile(MultipartFile file) {
if (file == null || file.isEmpty()) {
log.warn("上传的文件为空");
return "上传的文件为空";
}
if (file.getSize() > MAX_FILE_SIZE) {
log.warn("文件过大: {} bytes", file.getSize());
return String.format("文件过大，最大支持 %dMB", MAX_FILE_SIZE / 1024 / 1024);
}
String originalFilename = file.getOriginalFilename();
if (originalFilename != null && !isAllowedImage(originalFilename)) {
log.warn("不支持的文件格式: {}", originalFilename);
return "不支持的文件格式，仅支持: " + String.join(", ", ALLOWED_EXTENSIONS);
}
return null;
}
/**
* 检查文件扩展名是否允许
*/
private boolean isAllowedImage(String filename) {
if (filename == null) {
return false;
}
String lowerFilename = filename.toLowerCase();
return ALLOWED_EXTENSIONS.stream()
.anyMatch(lowerFilename::endsWith);
}
}