import { Sha256 } from '@aws-crypto/sha256-js' // 导入SHA256算法 import SparkMD5 from 'spark-md5' // 导入MD5算法 onmessage = async function (event) { const data = event.data // 获取文件数据 // chunk方式读取文件 const chunkSize = 1024 * 1024 // 每次读取1MB const fileSize = data.size // 文件大小 let offset = 0 // 偏移量 const hash = new Sha256() // 创建SHA256实例 const spark = new SparkMD5.ArrayBuffer() // 创建MD5实例 while (offset < fileSize) { const chunk = new Uint8Array(await data.slice(offset, offset + chunkSize).arrayBuffer()) // 读取文件块 hash.update(chunk) // 更新文件的SHA256值 spark.append(chunk) // 更新文件的MD5值 offset += chunkSize // 更新偏移量 // 计算进度 const progress = Math.min(100, Math.round((offset / fileSize) * 100)) console.log(progress + '%') // 打印进度 } const [sha256Uint8Array32, md5] = await Promise.all([ hash.digest(), // SHA256 Unit8Array(32) spark.end(), // MD5 ]) const hashArray = Array.from(new Uint8Array(sha256Uint8Array32)) const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join('') const obj = { md5, // MD5 sha256: hashHex, // SHA256 Hex sha256Int8Array32: Array.from(new Int8Array(sha256Uint8Array32)), // SHA256 Int8Array(32) } console.log(obj) // 打印结果对象 postMessage(obj) // 将结果对象发送给主线程 } 这段代码有Bug 就是剩下的不够1M就不会算
时间: 2024-02-06 13:03:53 浏览: 201
是的,这段代码有一个Bug,即在最后一次读取文件块时,如果剩余的文件大小不到1MB,则不会计算SHA256和MD5值。
为了解决这个问题,我们可以在循环结束后,再读取一次剩余的文件块,并计算SHA256和MD5值。
以下是修复后的代码:
```
import { Sha256 } from '@aws-crypto/sha256-js'; // 导入SHA256算法
import SparkMD5 from 'spark-md5'; // 导入MD5算法
onmessage = async function (event) {
const data = event.data; // 获取文件数据
// chunk方式读取文件
const chunkSize = 1024 * 1024; // 每次读取1MB
const fileSize = data.size; // 文件大小
let offset = 0; // 偏移量
const hash = new Sha256(); // 创建SHA256实例
const spark = new SparkMD5.ArrayBuffer(); // 创建MD5实例
while (offset < fileSize) {
const chunk = new Uint8Array(await data.slice(offset, offset + chunkSize).arrayBuffer()); // 读取文件块
hash.update(chunk); // 更新文件的SHA256值
spark.append(chunk); // 更新文件的MD5值
offset += chunkSize; // 更新偏移量
// 计算进度
const progress = Math.min(100, Math.round((offset / fileSize) * 100));
console.log(progress + '%'); // 打印进度
}
// 读取剩余的文件块
if (offset < fileSize) {
const chunk = new Uint8Array(await data.slice(offset).arrayBuffer()); // 读取文件块
hash.update(chunk); // 更新文件的SHA256值
spark.append(chunk); // 更新文件的MD5值
}
const [sha256Uint8Array32, md5] = await Promise.all([
hash.digest(), // SHA256 Unit8Array(32)
spark.end(), // MD5
]);
const hashArray = Array.from(new Uint8Array(sha256Uint8Array32));
const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join('');
const obj = {
md5, // MD5
sha256: hashHex, // SHA256 Hex
sha256Int8Array32: Array.from(new Int8Array(sha256Uint8Array32)), // SHA256 Int8Array(32)
};
console.log(obj); // 打印结果对象
postMessage(obj); // 将结果对象发送给主线程
}
```
阅读全文