sse接口

doc-biz/src/main/java/com/doc/chatglm3/util/SseEmitterUtf8.java

@@ -0,0 +1,31 @@
+package com.doc.chatglm3.util;
+
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
+
+import java.nio.charset.Charset;
+
+/**
+ * SseEmitterUTF8
+ * 重写SseEmitter 改为UTF-8编码
+ *
+ * @author wukai
+ * @date 2024年3月6日12:00:19
+ */
+public class SseEmitterUtf8 extends SseEmitter {
+
+    public SseEmitterUtf8(Long timeout) {
+        super(timeout);
+    }
+
+    @Override
+    protected void extendResponse(ServerHttpResponse outputMessage) {
+        super.extendResponse(outputMessage);
+
+        HttpHeaders headers = outputMessage.getHeaders();
+        headers.setContentType(new MediaType("text", "event-stream", Charset.forName("UTF-8")));
+    }
+}
+