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 Blob([data], { type: data.type }).slice(offset, offset + chunkSize) // 读取文件块 const buffer = await chunk.arrayBuffer() // 将文件块转为ArrayBuffer hash.update(buffer) // 更新文件的SHA256值 spark.append(buffer) // 更新文件的MD5值 offset += chunkSize // 更新偏移量 // 计算进度 const progress = Math.min(100, Math.round((offset / fileSize) * 100)) console.log(progress + '%') // 打印进度 } // 计算SHA256值和MD5值 const [ sha256Uint8Array32, // SHA256 Unit8Array(32) md5, // MD5 ] = await Promise.all([ hash.digest(), // 计算SHA256值 spark.end(), // 计算MD5值 ]) let obj = { md5, // MD5 sha256Int8Array32: Array.from(new Int8Array(sha256Uint8Array32)), // SHA256 Int8Array(32) } console.log(obj) // 打印结果对象 postMessage(obj) // 将结果对象发送给主线程 } 优化这段代码让我拿到文件的sha256值
时间: 2024-01-14 18:04:06 浏览: 307
这段代码的计算SHA256值的部分已经是使用AWS SDK提供的SHA256算法库,是比较高效和准确的,没有太多可以优化的地方。不过,可以进行以下两点优化:
1. 将每次更新SHA256值的数据从ArrayBuffer改为Uint8Array类型,因为Uint8Array比ArrayBuffer更适合用于进行二进制数据的处理和传输,可以减少类型转换的开销。
2. 每次读取文件块时,只需要获取该块的Uint8Array数据,而不需要再用Blob类型进行封装,这样可以减少一次类型转换的开销。
优化后的代码如下所示:
```
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 + '%'); // 打印进度
}
// 计算SHA256值和MD5值
const [ sha256Uint8Array32, md5 ] = await Promise.all([
hash.digest(), // 计算SHA256值
spark.end() // 计算MD5值
]);
const obj = {
md5, // MD5
sha256Int8Array32: Array.from(new Int8Array(sha256Uint8Array32)) // SHA256 Int8Array(32)
};
console.log(obj); // 打印结果对象
postMessage(obj); // 将结果对象发送给主线程
}
```
阅读全文