Egg.js构建一个stream流式接口服务

经常需要用到 stream 流式接口服务,比如：大文件下载、日志实时输出等等。本文将介绍如何使用Egg.js构建一个 stream 流式接口服务。

一、准备工作

目录结构：

txt 复制代码

app/
    /controller
        index.js
        test.txt
        test.sh

index.js 控制器
test.txt 测试文件，最好是20M以上的文件，这样才能看出流式返回的效果
test.sh 测试脚本，用于实时输出日志的测试脚本

二、流式文件处理

controller/index.js 文件内容如下：

js 复制代码

'use strict';

const Controller = require('egg').Controller;
const { createReadStream } = require('fs');
const { join } = require('path');

class HomeController extends Controller {

  async testStream() {
    const { ctx } = this;
    ctx.set('Content-Type', 'text/plain; charset=utf-8');
    const stream = createReadStream(join(__dirname, './test.txt'));
    ctx.body = stream;
  }

}

module.exports = HomeController;

三、流式日志处理

controller/index.js 文件内容如下：

js 复制代码

'use strict';

const Controller = require('egg').Controller;
const { createReadStream } = require('fs');
const { join } = require('path');
const { spawn } = require('child_process');

class HomeController extends Controller {

  async testStream() {
    ctx.set('Content-Type', 'text/plain; charset=utf-8');

    const shPath = join(__dirname, './test.sh');
    const stream = spawn('sh', [ shPath ]);
    ctx.body = stream.stdout;
  }

}

module.exports = HomeController;

controller/test.sh 文件内容如下：

bash 复制代码

#!/usr/bin/env sh

set -e

int=1
while(( $int<=10 ))
do
    echo $int
    sleep 2
    let "int++"
done

四、测试

前端使用 fetch 方法进行测试，为什么不用 axios ?因为 axios 是基于 XMLHttpRequest 的，不支持流式接口。具体实现请参考：前端实现 stream 流式请求

欢迎访问：天问博客