重生之我在NestJS中使用EventStream

ops/2024/9/23 3:25:01/

有一个需求是需要长连接等待后台的返回，我们使用的EventStream，在NestJS中使用很简单，框架基本上已经封装好了

0. 如果没有创建项目的，可以先创建一个项目，创建项目的直接跳转到下一个步骤去

全局安装 nest: npm install @nestjs/cli global
创建项目: nest new nest-demo-app

1. 在NestJS中的代码

在 app.constroller.ts 中新建一个方法，如下：

javascript">import { Observable } from 'rxjs';
import { Controller, Sse } from '@nestjs/common';@Controller()
class AppController {@Sse('test/event-stream') // 这个是访问的链接地址 http://localhost:3000/test/event-streamtestEventStream() {let count = 0;return new Observable(observer => {// 调用 observer.next() 这个方法向前端推送消息setInterval(() => {count++;observer.next({ data: new Date().toLocalTimeString() });}, 3000);// 调用 observer.complete() 这个方法，表示结束本次推送if (count > 5) {observer.complete();// 或者调用 observer.unsubscribe() 也可以结束推送}});}
}

2. 在前端调用这个接口

javascript">const stream = new EventSource('http://localhost:3000/test/event-stream')
let count = 0
stream.onopen = () => {console.log('连接打开');
}
stream.onmessage = e => {console.log(e.data); // 打印获取到的数据，也就是 NestJS 中 observer.next() 中发送的数据if (count++ >= 5) {stream.close(); // 主动关闭连接}
}
stream.onerror = e => {console.error(e); // 打印错误日志
}
stream.onclose = () => {console.log('连接关闭'); // 关闭连接 
}