Загрузка изображений с помощью node.js
Я пытаюсь написать script для загрузки изображений с помощью node.js. Это то, что у меня есть до сих пор:
var maxLength = 10 // 10mb
var download = function(uri, callback) {
http.request(uri)
.on('response', function(res) {
if (res.headers['content-length'] > maxLength*1024*1024) {
callback(new Error('Image too large.'))
} else if (!~[200, 304].indexOf(res.statusCode)) {
callback(new Error('Received an invalid status code.'))
} else if (!res.headers['content-type'].match(/image/)) {
callback(new Error('Not an image.'))
} else {
var body = ''
res.setEncoding('binary')
res
.on('error', function(err) {
callback(err)
})
.on('data', function(chunk) {
body += chunk
})
.on('end', function() {
// What about Windows?!
var path = '/tmp/' + Math.random().toString().split('.').pop()
fs.writeFile(path, body, 'binary', function(err) {
callback(err, path)
})
})
}
})
.on('error', function(err) {
callback(err)
})
.end();
}
Я, однако, хочу сделать это более надежным:
- Есть ли библиотеки, которые делают это и делают это лучше?
- Есть ли вероятность, что заголовки ответов лежат (о длине, о типе содержимого)?
- Есть ли какие-либо другие коды состояния, о которых я должен заботиться? Должен ли я беспокоиться о переадресации?
- Я думаю, что я где-то читал, что кодировка
binary
будет устаревать. Что мне делать тогда?
- Как я могу заставить это работать с окнами?
- Любые другие способы сделать этот script лучше?
Почему: для функции, похожей на imgur, где пользователи могут указывать мне URL-адрес, я загружаю это изображение и повторяю изображение в нескольких размерах.
Ответы
Ответ 1
Я бы предложил использовать модуль запроса. Загрузка файла так же проста, как следующий код:
var fs = require('fs'),
request = require('request');
var download = function(uri, filename, callback){
request.head(uri, function(err, res, body){
console.log('content-type:', res.headers['content-type']);
console.log('content-length:', res.headers['content-length']);
request(uri).pipe(fs.createWriteStream(filename)).on('close', callback);
});
};
download('https://www.google.com/images/srpr/logo3w.png', 'google.png', function(){
console.log('done');
});
Ответ 2
Я столкнулся с этой проблемой несколько дней назад, для чистого ответа NodeJS я бы предложил использовать Stream для объединения кусков вместе.
var http = require('http'),
Stream = require('stream').Transform,
fs = require('fs');
var url = 'http://www.google.com/images/srpr/logo11w.png';
http.request(url, function(response) {
var data = new Stream();
response.on('data', function(chunk) {
data.push(chunk);
});
response.on('end', function() {
fs.writeFileSync('image.png', data.read());
});
}).end();
Самые новые версии Node не будут работать с бинарными строками, поэтому слияние блоков со строками не является хорошей идеей при работе с двоичными данными.
* Просто будьте осторожны при использовании 'data.read()', он очистит поток для следующей операции read(). Если вы хотите использовать его более одного раза, сохраните его где-нибудь.
Ответ 3
Вы можете использовать AXIOS (а обещание -based HTTP клиент для Node.js) для загрузки изображений в порядке по своему выбору в качестве асинхронной среды:
npm i axios
Затем вы можете использовать следующий базовый пример, чтобы начать загрузку изображений:
const fs = require('fs');
const axios = require('axios');
/* ============================================================
Function: Download Image
============================================================ */
const download_image = (url, image_path) =>
axios({
url,
responseType: 'stream',
}).then(
response =>
new Promise((resolve, reject) => {
response.data
.pipe(fs.createWriteStream(image_path))
.on('finish', () => resolve())
.on('error', e => reject(e));
}),
);
/* ============================================================
Download Images in Order
============================================================ */
(async () => {
let example_image_1 = await download_image('https://example.com/test-1.png', 'example-1.png');
console.log(example_image_1.status); // true
console.log(example_image_1.error); // ''
let example_image_2 = await download_image('https://example.com/does-not-exist.png', 'example-2.png');
console.log(example_image_2.status); // false
console.log(example_image_2.error); // 'Error: Request failed with status code 404'
let example_image_3 = await download_image('https://example.com/test-3.png', 'example-3.png');
console.log(example_image_3.status); // true
console.log(example_image_3.error); // ''
})();
Ответ 4
если вы хотите выполнить загрузку, попробуйте это:
var fs = require('fs');
var request = require('request');
var progress = require('request-progress');
module.exports = function (uri, path, onProgress, onResponse, onError, onEnd) {
progress(request(uri))
.on('progress', onProgress)
.on('response', onResponse)
.on('error', onError)
.on('end', onEnd)
.pipe(fs.createWriteStream(path))
};
как использовать:
var download = require('../lib/download');
download("https://www.google.com/images/branding/googlelogo/2x/googlelogo_color_150x54dp.png", "~/download/logo.png", function (state) {
console.log("progress", state);
}, function (response) {
console.log("status code", response.statusCode);
}, function (error) {
console.log("error", error);
}, function () {
console.log("done");
});
Примечание: вы должны установить модули запроса и запроса-запроса, используя:
npm install request request-progress --save
Ответ 5
Основываясь на вышеизложенном, если кому-то нужно обрабатывать ошибки в потоках записи/чтения, я использовал эту версию. Обратите внимание на stream.read()
в случае ошибки записи, поэтому нам нужно закончить чтение и запуск close
в потоке чтения.
var download = function(uri, filename, callback){
request.head(uri, function(err, res, body){
if (err) callback(err, filename);
else {
var stream = request(uri);
stream.pipe(
fs.createWriteStream(filename)
.on('error', function(err){
callback(error, filename);
stream.read();
})
)
.on('close', function() {
callback(null, filename);
});
}
});
};
Ответ 6
var fs = require('fs'),
http = require('http'),
https = require('https');
var Stream = require('stream').Transform;
var downloadImageToUrl = (url, filename, callback) => {
var client = http;
if (url.toString().indexOf("https") === 0){
client = https;
}
client.request(url, function(response) {
var data = new Stream();
response.on('data', function(chunk) {
data.push(chunk);
});
response.on('end', function() {
fs.writeFileSync(filename, data.read());
});
}).end();
};
downloadImageToUrl('https://www.google.com/images/srpr/logo11w.png', 'public/uploads/users/abc.jpg');
Ответ 7
Это продолжение ответа Цезария. Если вы хотите загрузить его в определенный каталог, используйте это. Также используйте const вместо var. Так безопасно.
const fs = require('fs');
const request = require('request');
var download = function(uri, filename, callback){
request.head(uri, function(err, res, body){
request(uri).pipe(fs.createWriteStream(filename)).on('close', callback);
});
};
download('https://www.google.com/images/srpr/logo3w.png', './images/google.png', function(){
console.log('done');
});