feat(front): enhance file hash calculation by introducing engine selection for large files using native or wasm methods

This commit is contained in:
keven1024
2026-04-08 23:45:34 +08:00
parent f1dc65b1d0
commit ae2fbcc216
3 changed files with 27 additions and 37 deletions

View File

@@ -137,11 +137,14 @@ watchEffect(async () => {
}
})
const LARGE_FILE_THRESHOLD = 500 * 1024 * 1024 // 500 MB
const handleHash = async (fileId: string) => {
const uploadfile = uploadfiles.value.find((item) => item.fileId === fileId)
if (!uploadfile?.file) return
uploadfile.procressType = 'hash'
const res = await asyncWorker(calcFileHashWorker, { data: { file: uploadfile.file } })
const engine = uploadfile.file.size >= LARGE_FILE_THRESHOLD ? 'wasm' : 'native'
const res = await asyncWorker(calcFileHashWorker, { data: { file: uploadfile.file, engine } })
const { hash } = res?.data || {}
uploadfile.hash = hash
}

View File

@@ -1,47 +1,34 @@
import { noop } from 'lodash-es'
import { md5 } from 'js-md5'
import { createSHA1 } from 'hash-wasm'
interface CalcFileHashProps {
file: File
onProgress?: (current: number) => void
chunkSize?: number
engine?: 'native' | 'wasm'
}
const calcFileHash = async (props: CalcFileHashProps) => {
const { file, onProgress = noop, chunkSize = 100 } = props || {}
const blob = await file.arrayBuffer()
const hash = md5(blob)
return hash
// const finalChunkSize = chunkSize * 1024 * 1024;
// const chunks = Math.ceil(file.size / finalChunkSize);
// const spark = new SparkMD5.ArrayBuffer(); // 使用 SparkMD5 增量计算哈希
// const fileReader = new FileReader();
const { file, onProgress = noop, chunkSize = 100, engine = 'native' } = props || {}
// const readChunk = (start: number): Promise<ArrayBuffer> => {
// return new Promise((resolve, reject) => {
// const chunk = file.slice(start, Math.min(start + finalChunkSize, file.size));
// fileReader.onload = (e) => resolve(e.target?.result as ArrayBuffer);
// fileReader.onerror = reject;
// fileReader.readAsArrayBuffer(chunk);
// });
// };
if (engine === 'native') {
const buffer = await file.arrayBuffer()
const hashBuffer = await crypto.subtle.digest('SHA-1', buffer)
return Array.from(new Uint8Array(hashBuffer))
.map((b) => b.toString(16).padStart(2, '0'))
.join('')
}
// try {
// const progressCallback = (current: number) => {
// const percentage = Math.round((current / chunks) * 100);
// onProgress(percentage);
// };
// for (let i = 0; i < chunks; i++) {
// const chunk = await readChunk(i * chunkSize);
// spark.append(chunk);
// progressCallback(i + 1);
// }
// return spark.end();
// } catch (error) {
// throw error;
// }
const chunkBytes = chunkSize * 1024 * 1024
const hasher = await createSHA1()
let offset = 0
while (offset < file.size) {
const buffer = await file.slice(offset, offset + chunkBytes).arrayBuffer()
hasher.update(new Uint8Array(buffer))
offset += chunkBytes
onProgress(Math.min(offset, file.size) / file.size)
}
return hasher.digest('hex')
}
export default calcFileHash

View File

@@ -1,8 +1,8 @@
import calcFileHash from './calcFileHash'
// 监听主线程消息
self.onmessage = async (e: MessageEvent<{ file: File }>) => {
const { file } = e.data || {}
const hash = await calcFileHash({ file })
self.onmessage = async (e: MessageEvent<{ file: File; engine?: 'native' | 'wasm' }>) => {
const { file, engine } = e.data || {}
const hash = await calcFileHash({ file, engine })
self.postMessage({ hash })
}