diff --git a/src/main/java/xyz/wbsite/achat/config/ThreadConfig.java b/src/main/java/xyz/wbsite/achat/config/PoolConfig.java similarity index 98% rename from src/main/java/xyz/wbsite/achat/config/ThreadConfig.java rename to src/main/java/xyz/wbsite/achat/config/PoolConfig.java index 16e7986..ba79bb5 100644 --- a/src/main/java/xyz/wbsite/achat/config/ThreadConfig.java +++ b/src/main/java/xyz/wbsite/achat/config/PoolConfig.java @@ -16,7 +16,7 @@ import java.util.concurrent.ThreadPoolExecutor; * @since 1.8 */ @Configuration -public class ThreadConfig { +public class PoolConfig { // 获取CPU核心数 private final int cpuCount = Runtime.getRuntime().availableProcessors(); diff --git a/src/main/java/xyz/wbsite/achat/config/WebConfig.java b/src/main/java/xyz/wbsite/achat/config/WebConfig.java index 67b4b8d..4a60dfb 100644 --- a/src/main/java/xyz/wbsite/achat/config/WebConfig.java +++ b/src/main/java/xyz/wbsite/achat/config/WebConfig.java @@ -22,7 +22,7 @@ public class WebConfig implements WebMvcConfigurer { @Override public void addCorsMappings(CorsRegistry registry) { // 注意,如果授权认认证未通过会直接返回,此跨域配置则不会生效,前端仍然会提示跨域 - registry.addMapping("/chat/**") + registry.addMapping("/**") //允许的域,不要写*,否则cookie就无法使用了 .allowedOriginPatterns("http://localhost:5173") .allowedHeaders("*") diff --git a/src/main/java/xyz/wbsite/achat/core/chat/ChatServiceSampleImpl.java b/src/main/java/xyz/wbsite/achat/core/chat/ChatServiceSampleImpl.java index 9ffee44..fc5552e 100644 --- a/src/main/java/xyz/wbsite/achat/core/chat/ChatServiceSampleImpl.java +++ b/src/main/java/xyz/wbsite/achat/core/chat/ChatServiceSampleImpl.java @@ -87,6 +87,11 @@ public class ChatServiceSampleImpl implements ChatService { streamEmitter.onStart(chatId); char[] charArray = DEFAULT_PROMPT.toCharArray(); for (char c : charArray) { + try { + Thread.sleep(30); + } catch (InterruptedException e) { + throw new RuntimeException(e); + } streamEmitter.onPartial(String.valueOf(c)); } streamEmitter.onComplete(); diff --git a/src/main/java/xyz/wbsite/achat/core/chat/StreamEmitter.java b/src/main/java/xyz/wbsite/achat/core/chat/StreamEmitter.java index 6261e79..472c6ce 100644 --- a/src/main/java/xyz/wbsite/achat/core/chat/StreamEmitter.java +++ b/src/main/java/xyz/wbsite/achat/core/chat/StreamEmitter.java @@ -165,6 +165,7 @@ public class StreamEmitter extends SseEmitter { }) .build(); this.pushChunk(chunk); + this.complete(); this.status = Status.SUCCESS; }