2476 - downloadImages() method refactor #528

Merged
carlosjr merged 4 commits from 2476-floramondo_refactor into dev 2021-01-28 16:03:35 +00:00
3 changed files with 70 additions and 35 deletions

View File

@ -0,0 +1,2 @@
ALTER TABLE `vn`.`itemImageQueue`
ADD attempts INT default 0 NULL AFTER error;

View File

@ -18,28 +18,40 @@ module.exports = Self => {
Self.downloadImages = async() => { Self.downloadImages = async() => {
const models = Self.app.models; const models = Self.app.models;
const container = await models.TempContainer.container('salix-image');
const tempPath = path.join(container.client.root, container.name);
const maxAttempts = 3;
try { const images = await Self.find({
const tempPath = path.join('/tmp/salix-image'); where: {attempts: {eq: maxAttempts}}
// Create temporary path
await fs.mkdir(tempPath, {recursive: true});
const timer = setInterval(async() => {
const image = await Self.findOne({
where: {error: null, url: {neq: null}}
}); });
// Exit loop for (let image of images) {
if (!image) return clearInterval(timer); const currentStamp = new Date().getTime();
const updatedStamp = image.updated.getTime();
const graceTime = Math.abs(currentStamp - updatedStamp);
const maxTTL = 3600 * 48 * 1000; // 48 hours in ms;
if (graceTime >= maxTTL)
await Self.destroyById(image.itemFk);
}
download();
async function download() {
const image = await Self.findOne({
where: {url: {neq: null}, attempts: {lt: maxAttempts}},
order: 'attempts, updated'
});
if (!image) return;
const srcFile = image.url.split('/').pop(); const srcFile = image.url.split('/').pop();
const fileName = srcFile.split('.')[0]; const dotIndex = srcFile.lastIndexOf('.');
const fileName = srcFile.substring(0, dotIndex);
const file = `${fileName}.png`; const file = `${fileName}.png`;
const filePath = path.join(tempPath, file); const filePath = path.join(tempPath, file);
const writeStream = fs.createWriteStream(filePath);
writeStream.on('open', () => {
https.get(image.url, async response => { https.get(image.url, async response => {
if (response.statusCode != 200) { if (response.statusCode != 200) {
const error = new Error(`Could not download the image. Status code ${response.statusCode}`); const error = new Error(`Could not download the image. Status code ${response.statusCode}`);
@ -47,10 +59,9 @@ module.exports = Self => {
return await errorHandler(image.itemFk, error, filePath); return await errorHandler(image.itemFk, error, filePath);
} }
const writeStream = fs.createWriteStream(filePath);
writeStream.on('open', () => {
response.pipe(writeStream); response.pipe(writeStream);
}).on('error', async error => {
await errorHandler(image.itemFk, error, filePath);
});
}); });
writeStream.on('error', async error => { writeStream.on('error', async error => {
@ -58,31 +69,44 @@ module.exports = Self => {
}); });
writeStream.on('finish', async function() { writeStream.on('finish', async function() {
writeStream.end();
});
writeStream.on('close', async function() {
try { try {
await models.Image.registerImage('catalog', filePath, fileName, image.itemFk); await models.Image.registerImage('catalog', filePath, fileName, image.itemFk);
await image.destroy(); await image.destroy();
download();
} catch (error) { } catch (error) {
await errorHandler(image.itemFk, error, filePath); await errorHandler(image.itemFk, error, filePath);
} }
}); });
}, 1000); }).on('error', async error => {
} catch (error) { await errorHandler(image.itemFk, error, filePath);
throw new Error('Try-catch error: ', error); });
} }
async function errorHandler(rowId, error, filePath) { async function errorHandler(rowId, error, filePath) {
try { try {
const row = await Self.findById(rowId); const row = await Self.findById(rowId);
if (!row) if (!row) return;
throw new Error(`Could not update due error ${error}`);
await row.updateAttribute('error', error); if (row.attempts < maxAttempts) {
await row.updateAttributes({
error: error,
attempts: row.attempts + 1,
updated: new Date()
});
}
if (filePath && fs.existsSync(filePath)) if (filePath && fs.existsSync(filePath))
await fs.unlink(filePath); await fs.unlink(filePath);
download();
} catch (err) { } catch (err) {
throw new Error(`ErrorHandler error: ${err}`); throw new Error(`Image download failed: ${err}`);
} }
} }
}; };

View File

@ -9,17 +9,26 @@
}, },
"properties": { "properties": {
"itemFk": { "itemFk": {
"type": "Number", "type": "number",
"id": true, "id": true,
"description": "Identifier" "description": "Identifier"
}, },
"url": { "url": {
"type": "String", "type": "string",
"required": true "required": true
}, },
"error": { "error": {
"type": "String", "type": "string",
"required": true "required": true
},
"attempts": {
"type": "number"
},
"created": {
"type": "date"
},
"updated": {
"type": "date"
} }
}, },
"relations": { "relations": {