2476 - downloadImages() method refactor #528
|
@ -0,0 +1,2 @@
|
||||||
|
ALTER TABLE `vn`.`itemImageQueue`
|
||||||
|
ADD attempts INT default 0 NULL AFTER error;
|
|
@ -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}`);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
|
@ -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": {
|
||||||
|
|
Loading…
Reference in New Issue