feat(server): generate checksum for previous uploaded assets (#558)

* feat(server): generate checksum for previous uploaded assets

* fix(server): typo
This commit is contained in:
Thanh Pham 2022-09-02 20:32:21 +07:00 committed by GitHub
parent 0799aa2c72
commit c76f7804ab
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 96 additions and 10 deletions

View file

@ -0,0 +1,69 @@
import { AssetEntity } from '@app/database/entities/asset.entity';
import { generateChecksumQueueName } from '@app/job';
import { Process, Processor } from '@nestjs/bull';
import { Logger } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { createHash } from 'node:crypto';
import fs from 'node:fs';
import { IsNull, Repository } from 'typeorm';
// TODO: just temporary task to generate previous uploaded assets.
@Processor(generateChecksumQueueName)
export class GenerateChecksumProcessor {
constructor(
@InjectRepository(AssetEntity)
private assetRepository: Repository<AssetEntity>,
) {}
@Process()
async generateChecksum() {
let hasNext = true;
let pageSize = 200;
let offset = 0;
while (hasNext) {
const assets = await this.assetRepository.find({
where: {
checksum: IsNull()
},
skip: offset,
take: pageSize,
});
if (!assets?.length) {
hasNext = false; // avoid using break
} else {
for (const asset of assets) {
try {
await this.generateAssetChecksum(asset);
} catch (err: any) {
Logger.error(`Error generate checksum ${err}`);
}
}
if (assets.length < pageSize) {
hasNext = false;
} else {
offset += pageSize;
}
}
}
}
private async generateAssetChecksum(asset: AssetEntity) {
if (!asset.originalPath) return;
if (!fs.existsSync(asset.originalPath)) return;
const fileReadStream = fs.createReadStream(asset.originalPath);
const sha1Hash = createHash('sha1');
const deferred = new Promise<Buffer>((resolve, reject) => {
sha1Hash.once('error', (err) => reject(err));
sha1Hash.once('finish', () => resolve(sha1Hash.read()));
});
fileReadStream.pipe(sha1Hash);
const checksum = await deferred;
await this.assetRepository.update(asset.id, { checksum });
}
}