lsd/ks
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改ai对话中(空参考) 漏洞解决问题——已测试通过

Browse Source
This commit is contained in:
sh
2026-04-17 11:26:15 +08:00
parent 01b1c4365b
commit 6c6e61cb87
1 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
ruoyi-bussiness/src/main/java/com/ruoyi/cms/config/ChatClient.java
View File

@@ -4,8 +4,8 @@ import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject; import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.cms.domain.chat.ChatRequest; import com.ruoyi.cms.domain.chat.ChatRequest;
import com.ruoyi.common.utils.StringUtils; import com.ruoyi.common.utils.StringUtils;
import lombok.var;
import okhttp3.*; import okhttp3.*;
import okio.BufferedSource;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@@ -97,18 +97,34 @@ public class ChatClient {
} }
// 逐行读取 SSE 格式的响应 // 逐行读取 SSE 格式的响应
try (var bufferedSource = body.source()) { BufferedSource bufferedSource = body.source();
//判断 bufferedSource if (bufferedSource == null) {
if (bufferedSource != null) { System.out.println("bufferedSource 为空，直接返回");
return; return;
} }
try {
while (!bufferedSource.exhausted()) { while (!bufferedSource.exhausted()) {
String chunk = bufferedSource.readUtf8Line(); String chunk = bufferedSource.readUtf8Line();
if (chunk != null && !chunk.trim().isEmpty()) { if (chunk != null && !chunk.trim().isEmpty()) {
callback.onData(chunk); callback.onData(chunk);
} }
} }
} catch (IOException e) {
System.out.println("读取流异常"+ e);
} finally {
try {
bufferedSource.close();
} catch (IOException e) {
}
} }
/*try (var bufferedSource = body.source()) {
while (!bufferedSource.exhausted()) {
String chunk = bufferedSource.readUtf8Line();
if (chunk != null && !chunk.trim().isEmpty()) {
callback.onData(chunk);
}
}
}*/
// 通知流结束 // 通知流结束
callback.onComplete(); callback.onComplete();