如何使用fetchEventSource實(shí)現(xiàn)sse流式請(qǐng)求

更新時(shí)間：2024年08月30日 08:40:58 作者：寫(xiě)代碼的拉克絲

這篇文章主要介紹了如何使用fetchEventSource實(shí)現(xiàn)sse流式請(qǐng)求問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

使用fetchEventSource實(shí)現(xiàn)sse流式請(qǐng)求

最近在項(xiàng)目開(kāi)發(fā)的過(guò)程中，需要實(shí)現(xiàn)一個(gè)功能

根據(jù)進(jìn)度條的進(jìn)度動(dòng)態(tài)實(shí)時(shí)加載獲取到的數(shù)據(jù)并展示出來(lái)

用到的是fetchEventSource

以下主要寫(xiě)下js的實(shí)現(xiàn)的過(guò)程（前端用到的ui組件是antd-design-vue1.78版本 vue2.0項(xiàng)目）

首先下載插件：

npm install --save @microsoft/fetch-event-source

<div class="progress_container" v-if="!progressDone">
        <div class="progress_row">
          <a-progress type="line" :percent="progressValue" status="active" />
        </div>
        <div class="progress_text">
          {{ progressTxt }}
        </div>
</div>


<script>
  import { fetchEventSource } from '@microsoft/fetch-event-source'
  import storage from 'store'
  import { ACCESS_TOKEN } from '@/store/mutation-types'
   export default {
        data () {
          return {
            speed: 30,
            progressValue: 0,//進(jìn)度條初始值
            progressTxt: '',//進(jìn)度條文字展示默認(rèn)值
            eventSource: null,
            abortController: null,
            checkMessage: [],
            checkResult: {},
            resultData: [],
            submitUnable: true,
            progressDone: false
          }
        },
      created () {
        this.connectSSE()
      },
      destroyed () {
          this.stopSSE()
      },
      methods: {
        connectSSE () {
          this.abortController = new AbortController() // 終止or結(jié)束會(huì)話，做了個(gè)全局的...
          const url = process.env.VUE_APP_API_BASE_URL + `自己的請(qǐng)求數(shù)據(jù)接口`
          const token = storage.get(ACCESS_TOKEN)
          // 開(kāi)始會(huì)話鏈接，也做了個(gè)全局的......
          this.eventSource = fetchEventSource(url, {
            method: 'POST',
            headers: {
              'Content-Type': 'application/json',
              [ACCESS_TOKEN]: token
            },
            // 傳參只能發(fā)送文本格式的數(shù)據(jù)
            body: JSON.stringify({
           
            }),
            signal: this.abortController.signal,
            onopen: (event) => {
              console.log(event, 'SSE在連接打開(kāi)時(shí)被調(diào)用')
            },
            onmessage: (event) => {
              // console.log('SSE會(huì)在通過(guò)事件源收到數(shù)據(jù)時(shí)觸發(fā)')
              // console.log('接口返回', JSON.parse(event.data))
              //我這里是判斷我請(qǐng)求回來(lái)的數(shù)據(jù)里是否包含'[DONE]'，他是數(shù)組里的最后一項(xiàng)，表示加載完成
              if (!event.data.includes('[DONE]')) {
                const data = JSON.parse(event.data)
                const text = data.message
                const percent = data.percent
                const message = data.message
                if (text === 'connect') {
                  // 連接成功，開(kāi)始檢測(cè)邏輯
                  this.progressDone = false
                  this.progressValue = 0
                  this.progressTxt = '數(shù)據(jù)檢查中，該過(guò)程可能需要一段時(shí)間，請(qǐng)勿關(guān)閉窗口'
                
                  this.getTurnCheck()
                 return false
                }
                //定時(shí)器，控制進(jìn)度條和文字的展示速度
                this.speed += 300
                this.timer = setTimeout(() => {
                  this.progressValue = Number(percent)
                  this.progressTxt = message
                  if (this.progressValue === 100 && !this.progressDone) {
                 
                    this.stopSSE()
                  }
                }, this.speed)
              }
              
              if (event.data.includes('[DONE]')) {
              }
              
            },
            onerror: () => {
              console.log('SSE拋出異常onerror')
              this.stopSSE()
            },
            // onclose函數(shù)是會(huì)話結(jié)束正常關(guān)閉觸發(fā)，但幾乎沒(méi)有做到這步，不執(zhí)行這個(gè)回調(diào)，不知道什么原因
            onclose: () => {
              console.log('onclose') 
            }
          })
        },
        // 停止SSE
        async stopSSE () {
          // 關(guān)閉業(yè)務(wù)，執(zhí)行關(guān)閉sse數(shù)據(jù)接口
          await this.getCloseSse()
          if (this.abortController) {
            this.timer && clearTimeout(this.timer)
            console.log('SSE結(jié)束會(huì)話了')
            this.abortController.abort() // 結(jié)束會(huì)話
          }
        },
        //檢查數(shù)據(jù)的接口
        async getTurnCheck () {
          const params = {
            
          }
      
        },
        //關(guān)閉sse接口
        async getCloseSse () {
          const params = {
            
          }
          const [err, res] = await to(closeSse(params))
          if (err) return false
        }

      },


    }
</script>