添加Mutex互斥锁
This commit is contained in:
ocean
parent
1f664cd191
commit
86750a4429
@ -13,6 +13,8 @@ import com.all.pdfreader.pro.app.room.entity.PdfDocumentEntity
|
|||||||
import com.all.pdfreader.pro.app.room.repository.PdfRepository
|
import com.all.pdfreader.pro.app.room.repository.PdfRepository
|
||||||
import com.all.pdfreader.pro.app.util.FileUtils.isPdfEncrypted
|
import com.all.pdfreader.pro.app.util.FileUtils.isPdfEncrypted
|
||||||
import kotlinx.coroutines.Dispatchers
|
import kotlinx.coroutines.Dispatchers
|
||||||
|
import kotlinx.coroutines.sync.Mutex
|
||||||
|
import kotlinx.coroutines.sync.withLock
|
||||||
import kotlinx.coroutines.withContext
|
import kotlinx.coroutines.withContext
|
||||||
import java.io.File
|
import java.io.File
|
||||||
import java.io.FileOutputStream
|
import java.io.FileOutputStream
|
||||||
@ -23,210 +25,215 @@ class PdfScanner(
|
|||||||
) {
|
) {
|
||||||
|
|
||||||
private val TAG = "ocean-PdfScanner"
|
private val TAG = "ocean-PdfScanner"
|
||||||
|
private val scanMutex = Mutex()
|
||||||
suspend fun scanAndLoadPdfFiles(isNeedFullScan: Boolean, callback: (Boolean) -> Unit = {}) {
|
suspend fun scanAndLoadPdfFiles(isNeedFullScan: Boolean, callback: (Boolean) -> Unit = {}) {
|
||||||
if (!StoragePermissionHelper.hasBasicStoragePermission(context)) {
|
if (!StoragePermissionHelper.hasBasicStoragePermission(context)) {
|
||||||
LogUtil.logDebug(TAG, "权限不足")
|
LogUtil.logDebug(TAG, "权限不足")
|
||||||
callback.invoke(false)
|
callback.invoke(false)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
scanMutex.withLock {// 保证同一时间只有一次扫描
|
||||||
withContext(Dispatchers.IO) {
|
withContext(Dispatchers.IO) {
|
||||||
try {
|
try {
|
||||||
val startScannerTime = System.currentTimeMillis()
|
val startScannerTime = System.currentTimeMillis()
|
||||||
LogUtil.logDebug(TAG, "🔍开始扫描PDF文件...")
|
LogUtil.logDebug(TAG, "🔍开始扫描PDF文件...")
|
||||||
LogUtil.logDebug(TAG, "获取数据库数据")
|
LogUtil.logDebug(TAG, "获取数据库数据")
|
||||||
val cachedDocs = pdfRepository.getAllDocumentsOnce()
|
val cachedDocs = pdfRepository.getAllDocumentsOnce()
|
||||||
var needFullScan = isNeedFullScan
|
var needFullScan = isNeedFullScan
|
||||||
if (!needFullScan) {
|
if (!needFullScan) {
|
||||||
cachedDocs.forEach { doc ->
|
cachedDocs.forEach { doc ->
|
||||||
val file = File(doc.filePath)//得到数据库存储的文件路径
|
val file = File(doc.filePath)//得到数据库存储的文件路径
|
||||||
if (file.exists()) {
|
if (file.exists()) {
|
||||||
LogUtil.logDebug(TAG, "$file 文件存在，检测是否存在变化")
|
LogUtil.logDebug(TAG, "$file 文件存在，检测是否存在变化")
|
||||||
var needUpdate = false
|
var needUpdate = false
|
||||||
var updatedDoc = doc.copy()
|
var updatedDoc = doc.copy()
|
||||||
// 修改时间 与 文件大小
|
// 修改时间 与 文件大小
|
||||||
if (file.lastModified() != doc.lastModified || file.length() != doc.fileSize) {
|
if (file.lastModified() != doc.lastModified || file.length() != doc.fileSize) {
|
||||||
LogUtil.logDebug(TAG, "文件时间或者大小被修改 -> ${doc.fileName}")
|
|
||||||
updatedDoc = updatedDoc.copy(
|
|
||||||
lastModified = file.lastModified(),
|
|
||||||
fileSize = file.length()
|
|
||||||
)
|
|
||||||
needUpdate = true
|
|
||||||
}
|
|
||||||
if (needUpdate) {
|
|
||||||
// 是否加密
|
|
||||||
val currentIsPassword = isPdfEncrypted(file)
|
|
||||||
if (doc.isPassword != currentIsPassword) {
|
|
||||||
LogUtil.logDebug(TAG, "密码状态变化 -> ${doc.fileName}")
|
|
||||||
updatedDoc = updatedDoc.copy(isPassword = currentIsPassword)
|
|
||||||
}
|
|
||||||
// 缩略图（仅非加密文件）
|
|
||||||
if (!currentIsPassword) {
|
|
||||||
val newThumbnail = generateThumbnail(context, file)
|
|
||||||
if (doc.thumbnailPath != newThumbnail) {
|
|
||||||
updatedDoc = updatedDoc.copy(thumbnailPath = newThumbnail)
|
|
||||||
}
|
|
||||||
} else if (doc.thumbnailPath != null) {
|
|
||||||
updatedDoc = updatedDoc.copy(thumbnailPath = null)
|
|
||||||
}
|
|
||||||
pdfRepository.insertOrUpdateDocument(updatedDoc)
|
|
||||||
LogUtil.logDebug(TAG, "✅数据库已更新: ${doc.fileName}")
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// 文件不存在 → 删除数据库记录，并触发全盘扫描
|
|
||||||
LogUtil.logDebug(TAG, "文件不存在 -> ${doc.fileName}, 删除记录")
|
|
||||||
pdfRepository.deleteDocument(doc.fileHash)
|
|
||||||
needFullScan = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (needFullScan || cachedDocs.isEmpty()) {
|
|
||||||
LogUtil.logDebug(TAG, "数据库不完整，执行全盘扫描...")
|
|
||||||
// 扫描应用私有目录（无需权限）
|
|
||||||
val privateFiles = FileUtils.scanPdfFiles(context)
|
|
||||||
LogUtil.logDebug(TAG, "📁应用私有目录找到: ${privateFiles.size} 个PDF文件")
|
|
||||||
// 扫描MediaStore（需要权限）
|
|
||||||
val mediaStoreFiles = FileUtils.scanPdfFilesFromMediaStore(context)
|
|
||||||
LogUtil.logDebug(TAG, "📱MediaStore找到: ${mediaStoreFiles.size} 个PDF文件")
|
|
||||||
// 合并并去重
|
|
||||||
val allFiles = (privateFiles + mediaStoreFiles).distinctBy { it.absolutePath }
|
|
||||||
LogUtil.logDebug(TAG, "📊总计扫描到: ${allFiles.size} 个PDF文件")
|
|
||||||
// 处理每个PDF文件
|
|
||||||
allFiles.forEachIndexed { index, file ->
|
|
||||||
LogUtil.logDebug(
|
|
||||||
TAG,
|
|
||||||
"🔄处理文件 ${index + 1}/${allFiles.size}: ${file.name} - ${file.absolutePath}"
|
|
||||||
)
|
|
||||||
|
|
||||||
if (FileUtils.isPdfFile(file)) {
|
|
||||||
val fileHash = FileUtils.calculateFileHash(file.absolutePath)
|
|
||||||
LogUtil.logDebug(TAG, "🔑文件哈希: $fileHash")
|
|
||||||
|
|
||||||
if (fileHash != null) {
|
|
||||||
val existingDoc = pdfRepository.getDocumentByPath(file.absolutePath)
|
|
||||||
|
|
||||||
if (existingDoc == null) {
|
|
||||||
LogUtil.logDebug(
|
LogUtil.logDebug(
|
||||||
TAG, "🆕发现新PDF文件: ${file.name}"
|
TAG, "文件时间或者大小被修改 -> ${doc.fileName}"
|
||||||
)
|
)
|
||||||
var thumbnailPath: String? = null
|
updatedDoc = updatedDoc.copy(
|
||||||
val isPassword = isPdfEncrypted(file)
|
lastModified = file.lastModified(), fileSize = file.length()
|
||||||
LogUtil.logDebug(TAG, "isPassword->${isPassword}")
|
|
||||||
if (!isPassword) {//没有密码的情况下才去获取缩略图
|
|
||||||
thumbnailPath = generateThumbnail(context, file) ?: ""
|
|
||||||
}
|
|
||||||
LogUtil.logDebug(TAG, "thumbnailPath->${thumbnailPath}")
|
|
||||||
val metadata =
|
|
||||||
PdfMetadataExtractor.extractMetadata(file.absolutePath)
|
|
||||||
val document = PdfDocumentEntity(
|
|
||||||
fileHash = fileHash,
|
|
||||||
filePath = file.absolutePath,
|
|
||||||
fileName = file.name,
|
|
||||||
fileSize = file.length(),
|
|
||||||
lastModified = file.lastModified(),
|
|
||||||
pageCount = metadata?.pageCount ?: 0,
|
|
||||||
thumbnailPath = thumbnailPath,
|
|
||||||
metadataTitle = metadata?.title,
|
|
||||||
metadataAuthor = metadata?.author,
|
|
||||||
metadataSubject = metadata?.subject,
|
|
||||||
metadataKeywords = metadata?.keywords,
|
|
||||||
metadataCreationDate = metadata?.creationDate?.time,
|
|
||||||
metadataModificationDate = metadata?.modificationDate?.time,
|
|
||||||
isPassword = isPassword
|
|
||||||
)
|
)
|
||||||
pdfRepository.insertOrUpdateDocument(document)
|
needUpdate = true
|
||||||
LogUtil.logDebug(
|
}
|
||||||
TAG, " ✅ 已保存到数据库: ${file.name}"
|
if (needUpdate) {
|
||||||
)
|
// 是否加密
|
||||||
} else {
|
|
||||||
LogUtil.logDebug(TAG, " 📋 文件已存在: ${file.name}")
|
|
||||||
// 🔹 文件已存在，检查是否需要更新
|
|
||||||
var needUpdate = false
|
|
||||||
var updatedDoc = existingDoc.copy()
|
|
||||||
|
|
||||||
// 路径/修改时间更新
|
|
||||||
if (existingDoc.filePath != file.absolutePath || existingDoc.lastModified != file.lastModified()) {
|
|
||||||
LogUtil.logDebug(TAG, "✅ 路径/修改时间需要更新")
|
|
||||||
updatedDoc = updatedDoc.copy(
|
|
||||||
filePath = file.absolutePath,
|
|
||||||
lastModified = file.lastModified()
|
|
||||||
)
|
|
||||||
needUpdate = true
|
|
||||||
}
|
|
||||||
|
|
||||||
// 是否加密更新
|
|
||||||
val currentIsPassword = isPdfEncrypted(file)
|
val currentIsPassword = isPdfEncrypted(file)
|
||||||
if (existingDoc.isPassword != currentIsPassword) {
|
if (doc.isPassword != currentIsPassword) {
|
||||||
LogUtil.logDebug(TAG, "✅ 密码状态需要更新")
|
LogUtil.logDebug(TAG, "密码状态变化 -> ${doc.fileName}")
|
||||||
updatedDoc = updatedDoc.copy(isPassword = currentIsPassword)
|
updatedDoc = updatedDoc.copy(isPassword = currentIsPassword)
|
||||||
needUpdate = true
|
|
||||||
}
|
}
|
||||||
|
// 缩略图（仅非加密文件）
|
||||||
if (!currentIsPassword) {
|
if (!currentIsPassword) {
|
||||||
// 如果不是加密 PDF，再生成缩略图
|
|
||||||
val newThumbnail = generateThumbnail(context, file)
|
val newThumbnail = generateThumbnail(context, file)
|
||||||
if (existingDoc.thumbnailPath != newThumbnail) {
|
if (doc.thumbnailPath != newThumbnail) {
|
||||||
LogUtil.logDebug(TAG, "✅ 缩略图需要更新")
|
|
||||||
updatedDoc =
|
updatedDoc =
|
||||||
updatedDoc.copy(thumbnailPath = newThumbnail)
|
updatedDoc.copy(thumbnailPath = newThumbnail)
|
||||||
|
}
|
||||||
|
} else if (doc.thumbnailPath != null) {
|
||||||
|
updatedDoc = updatedDoc.copy(thumbnailPath = null)
|
||||||
|
}
|
||||||
|
pdfRepository.insertOrUpdateDocument(updatedDoc)
|
||||||
|
LogUtil.logDebug(TAG, "✅数据库已更新: ${doc.fileName}")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 文件不存在 → 删除数据库记录，并触发全盘扫描
|
||||||
|
LogUtil.logDebug(TAG, "文件不存在 -> ${doc.fileName}, 删除记录")
|
||||||
|
pdfRepository.deleteDocument(doc.fileHash)
|
||||||
|
needFullScan = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (needFullScan || cachedDocs.isEmpty()) {
|
||||||
|
LogUtil.logDebug(TAG, "数据库不完整，执行全盘扫描...")
|
||||||
|
// 扫描应用私有目录（无需权限）
|
||||||
|
val privateFiles = FileUtils.scanPdfFiles(context)
|
||||||
|
LogUtil.logDebug(TAG, "📁应用私有目录找到: ${privateFiles.size} 个PDF文件")
|
||||||
|
// 扫描MediaStore（需要权限）
|
||||||
|
val mediaStoreFiles = FileUtils.scanPdfFilesFromMediaStore(context)
|
||||||
|
LogUtil.logDebug(TAG, "📱MediaStore找到: ${mediaStoreFiles.size} 个PDF文件")
|
||||||
|
// 合并并去重
|
||||||
|
val allFiles =
|
||||||
|
(privateFiles + mediaStoreFiles).distinctBy { it.absolutePath }
|
||||||
|
LogUtil.logDebug(TAG, "📊总计扫描到: ${allFiles.size} 个PDF文件")
|
||||||
|
// 处理每个PDF文件
|
||||||
|
allFiles.forEachIndexed { index, file ->
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG,
|
||||||
|
"🔄处理文件 ${index + 1}/${allFiles.size}: ${file.name} - ${file.absolutePath}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if (FileUtils.isPdfFile(file)) {
|
||||||
|
val fileHash = FileUtils.calculateFileHash(file.absolutePath)
|
||||||
|
LogUtil.logDebug(TAG, "🔑文件哈希: $fileHash")
|
||||||
|
|
||||||
|
if (fileHash != null) {
|
||||||
|
val existingDoc =
|
||||||
|
pdfRepository.getDocumentByPath(file.absolutePath)
|
||||||
|
|
||||||
|
if (existingDoc == null) {
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, "🆕发现新PDF文件: ${file.name}"
|
||||||
|
)
|
||||||
|
var thumbnailPath: String? = null
|
||||||
|
val isPassword = isPdfEncrypted(file)
|
||||||
|
LogUtil.logDebug(TAG, "isPassword->${isPassword}")
|
||||||
|
if (!isPassword) {//没有密码的情况下才去获取缩略图
|
||||||
|
thumbnailPath = generateThumbnail(context, file) ?: ""
|
||||||
|
}
|
||||||
|
LogUtil.logDebug(TAG, "thumbnailPath->${thumbnailPath}")
|
||||||
|
val metadata =
|
||||||
|
PdfMetadataExtractor.extractMetadata(file.absolutePath)
|
||||||
|
val document = PdfDocumentEntity(
|
||||||
|
fileHash = fileHash,
|
||||||
|
filePath = file.absolutePath,
|
||||||
|
fileName = file.name,
|
||||||
|
fileSize = file.length(),
|
||||||
|
lastModified = file.lastModified(),
|
||||||
|
pageCount = metadata?.pageCount ?: 0,
|
||||||
|
thumbnailPath = thumbnailPath,
|
||||||
|
metadataTitle = metadata?.title,
|
||||||
|
metadataAuthor = metadata?.author,
|
||||||
|
metadataSubject = metadata?.subject,
|
||||||
|
metadataKeywords = metadata?.keywords,
|
||||||
|
metadataCreationDate = metadata?.creationDate?.time,
|
||||||
|
metadataModificationDate = metadata?.modificationDate?.time,
|
||||||
|
isPassword = isPassword
|
||||||
|
)
|
||||||
|
pdfRepository.insertOrUpdateDocument(document)
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, " ✅ 已保存到数据库: ${file.name}"
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
LogUtil.logDebug(TAG, " 📋 文件已存在: ${file.name}")
|
||||||
|
// 🔹 文件已存在，检查是否需要更新
|
||||||
|
var needUpdate = false
|
||||||
|
var updatedDoc = existingDoc.copy()
|
||||||
|
|
||||||
|
// 路径/修改时间更新
|
||||||
|
if (existingDoc.filePath != file.absolutePath || existingDoc.lastModified != file.lastModified()) {
|
||||||
|
LogUtil.logDebug(TAG, "✅ 路径/修改时间需要更新")
|
||||||
|
updatedDoc = updatedDoc.copy(
|
||||||
|
filePath = file.absolutePath,
|
||||||
|
lastModified = file.lastModified()
|
||||||
|
)
|
||||||
needUpdate = true
|
needUpdate = true
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
updatedDoc = updatedDoc.copy(thumbnailPath = null)
|
|
||||||
needUpdate = true
|
|
||||||
}
|
|
||||||
|
|
||||||
// 执行更新
|
// 是否加密更新
|
||||||
if (needUpdate) {
|
val currentIsPassword = isPdfEncrypted(file)
|
||||||
pdfRepository.insertOrUpdateDocument(updatedDoc)
|
if (existingDoc.isPassword != currentIsPassword) {
|
||||||
LogUtil.logDebug(
|
LogUtil.logDebug(TAG, "✅ 密码状态需要更新")
|
||||||
TAG, "✅ 数据库已更新: ${file.name}"
|
updatedDoc =
|
||||||
)
|
updatedDoc.copy(isPassword = currentIsPassword)
|
||||||
} else {
|
needUpdate = true
|
||||||
LogUtil.logDebug(
|
}
|
||||||
TAG, "⏩ 无需更新: ${file.name}"
|
|
||||||
)
|
if (!currentIsPassword) {
|
||||||
|
// 如果不是加密 PDF，再生成缩略图
|
||||||
|
val newThumbnail = generateThumbnail(context, file)
|
||||||
|
if (existingDoc.thumbnailPath != newThumbnail) {
|
||||||
|
LogUtil.logDebug(TAG, "✅ 缩略图需要更新")
|
||||||
|
updatedDoc =
|
||||||
|
updatedDoc.copy(thumbnailPath = newThumbnail)
|
||||||
|
needUpdate = true
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
updatedDoc = updatedDoc.copy(thumbnailPath = null)
|
||||||
|
needUpdate = true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 执行更新
|
||||||
|
if (needUpdate) {
|
||||||
|
pdfRepository.insertOrUpdateDocument(updatedDoc)
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, "✅ 数据库已更新: ${file.name}"
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, "⏩ 无需更新: ${file.name}"
|
||||||
|
)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// 打印数据库中的总记录数
|
||||||
|
pdfRepository.getAllDocumentsOnce().forEach { doc ->
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, " 📖 ${doc.fileName} - ${doc.pageCount}页 - ${
|
||||||
|
FileUtils.formatFileSize(
|
||||||
|
doc.fileSize
|
||||||
|
)
|
||||||
|
} - ${doc.thumbnailPath}"
|
||||||
|
)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// 打印数据库中的总记录数
|
// 标记扫描完成
|
||||||
pdfRepository.getAllDocumentsOnce().forEach { doc ->
|
ScanManager.markScanComplete(context)
|
||||||
LogUtil.logDebug(
|
val lastScanTime = ScanManager.getLastScanTime(context)
|
||||||
TAG, " 📖 ${doc.fileName} - ${doc.pageCount}页 - ${
|
LogUtil.logDebug(
|
||||||
FileUtils.formatFileSize(
|
TAG, "✅ 扫描完成，记录时间: ${java.util.Date(lastScanTime)}"
|
||||||
doc.fileSize
|
)
|
||||||
)
|
var string = if (needFullScan) {
|
||||||
} - ${doc.thumbnailPath}"
|
"全盘扫描"
|
||||||
)
|
} else if (cachedDocs.isEmpty()) {
|
||||||
|
"数据库没有值，第一次全盘扫描"
|
||||||
|
} else {
|
||||||
|
"快速扫描"
|
||||||
}
|
}
|
||||||
|
// 计算扫描耗时
|
||||||
|
val scannerTime = System.currentTimeMillis() - startScannerTime
|
||||||
|
LogUtil.logDebug(
|
||||||
|
TAG, "$string 本次扫描耗时: $scannerTime ms (${scannerTime / 1000.0} 秒)"
|
||||||
|
)
|
||||||
|
PDFReaderApplication.isNeedFullScan = false
|
||||||
|
callback.invoke(true)
|
||||||
|
} catch (e: Exception) {
|
||||||
|
Log.e(TAG, "❌ 扫描出错: ${e.message}", e)
|
||||||
|
callback.invoke(false)
|
||||||
}
|
}
|
||||||
// 标记扫描完成
|
|
||||||
ScanManager.markScanComplete(context)
|
|
||||||
val lastScanTime = ScanManager.getLastScanTime(context)
|
|
||||||
LogUtil.logDebug(
|
|
||||||
TAG, "✅ 扫描完成，记录时间: ${java.util.Date(lastScanTime)}"
|
|
||||||
)
|
|
||||||
var string = if (needFullScan) {
|
|
||||||
"全盘扫描"
|
|
||||||
} else if (cachedDocs.isEmpty()) {
|
|
||||||
"数据库没有值，第一次全盘扫描"
|
|
||||||
} else {
|
|
||||||
"快速扫描"
|
|
||||||
}
|
|
||||||
// 计算扫描耗时
|
|
||||||
val scannerTime = System.currentTimeMillis() - startScannerTime
|
|
||||||
LogUtil.logDebug(
|
|
||||||
TAG,
|
|
||||||
"$string 本次扫描耗时: $scannerTime ms (${scannerTime / 1000.0} 秒)"
|
|
||||||
)
|
|
||||||
PDFReaderApplication.isNeedFullScan = false
|
|
||||||
callback.invoke(true)
|
|
||||||
} catch (e: Exception) {
|
|
||||||
Log.e(TAG, "❌ 扫描出错: ${e.message}", e)
|
|
||||||
callback.invoke(false)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user