node實(shí)現(xiàn)分片下載的示例代碼

更新時(shí)間：2018年10月17日 09:55:45 作者：keller35

這篇文章主要介紹了node實(shí)現(xiàn)分片下載的示例代碼，使用場(chǎng)景包括基于瀏覽器的流文件片段傳輸、基于客戶端的分片下載等。感興趣的小伙伴們可以參考一下

本文基于http Range Requests協(xié)議，實(shí)現(xiàn)了分片下載的功能。

使用場(chǎng)景包括基于瀏覽器的流文件片段傳輸、基于客戶端的分片下載等。

原理

http通過(guò)Range Requests相關(guān)的header，可以與服務(wù)器進(jìn)行協(xié)商，實(shí)現(xiàn)分部分的請(qǐng)求。

這里就不細(xì)說(shuō)具體協(xié)議內(nèi)容了，具體可以參考這兩篇文章，解釋的非常詳細(xì)：

下面貼一下實(shí)現(xiàn)過(guò)程。

服務(wù)端代碼

服務(wù)端用node實(shí)現(xiàn)：

app.use(async ctx => {
 const file = path.join(__dirname, `${PATH}${ctx.path}`);
 // 1、404檢查
 try {
  fs.accessSync(file);
 } catch (e) {
  return ctx.response.status = 404;
 }
 const method = ctx.request.method;
 const { size } = fs.statSync(file);
 // 2、響應(yīng)head請(qǐng)求，返回文件大小
 if ('HEAD' == method) {
  return ctx.set('Content-Length', size);
 }
 const range = ctx.headers['range'];
 // 3、通知瀏覽器可以進(jìn)行分部分請(qǐng)求
 if (!range) {
  return ctx.set('Accept-Ranges', 'bytes');
 }
 const { start, end } = getRange(range);
 // 4、檢查請(qǐng)求范圍
 if (start >= size || end >= size) {
  ctx.response.status = 416;
  return ctx.set('Content-Range', `bytes */${size}`);
 }
 // 5、206分部分響應(yīng)
 ctx.response.status = 206;
 ctx.set('Accept-Ranges', 'bytes');
 ctx.set('Content-Range', `bytes ${start}-${end ? end : size - 1}/${size}`);
 ctx.body = fs.createReadStream(file, { start, end });
});

app.listen(3000, () => console.log('partial content server start'));

function getRange(range) {
 var match = /bytes=([0-9]*)-([0-9]*)/.exec(range);
 const requestRange = {};
 if (match) {
  if (match[1]) requestRange.start = Number(match[1]);
  if (match[2]) requestRange.end = Number(match[2]);
 }
 return requestRange;
}

代碼實(shí)現(xiàn)的功能邏輯大致是：

對(duì)請(qǐng)求的資源做檢查，不存在則響應(yīng)404
對(duì)于HEAD請(qǐng)求，返回資源大小
如果GET請(qǐng)求沒(méi)有告知range，返回Content-Length，告知瀏覽器可以進(jìn)行分片請(qǐng)求
如果請(qǐng)求設(shè)置了range，則檢查range是否合法，不合法返回合法的rangge
一切正常，獲取文件range范圍部分，做流響應(yīng)

代碼很簡(jiǎn)單，把Range Requests協(xié)議對(duì)應(yīng)實(shí)現(xiàn)一遍就ok了，當(dāng)然這里沒(méi)有完全實(shí)現(xiàn)協(xié)議的內(nèi)容，但已經(jīng)滿足了這里演示的需求。

服務(wù)端代碼ok了，用一個(gè)瀏覽器的demo來(lái)檢驗(yàn)一下。

瀏覽器例子

現(xiàn)代瀏覽器基本都實(shí)現(xiàn)了Range Requests，這里用audio標(biāo)簽作為例子。

<html>
 <head>
  <title>分片流傳輸</title>
  <script type="text/javascript">
   function jump() {
    const player = document.getElementById('musicPlayer');
    // 從30s開(kāi)始播放
    player.currentTime = 30;
   }
  </script>
 </head>
 <body>
  <audio id="musicPlayer" src="http:127.0.0.1:3000/source.mp3" controls></audio>
  <button onclick="jump()">切到30s</button>
 </body>
</html>

最終的效果是這樣的：

對(duì)比兩張圖，當(dāng)html加載完成，瀏覽器自動(dòng)請(qǐng)求資源，此時(shí)header有Range: bytes=0-，表示從第0 byte開(kāi)始加載資源；當(dāng)點(diǎn)擊跳到30s處播放時(shí)，此時(shí)header變成了Range: bytes=3145728-。

同樣用這個(gè)服務(wù)端代碼，還可以實(shí)現(xiàn)一個(gè)客戶端，模擬一下分包下載。

node分包下載

這個(gè)例子演示了，對(duì)一個(gè)資源，并發(fā)的實(shí)現(xiàn)分部分的下載，然后再合并成一個(gè)文件。

這里也是用node實(shí)現(xiàn)：

import request from 'request';
import path from 'path';
import fs from 'fs';

const SINGLE = 1024 * 1000;
const SOURCE = 'http://127.0.0.1:3000/source.mp3';

request({
 method: 'HEAD',
 uri: SOURCE,
}, (err, res) => {
 if (err) return console.error(err);
 const file = path.join(__dirname, './download/source.mp3');
 try {
  fs.closeSync(fs.openSync(file, 'w'));
 } catch (err) {
  return console.error(err);
 }
 const size = Number(res.headers['content-length']);
 const length = parseInt(size / SINGLE);
 for (let i=0; i<length; i++) {
  let start = i * SINGLE;
  let end = i == length ? (i + 1) * SINGLE - 1 : size - 1;
  request({
   method: 'GET',
   uri: SOURCE,
   headers: {
    'range': `bytes=${start}-${end}`
   },
  }).on('response', (resp) => {
   const range = resp.headers['content-range'];
   const match = /bytes ([0-9]*)-([0-9]*)/.exec(range);
   start = match[1];
   end = match[2];
  }).pipe(fs.createWriteStream(file, {start, end}));
 }
});

代碼比較簡(jiǎn)單，就是開(kāi)啟多個(gè)http請(qǐng)求，并發(fā)的下載資源，然后根據(jù)響應(yīng)的content-range，寫(xiě)到文件的對(duì)應(yīng)位置。

參考文章：

https://tools.ietf.org/html/rfc7233
http://www.dbjr.com.cn/article/68284.htm

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: