zm/nuttyreading-java
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ai问答改成websocket

Browse Source
This commit is contained in:
wuchunlei
2025-05-22 17:53:19 +08:00
parent 65aaaa7110
commit 5443e2c764
6 changed files with 97 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pom.xml
View File

@@ -76,6 +76,10 @@
<artifactId>jacob</artifactId>
<version>1.18</version>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>

23
src/main/java/com/peanut/common/utils/RagFlowApiUtil.java
View File

@@ -1,9 +1,9 @@
package com.peanut.common.utils;
import com.alibaba.fastjson.JSONObject;
import com.peanut.config.WebSocket;
import com.peanut.modules.common.entity.AiChatContent;
import com.peanut.modules.common.service.AiChatContentService;
import org.apache.commons.lang.StringUtils;
import org.apache.http.Consts;
import org.apache.http.HttpEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
@@ -18,7 +18,6 @@ import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Flux;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
@@ -33,6 +32,8 @@ public class RagFlowApiUtil {
private String authorization;
@Autowired
private AiChatContentService aiChatContentService;
@Autowired
private WebSocket webSocket;
//聊天助手列表
public List<Map<String,Object>> getChatAssistants(String chatId) throws Exception{
@@ -85,15 +86,6 @@ public class RagFlowApiUtil {
if ("0".equals(jsonObject.get("code").toString())){
List l = jsonObject.getJSONArray("data");
list.addAll(l);
// for (Object o : l) {
// Map<String,Object> map = new HashMap<>();
// Map<String,Object> m = (Map<String,Object>)o;
// map.put("chatId",chatId);
// map.put("id",m.get("id"));
// map.put("name",m.get("name"));
// map.put("messages",m.get("messages"));
// list.add(map);
// }
}
return list;
}
@@ -146,7 +138,7 @@ public class RagFlowApiUtil {
}
//与助手聊天流式
public Flux<String> chatToAssistantStream(String chatId,String chatName,String sessionId,String sessionName,String question,String patientName) {
public void chatToAssistantStream(String chatId,String chatName,String sessionId,String sessionName,String question,String patientName) {
try {
String userId = ShiroUtils.getUId()+"";
Map<String, Object> entity = new HashMap<>();
@@ -165,7 +157,7 @@ public class RagFlowApiUtil {
content.setContent(question);
aiChatContentService.save(content);
List<String> list = new ArrayList<>();
return WebClient.create().post()
WebClient.create().post()
.uri(url+"/api/v1/chats/"+chatId+"/completions")
.header("Authorization", authorization)
.header("Content-Type", "application/json;chartset=utf-8")
@@ -176,11 +168,13 @@ public class RagFlowApiUtil {
JSONObject jsonObject = JSONObject.parseObject(data);
if ("0".equals(jsonObject.get("code").toString())){
if (!"true".equals(jsonObject.get("data").toString())){
webSocket.sendMessage(data);
list.add(((JSONObject)jsonObject.get("data")).get("answer").toString());
}
}
})
.doFinally(data -> {
webSocket.sendMessage("{\"code\":0,\"data\":true}");
AiChatContent answer = new AiChatContent();
answer.setUserId(Integer.parseInt(userId));
answer.setChatAssistantId(chatId);
@@ -191,12 +185,11 @@ public class RagFlowApiUtil {
answer.setType(1);
answer.setContent(list.get(list.size()-1));
aiChatContentService.save(answer);
});
}).subscribe();
}catch (Exception e){
e.printStackTrace();
TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
}
return null;
}

2
src/main/java/com/peanut/config/ShiroConfig.java
View File

@@ -17,7 +17,6 @@ import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.servlet.Filter;
import java.util.HashMap;
import java.util.LinkedHashMap;
@@ -58,6 +57,7 @@ public class ShiroConfig {
filterMap.put("/common/apkConfig/getApkUrl","anon");//获取apk下载地址
filterMap.put("/common/sysFeedback/addSysFeedback","anon");//问题反馈-密码问题
filterMap.put("/websocket/**","anon");
filterMap.put("/oss/**","anon");
filterMap.put("/image/**","anon");

64
src/main/java/com/peanut/config/WebSocket.java Normal file
View File

@@ -0,0 +1,64 @@
package com.peanut.config;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.CopyOnWriteArraySet;
/**
*WebSocket的服务端
*/
@Slf4j
@Component
@ServerEndpoint("/websocket")
public class WebSocket {
private Session session;
private static CopyOnWriteArraySet<WebSocket> webSocketSet = new CopyOnWriteArraySet<>();
@OnOpen
public void onOpen(Session session) {
this.session = session;
webSocketSet.add(this);
log.info("【websocket消息】有新的连接, 总数:{}", webSocketSet.size());
}
//前端关闭时一个websocket时
@OnClose
public void onClose() {
webSocketSet.remove(this);
log.info("【websocket消息】连接断开, 总数:{}", webSocketSet.size());
}
//前端向后端发送消息
@OnMessage
public void onMessage(String message) {
log.info("【websocket消息】收到客户端发来的消息:{}", message);
}
//新增一个方法用于主动向客户端发送消息
public static void sendMessage(String message) {
for (WebSocket webSocket: webSocketSet) {
log.info("【websocket消息】, message={}", message);
try {
webSocket.session.getBasicRemote().sendText(message);
} catch (Exception e) {
e.printStackTrace();
}
}
}
}

16
src/main/java/com/peanut/config/WebSocketConfig.java Normal file
View File

@@ -0,0 +1,16 @@
package com.peanut.config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;
@Configuration
@EnableWebSocket
public class WebSocketConfig{
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}

12
src/main/java/com/peanut/modules/common/controller/RagFlowApiController.java
View File

@@ -9,17 +9,12 @@ import com.peanut.modules.common.service.AiChatContentService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Flux;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
@Slf4j
@RestController("commonRagFlowApi")
@@ -82,10 +77,11 @@ public class RagFlowApiController {
}
//与助手聊天流式
@RequestMapping(value = "/chatToAssistantStream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
@RequestMapping(value = "/chatToAssistantStream")
@Transactional
public Flux<String> chatToAssistantStream(String chatId,String chatName,String sessionId,String sessionName,String question,String patientName){
return ragFlowApiUtil.chatToAssistantStream(chatId,chatName,sessionId,sessionName,question,patientName);
public R chatToAssistantStream(String chatId,String chatName,String sessionId,String sessionName,String question,String patientName){
ragFlowApiUtil.chatToAssistantStream(chatId,chatName,sessionId,sessionName,question,patientName);
return R.ok();
}