前端http请求后端，后端SSE推送流

前端请求后端实时推送流给前端
直接上后端Java代码：

controller层代码：

@PostMapping(value = "/chat", produces = "text/event-stream;charset=UTF-8")
public void chat(@RequestBody AiDialogueVo aiDialogueVo, HttpServletResponse rp) {
aiDialogueVo.setUserId(getUserId());
aiDialogueVo.setMessageId(String.valueOf(new SnowFlake(4, 4).nextId()));
fastGptService.chat(aiDialogueVo,rp);
}

实现类代码：

@Override
public void chat(AiDialogueVo aiDialogueVo, HttpServletResponse rp) {
FastUserApp fastUserApp = fastUserAppMapper.selectFastUserAppByUserId(aiDialogueVo.getUserId());
try {
aiDialogueVo.getReqData().set("chatId",fastUserApp.getMongoAppId());
SSEListener sseListener = new SSEListener(aiDialogueVo, rp);
ExecuteSSEUtil.executeSSE(url,"Authorization",
fastUserApp .getToken().replaceAll("\r|\n", ""),
sseListener, JSON.toJSONString(aiDialogueVo.getReqData()));
} catch (Exception e) {
e.printStackTrace();
}
}

工具类ExecuteSSEUtil：

import lombok.extern.slf4j.Slf4j;
import okhttp3.MediaType;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.sse.EventSource;
import okhttp3.sse.EventSources;

@Slf4j
public class ExecuteSSEUtil {

public static void executeSSE(String url, String reqKey, String authToken, SSEListener eventSourceListener, String body) throws Exception {
RequestBody formBody = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), body);
Request.Builder requestBuilder = new Request.Builder();
requestBuilder.addHeader(reqKey, authToken);
requestBuilder.addHeader("Content-Type", "application/json");
Request request = requestBuilder.url(url).post(formBody).build();
EventSource.Factory factory = EventSources.createFactory(OkHttpUtil.getInstance());
//创建事件
factory.newEventSource(request, eventSourceListener);
eventSourceListener.getCountDownLatch().await();
}
}

工具类OkHttpUtil：

import okhttp3.ConnectionPool;
import okhttp3.OkHttpClient;

import java.net.Proxy;
import java.util.concurrent.TimeUnit;

public class OkHttpUtil {
private static OkHttpClient okHttpClient;

public static ConnectionPool connectionPool = new ConnectionPool(10, 5, TimeUnit.MINUTES);

public static OkHttpClient getInstance() {
if (okHttpClient == null) { //加同步安全
synchronized (OkHttpClient.class) {
if (okHttpClient == null) { //okhttp可以缓存数据....指定缓存路径
okHttpClient = new OkHttpClient.Builder()//构建器
.proxy(Proxy.NO_PROXY) //来屏蔽系统代理
.connectionPool(connectionPool)
.connectTimeout(6000, TimeUnit.SECONDS)//连接超时
.writeTimeout(6000, TimeUnit.SECONDS)//写入超时
.readTimeout(6000, TimeUnit.SECONDS)//读取超时
.build();
okHttpClient.dispatcher().setMaxRequestsPerHost(2000);
okHttpClient.dispatcher().setMaxRequests(2000);
}
}
}
return okHttpClient;
}
}

监听：

package com.yk.common.sse;


import com.alibaba.fastjson2.JSON;
import com.yk.common.sse.vo.AiDialogueVo;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import okhttp3.Response;
import okhttp3.sse.EventSource;
import okhttp3.sse.EventSourceListener;
import javax.servlet.http.HttpServletResponse;
import java.util.concurrent.CountDownLatch;

@Slf4j
@Data
public class SSEListener extends EventSourceListener {

private CountDownLatch countDownLatch = new CountDownLatch(1);

private AiDialogueVo aiDialogueVo;

private HttpServletResponse rp;

private StringBuffer output = new StringBuffer();

public SSEListener(AiDialogueVo aiDialogueVo, HttpServletResponse response) {
this.aiDialogueVo = aiDialogueVo;
this.rp = response;
}

/**
* {@inheritDoc}
* 建立sse连接
*/
@Override
public void onOpen(final EventSource eventSource, final Response
response) {
if (rp != null) {
rp.setContentType("text/event-stream");
rp.setCharacterEncoding("UTF-8");
rp.setStatus(200);
log.info("建立sse连接..." + JSON.toJSONString(aiDialogueVo));
} else {
log.info("客户端非sse推送" + JSON.toJSONString(aiDialogueVo));
}
}


/**
* 事件
*
* @param eventSource
* @param id
* @param type
* @param data
*/
@Override
public void onEvent(EventSource eventSource, String id, String type, String data) {
try {
output.append(data);
if ("finish".equals(type)) {
log.info("请求结束{} {}", aiDialogueVo.getEngineId(), output.toString());
}
if ("error".equals(type)) {
log.info("{}: {}source {}", aiDialogueVo.getEngineId(), data, JSON.toJSONString(aiDialogueVo));
}
if (rp != null) {
// System.out.println("data:" + data);
if ("\n".equals(data)) {
rp.getWriter().write("event:" + type + "\n");
rp.getWriter().write("id:" + aiDialogueVo.getEngineId() + "\n");
rp.getWriter().write("data:\n\n");
rp.getWriter().flush();
} else {
String[] dataArr = data.split("\\n");
for (int i = 0; i < dataArr.length; i++) {

if (i == 0) {
rp.getWriter().write("event:" + type + "\n");
rp.getWriter().write("id:" + aiDialogueVo.getMessageId() + "\n");
}
if (i == dataArr.length - 1) {
rp.getWriter().write("data:" + dataArr[i]+ "\n\n");
rp.getWriter().flush();
} else {
rp.getWriter().write("data:" + dataArr[i]+ "\n");
rp.getWriter().flush();
}
}
}
}
} catch (Exception e) {
log.error("消息错误[" + JSON.toJSONString(aiDialogueVo) + "]", e);
countDownLatch.countDown();
throw new RuntimeException(e);
}

}

/**
* {@inheritDoc}
*/
@Override
public void onClosed(final EventSource eventSource) {
log.info("sse连接关闭:{}", aiDialogueVo.getMessageId());
log.info("结果输出:{}" + output.toString());
countDownLatch.countDown();
}

/**
* {@inheritDoc}
*/
@Override
public void onFailure(final EventSource eventSource, final Throwable t, final Response response) {
log.error("使用事件源时出现异常... [响应：{}]...", aiDialogueVo.getMessageId());
countDownLatch.countDown();
}

public CountDownLatch getCountDownLatch() {
return this.countDownLatch;
}
}
监听里面的vo类自己改一下就可以了，改成你自己需要的参数