網(wǎng)上雖然有各種實例鸳吸，可在實際使用中，存在各種問題速勇，同時對錯誤并沒有完善的處理晌砾。所以經(jīng)過反復(fù)的調(diào)試，成功后烦磁，在此做個知識總結(jié)养匈。

需求

• 開發(fā)語言: Typescript
• 在 Electron Render 線程中調(diào)用
• 用 Promise 方式實現(xiàn)
• 下載用 pipe 方式，更好的支持大文件
• 上傳支持附加請求參數(shù)
• 上傳支持附加頭部信息
• 完善的錯誤捕獲邏輯都伪，拋出統(tǒng)一的錯誤類型呕乎，方便捕獲處理

依賴庫

# For download
yarn add request

# For upload
yarn add form-data

邏輯代碼

• transfer.ts

import FormData from 'form-data/lib/form_data';
import request from 'request';
import * as fs from 'fs';
import { IncomingMessage } from 'http';

export function TransferError(message: string = '', code: number = -1) {
  this.name = 'TransferError';
  this.message = message;
  this.code = code;
}
TransferError.prototype = Error.prototype;

// Transfer file between local and remote.
class Transfer {
  debug: Function | null = null; // Example: transfer.debug = console.log

  // Upload file to remote.
  upload(
    url: string,
    filePath: string,
    params: any = {},
    addHeaders: any = {}
  ) {
    return new Promise((resolve, reject) => {
      const stream = fs.createReadStream(filePath);
      stream.on('error', (e: any) => reject(new TransferError(e)));

      const form = new FormData();
      form.append('file', stream);
      for (const k in params) {
        form.append(k, params[k]);
      }

      const u = new URL(url);
      const data = {
        protocol: u.protocol,
        host: u.hostname,
        port: u.port,
        path: u.pathname,
        headers: {},
      };
      for (const k in addHeaders) {
        data.headers[k] = addHeaders[k];
      }

      if (this.debug) {
        this.debug(`[transfer] upload - data: ${JSON.stringify(data)}`);
      }

      form
        .submit(data, (e: any, resp: IncomingMessage) => {
          if (e || resp.statusCode !== 200) {
            if (resp) {
              reject(new TransferError(resp.statusMessage, resp.statusCode));
            } else {
              reject(new TransferError(e));
            }
          } else {
            resolve(resp.read());
          }
        })
        .on('error', (e: any) => {
          reject(new TransferError(e));
        });
    });
  }

  // Download file from remote.
  download(url: string, destPath: string) {
    return new Promise((resolve, reject) => {
      const r = request(url);
      r.on('response', (resp: IncomingMessage) => {
        if (resp.statusCode === 200) {
          const writer = fs.createWriteStream(destPath);
          r.pipe(writer)
            .on('error', (e: any) => reject(new TransferError(e)))
            .on('finish', resolve);
        } else {
          reject(new TransferError(resp.statusMessage, resp.statusCode));
        }
      }).on('error', (e: any) => reject(new TransferError(e)));
    });
  }
}

export const transfer = new Transfer();

測試代碼

• transfer_test.ts

import { transfer, TransferError } from './transfer';

async function upload() {
  const url = 'http://192.168.10.11:8081/upload';
  const filePath = '/Users/mac/Desktop/upload.jpg';
  const openid = 'cd2b61ce043f6eed3f9d86aa6e670dba91a436ac';
  const token =
    'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VyIjp7ImlkIjoxLCJvcGVuaWQiOiJlZjE2NDU3YS0wNDgxLTRlYzUtOTQzMS0yZDhlYTA3MDdkODkifSwiZXhwIjoxNTc2NTAwNTA4LCJpc3MiOiJub3RlZG93bi5hdXRoIn0.86IwjCon8F3s5Pngn6gK5uuPrxEUmgFvzpkm_Jfhrzc';
  const addHeaders = {
    Authorization: `token ${token}`,
  };

  transfer.debug = console.log;
  try {
    const resp = await transfer.upload(url, filePath, { openid }, addHeaders);
    console.log('upload success', resp);
  } catch (e) {
    if (e instanceof TransferError) {
      console.log(e);
    } else {
      console.error(e);
    }
  }
}

async function download() {
  const url =
    'http://192.168.10.11:8081/static/86500d81f4a0921e27b4fbd89ce1eb1cc164109f.jpg';
  const filePath = '/Users/mac/Desktop/download.jpg';

  try {
    const resp = await transfer.download(url, filePath);
    console.log('download success', resp);
  } catch (e) {
    if (e instanceof TransferError) {
      console.log(e);
    } else {
      console.error(e);
    }
  }
}

export default function test() {
  upload();
  download();
}