package jnpf.dmsService.search;
|
|
import jnpf.dmsEntity.search.model.DmsSearchIndexSource;
|
import jnpf.dmsMapper.DmsSearchIndexSourceMapper;
|
import jnpf.dmsService.search.extract.DmsTextExtractionResult;
|
import jnpf.dmsService.search.extract.DmsTextExtractionService;
|
import jnpf.dmsService.search.file.DmsSearchFileSourceService;
|
import jnpf.dmsService.search.file.DmsTemporarySearchFile;
|
import jnpf.dmsService.search.meili.DmsSearchIndexDocument;
|
import jnpf.dmsService.search.ocr.DmsOcrAdapter;
|
import jnpf.dmsService.search.ocr.DmsOcrAdapterRegistry;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.stereotype.Service;
|
|
import java.io.IOException;
|
import java.util.Optional;
|
|
@Service
|
@RequiredArgsConstructor
|
public class DmsSearchIndexProcessor {
|
private final DmsSearchIndexSourceMapper sourceMapper;
|
private final DmsSearchFileSourceService fileSourceService;
|
private final DmsTextExtractionService extractionService;
|
private final DmsOcrAdapterRegistry ocrAdapterRegistry;
|
private final DmsSearchIndexWriter writer;
|
|
public void process(DmsSearchIndexTask task) throws IOException {
|
DmsSearchIndexSource source = sourceMapper.selectSource(task.getTenantId(),
|
task.getDocumentId(), task.getFileVersionId());
|
if (source == null) {
|
return; // No longer current; expiry cleanup has its own durable retry.
|
}
|
try (DmsTemporarySearchFile file = fileSourceService.fetch(task)) {
|
DmsTextExtractionResult extraction = extractionService.extract(
|
file.getPath(), file.getFileName());
|
String content = extraction.getText();
|
if (extraction.isOcrRequired()) {
|
Optional<DmsOcrAdapter> adapter = ocrAdapterRegistry.configuredAdapter();
|
if (adapter.isPresent()) {
|
content = adapter.get().extract(file.getPath(), file.getFileName(), file.getMimeType());
|
}
|
}
|
writer.publish(document(task, source, content));
|
}
|
}
|
|
private DmsSearchIndexDocument document(DmsSearchIndexTask task,
|
DmsSearchIndexSource source, String content) {
|
DmsSearchIndexDocument document = new DmsSearchIndexDocument();
|
document.setId(source.getFileVersionId().toString());
|
document.setTenantId(source.getTenantId());
|
document.setDocumentId(source.getDocumentId().toString());
|
document.setFileVersionId(source.getFileVersionId().toString());
|
document.setFolderId(source.getFolderId() == null ? null : source.getFolderId().toString());
|
document.setFolderPath(source.getFolderPath());
|
document.setDocumentNo(source.getDocumentNo());
|
document.setTitle(source.getDocumentName());
|
document.setDocumentType(source.getDocumentType());
|
document.setResponsibleDeptId(source.getResponsibleDeptId());
|
document.setVersionLabel(source.getVersionLabel());
|
document.setFileName(source.getFileName());
|
document.setMimeType(source.getMimeType());
|
document.setLifecycleStatus(source.getLifecycleStatus());
|
document.setCreatedAt(source.getCreatedAt());
|
document.setContentChecksum(task.getChecksumSha256());
|
document.setContent(content == null ? "" : content);
|
return document;
|
}
|
}
|
