diff --git a/servers/boe-server-all/src/main/java/com/xboe/config/ElasticSearchIndexInitializer.java b/servers/boe-server-all/src/main/java/com/xboe/config/ElasticSearchIndexInitializer.java index c25cb91a..bccc491e 100644 --- a/servers/boe-server-all/src/main/java/com/xboe/config/ElasticSearchIndexInitializer.java +++ b/servers/boe-server-all/src/main/java/com/xboe/config/ElasticSearchIndexInitializer.java @@ -26,12 +26,11 @@ public class ElasticSearchIndexInitializer { */ @EventListener(ApplicationReadyEvent.class) public void initializeElasticSearchIndices() { - String indexName = "ai_chat_messages"; - if (elasticSearchIndexService.checkIndexExists(indexName)) { + if (elasticSearchIndexService.checkIndexExists()) { log.info("ElasticSearch索引 ai_chat_messages 已存在"); } else { log.info("ElasticSearch索引 ai_chat_messages 不存在，开始创建..."); - elasticSearchIndexService.createIndex(indexName); + elasticSearchIndexService.createIndex(); } } } \ No newline at end of file diff --git a/servers/boe-server-all/src/main/java/com/xboe/constants/CaseAiConstants.java b/servers/boe-server-all/src/main/java/com/xboe/constants/CaseAiConstants.java new file mode 100644 index 00000000..6190d299 --- /dev/null +++ b/servers/boe-server-all/src/main/java/com/xboe/constants/CaseAiConstants.java @@ -0,0 +1,6 @@ +package com.xboe.constants; + +public class CaseAiConstants { + + public static final String CASE_AI_INDEX_NAME = "ai_chat_messages"; +} diff --git a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/api/CaseAiChatApi.java b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/api/CaseAiChatApi.java index 68d4d4b8..8b436724 100644 --- a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/api/CaseAiChatApi.java +++ b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/api/CaseAiChatApi.java @@ -3,6 +3,7 @@ package com.xboe.module.boecase.api; import com.xboe.core.api.ApiBaseController; import com.xboe.core.JsonResponse; import com.xboe.module.boecase.dto.CaseAiChatDto; +import com.xboe.module.boecase.entity.AiChatConversationData; import com.xboe.module.boecase.service.ICaseAiChatService; import com.xboe.module.boecase.service.ICaseAiPermissionService; import com.xboe.module.boecase.service.IElasticSearchIndexService; @@ -94,16 +95,32 @@ public class CaseAiChatApi extends ApiBaseController { */ @PostMapping("/index/refresh") public JsonResponse deleteAndCreateEsIndex() { - String indexName = "ai_chat_messages"; - if (elasticSearchIndexService.checkIndexExists(indexName)) { - boolean deleteResult = elasticSearchIndexService.deleteIndex(indexName); + if (elasticSearchIndexService.checkIndexExists()) { + boolean deleteResult = elasticSearchIndexService.deleteIndex(); if (deleteResult) { - elasticSearchIndexService.createIndex(indexName); + elasticSearchIndexService.createIndex(); return success("刷新成功"); } } else { - elasticSearchIndexService.createIndex(indexName); + elasticSearchIndexService.createIndex(); } return error("刷新失败"); } + + @PostMapping("/es/create") + public JsonResponse createNewConversation(@RequestBody CaseAiMessageVo caseAiMessageVo, + @RequestParam String conversationId, + @RequestParam String userId) { + AiChatConversationData aiChatConversationData = new AiChatConversationData(); + aiChatConversationData.setConversationId(conversationId); + aiChatConversationData.setQuery(caseAiMessageVo.getQuery()); + aiChatConversationData.appendAnswer(caseAiMessageVo.getAnswer()); + aiChatConversationData.setCaseRefers(caseAiMessageVo.getCaseRefer()); + aiChatConversationData.setSuggestions(caseAiMessageVo.getSuggestions()); + aiChatConversationData.setUserId(userId); + if (elasticSearchIndexService.createData(aiChatConversationData)) { + return success("创建成功"); + } + return error("创建失败"); + } } diff --git a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/IElasticSearchIndexService.java b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/IElasticSearchIndexService.java index 6124c1ab..1f389eaa 100644 --- a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/IElasticSearchIndexService.java +++ b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/IElasticSearchIndexService.java @@ -1,5 +1,10 @@ package com.xboe.module.boecase.service; +import com.xboe.module.boecase.entity.AiChatConversationData; +import com.xboe.module.boecase.vo.CaseAiMessageVo; + +import java.util.List; + /** * es索引 */ @@ -10,18 +15,32 @@ public interface IElasticSearchIndexService { * @param indexName * @return */ - boolean checkIndexExists(String indexName); + boolean checkIndexExists(); /** * 创建索引 * @param indexName */ - boolean createIndex(String indexName); + boolean createIndex(); /** * 删除索引 * @param indexName * @return */ - boolean deleteIndex(String indexName); + boolean deleteIndex(); + + /** + * 新增数据 + * @param data + * @return + */ + boolean createData(AiChatConversationData data); + + /** + * 查询数据 + * @param conversationId + * @return + */ + List queryData(String conversationId); } diff --git a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/CaseAiChatServiceImpl.java b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/CaseAiChatServiceImpl.java index ff08cf73..3be0374c 100644 --- a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/CaseAiChatServiceImpl.java +++ b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/CaseAiChatServiceImpl.java @@ -14,6 +14,7 @@ import com.xboe.module.boecase.entity.Cases; import com.xboe.module.boecase.properties.CaseAiProperties; import com.xboe.module.boecase.service.IAiAccessTokenService; import com.xboe.module.boecase.service.ICaseAiChatService; +import com.xboe.module.boecase.service.IElasticSearchIndexService; import com.xboe.module.boecase.vo.CaseAiMessageVo; import com.xboe.module.boecase.vo.CaseReferVo; import com.xboe.module.boecase.entity.AiChatConversationData; @@ -82,8 +83,8 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService { @Autowired private CaseAiConversationsDao caseAiConversationsDao; - @Autowired(required = false) - private RestHighLevelClient elasticsearchClient; + @Autowired + private IElasticSearchIndexService elasticSearchIndexService; @Autowired private CaseDocumentLogDao caseDocumentLogDao; @@ -142,7 +143,7 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService { public void onClosed(@NotNull EventSource eventSource) { log.info("调用接口 [{}] 接口关闭", request.url()); // 对话完成，保存到ES - saveConversationToES(conversationData); + elasticSearchIndexService.createData(conversationData); sseEmitter.complete(); } @@ -309,46 +310,11 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService { @Override public List getConversationMessages(String conversationId) { - List messages = new ArrayList<>(); - - if (elasticsearchClient == null) { - log.warn("未配置Elasticsearch客户端，无法查询消息记录"); - return messages; + if (StringUtils.isEmpty(conversationId)) { + log.warn("conversationId 为空, 不查询"); + return new ArrayList<>(); } - - try { - // 根据conversationId从数据库查询AI会话ID - String aiConversationId = caseAiConversationsDao.findAiConversationIdById(conversationId); - if (StringUtils.isEmpty(aiConversationId)) { - log.warn("未找到conversationId: {}对应的AI会话ID", conversationId); - return messages; - } - - // 从 ES 中查询消息记录 - SearchRequest searchRequest = new SearchRequest("ai_chat_messages"); // ES索引名，可以根据实际情况调整 - SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder(); - searchSourceBuilder.query(QueryBuilders.termQuery("conversationId", aiConversationId)); - searchSourceBuilder.size(1000); // 设置最大返回数量 - searchRequest.source(searchSourceBuilder); - - SearchResponse searchResponse = elasticsearchClient.search(searchRequest, RequestOptions.DEFAULT); - SearchHits hits = searchResponse.getHits(); - - for (SearchHit hit : hits) { - Map sourceMap = hit.getSourceAsMap(); - CaseAiMessageVo messageVo = parseMessageFromES(sourceMap); - if (messageVo != null) { - messages.add(messageVo); - } - } - - log.info("从 ES 中查询到 {} 条消息记录", messages.size()); - - } catch (Exception e) { - log.error("从 ES 查询会话消息记录异常", e); - } - - return messages; + return elasticSearchIndexService.queryData(conversationId); } /** @@ -556,52 +522,6 @@ public class CaseAiChatServiceImpl implements ICaseAiChatService { } } - /** - * 保存对话记录到ES - */ - private void saveConversationToES(AiChatConversationData conversationData) { - if (elasticsearchClient == null) { - log.warn("未配置Elasticsearch客户端，无法保存对话记录"); - return; - } - - try { - // 构建要保存的数据 - JSONObject esData = new JSONObject(); - esData.put("query", conversationData.getQuery()); - esData.put("answer", conversationData.getAnswerAsString()); - esData.put("conversationId", conversationData.getConversationId()); - esData.put("userId", conversationData.getUserId()); - esData.put("timestamp", LocalDateTime.now().toString()); - - // 构建 caseRefer 数据 - JSONArray caseReferArray = new JSONArray(); - for (CaseReferVo caseRefer : conversationData.getCaseRefers()) { - JSONObject caseReferObj = new JSONObject(); - caseReferObj.put("caseId", caseRefer.getCaseId()); - caseReferObj.put("title", caseRefer.getTitle()); - caseReferObj.put("authorName", caseRefer.getAuthorName()); - caseReferObj.put("keywords", caseRefer.getKeywords()); - caseReferObj.put("content", caseRefer.getContent()); - caseReferArray.add(caseReferObj); - } - esData.put("caseRefer", caseReferArray); - - // 添加建议 - esData.put("suggestions", conversationData.getSuggestions()); - - // 保存到ES - IndexRequest indexRequest = new IndexRequest("ai_chat_messages"); - indexRequest.source(esData.toJSONString(), XContentType.JSON); - - IndexResponse indexResponse = elasticsearchClient.index(indexRequest, RequestOptions.DEFAULT); - log.info("保存对话记录到ES成功，文档ID: {}", indexResponse.getId()); - - } catch (Exception e) { - log.error("保存对话记录到ES异常", e); - } - } - /** * 当 SSE 失败时，作为普通 HTTP 请求处理 */ diff --git a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/ElasticSearchIndexServiceImpl.java b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/ElasticSearchIndexServiceImpl.java index 9b5b38e9..2dd12638 100644 --- a/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/ElasticSearchIndexServiceImpl.java +++ b/servers/boe-server-all/src/main/java/com/xboe/module/boecase/service/impl/ElasticSearchIndexServiceImpl.java @@ -1,8 +1,18 @@ package com.xboe.module.boecase.service.impl; +import com.alibaba.fastjson.JSONArray; +import com.alibaba.fastjson.JSONObject; +import com.xboe.constants.CaseAiConstants; +import com.xboe.module.boecase.entity.AiChatConversationData; import com.xboe.module.boecase.service.IElasticSearchIndexService; +import com.xboe.module.boecase.vo.CaseAiMessageVo; +import com.xboe.module.boecase.vo.CaseReferVo; import lombok.extern.slf4j.Slf4j; import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest; +import org.elasticsearch.action.index.IndexRequest; +import org.elasticsearch.action.index.IndexResponse; +import org.elasticsearch.action.search.SearchRequest; +import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.action.support.master.AcknowledgedResponse; import org.elasticsearch.client.RequestOptions; import org.elasticsearch.client.RestHighLevelClient; @@ -11,10 +21,18 @@ import org.elasticsearch.client.indices.CreateIndexResponse; import org.elasticsearch.client.indices.GetIndexRequest; import org.elasticsearch.common.settings.Settings; import org.elasticsearch.common.xcontent.XContentType; +import org.elasticsearch.index.query.QueryBuilders; +import org.elasticsearch.search.SearchHit; +import org.elasticsearch.search.SearchHits; +import org.elasticsearch.search.builder.SearchSourceBuilder; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.io.IOException; +import java.time.LocalDateTime; +import java.util.ArrayList; +import java.util.List; +import java.util.Map; @Service @Slf4j @@ -24,13 +42,13 @@ public class ElasticSearchIndexServiceImpl implements IElasticSearchIndexService private RestHighLevelClient elasticsearchClient; @Override - public boolean checkIndexExists(String indexName) { + public boolean checkIndexExists() { if (elasticsearchClient == null) { - log.warn("ElasticSearch客户端未配置"); + log.error("ElasticSearch客户端未配置"); return false; } // 检查索引是否存在 - GetIndexRequest getIndexRequest = new GetIndexRequest(indexName); + GetIndexRequest getIndexRequest = new GetIndexRequest(CaseAiConstants.CASE_AI_INDEX_NAME); try { return elasticsearchClient.indices().exists(getIndexRequest, RequestOptions.DEFAULT); } catch (IOException e) { @@ -40,13 +58,13 @@ public class ElasticSearchIndexServiceImpl implements IElasticSearchIndexService } @Override - public boolean createIndex(String indexName) { + public boolean createIndex() { if (elasticsearchClient == null) { - log.warn("ElasticSearch客户端未配置"); + log.error("ElasticSearch客户端未配置"); return false; } // 创建索引 - CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName); + CreateIndexRequest createIndexRequest = new CreateIndexRequest(CaseAiConstants.CASE_AI_INDEX_NAME); // 设置索引配置 createIndexRequest.settings(Settings.builder() @@ -71,29 +89,29 @@ public class ElasticSearchIndexServiceImpl implements IElasticSearchIndexService } if (createIndexResponse.isAcknowledged()) { - log.info("ElasticSearch索引 [{}] 创建成功", indexName); + log.info("ElasticSearch索引 [{}] 创建成功", CaseAiConstants.CASE_AI_INDEX_NAME); return true; } else { - log.warn("ElasticSearch索引 [{}] 创建可能失败，响应未确认", indexName); + log.error("ElasticSearch索引 [{}] 创建可能失败，响应未确认", CaseAiConstants.CASE_AI_INDEX_NAME); return false; } } @Override - public boolean deleteIndex(String indexName) { + public boolean deleteIndex() { if (elasticsearchClient == null) { - log.warn("ElasticSearch客户端未配置"); + log.error("ElasticSearch客户端未配置"); return false; } // 执行删除索引请求 - DeleteIndexRequest deleteRequest = new DeleteIndexRequest(indexName); + DeleteIndexRequest deleteRequest = new DeleteIndexRequest(CaseAiConstants.CASE_AI_INDEX_NAME); try { AcknowledgedResponse deleteResponse = elasticsearchClient.indices().delete(deleteRequest, RequestOptions.DEFAULT); if (deleteResponse.isAcknowledged()) { - log.info("成功删除Elasticsearch索引: {}", indexName); + log.info("成功删除Elasticsearch索引: {}", CaseAiConstants.CASE_AI_INDEX_NAME); return true; } else { - log.warn("删除索引 [{}] 未被确认（可能部分节点未响应）", indexName); + log.error("删除索引 [{}] 未被确认（可能部分节点未响应）", CaseAiConstants.CASE_AI_INDEX_NAME); return false; } } catch (IOException e) { @@ -102,6 +120,130 @@ public class ElasticSearchIndexServiceImpl implements IElasticSearchIndexService } } + @Override + public boolean createData(AiChatConversationData conversationData) { + if (elasticsearchClient == null) { + log.error("未配置Elasticsearch客户端，无法保存对话记录"); + return false; + } + + try { + // 构建要保存的数据 + JSONObject esData = new JSONObject(); + esData.put("query", conversationData.getQuery()); + esData.put("answer", conversationData.getAnswerAsString()); + esData.put("conversationId", conversationData.getConversationId()); + esData.put("userId", conversationData.getUserId()); + esData.put("timestamp", LocalDateTime.now().toString()); + + // 构建 caseRefer 数据 + JSONArray caseReferArray = new JSONArray(); + for (CaseReferVo caseRefer : conversationData.getCaseRefers()) { + JSONObject caseReferObj = new JSONObject(); + caseReferObj.put("caseId", caseRefer.getCaseId()); + caseReferObj.put("title", caseRefer.getTitle()); + caseReferObj.put("authorName", caseRefer.getAuthorName()); + caseReferObj.put("keywords", caseRefer.getKeywords()); + caseReferObj.put("content", caseRefer.getContent()); + caseReferArray.add(caseReferObj); + } + esData.put("caseRefer", caseReferArray); + + // 添加建议 + esData.put("suggestions", conversationData.getSuggestions()); + + // 保存到ES + IndexRequest indexRequest = new IndexRequest("ai_chat_messages"); + String dataStr = esData.toJSONString(); + log.info("保存对话记录到ES：{}", dataStr); + indexRequest.source(dataStr, XContentType.JSON); + + IndexResponse indexResponse = elasticsearchClient.index(indexRequest, RequestOptions.DEFAULT); + log.info("保存对话记录到ES成功，文档ID: {}", indexResponse.getId()); + return true; + } catch (Exception e) { + log.error("保存对话记录到ES异常", e); + return false; + } + } + + @Override + public List queryData(String conversationId) { + List list = new ArrayList<>(); + if (elasticsearchClient == null) { + log.error("未配置Elasticsearch客户端，无法查询对话记录"); + return list; + } + try { + // 从 ES 中查询消息记录 + SearchRequest searchRequest = new SearchRequest(CaseAiConstants.CASE_AI_INDEX_NAME); // ES索引名，可以根据实际情况调整 + SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder(); + searchSourceBuilder.query(QueryBuilders.termQuery("conversationId", conversationId)); + searchSourceBuilder.size(1000); // 设置最大返回数量 + searchRequest.source(searchSourceBuilder); + + SearchResponse searchResponse = elasticsearchClient.search(searchRequest, RequestOptions.DEFAULT); + SearchHits hits = searchResponse.getHits(); + + for (SearchHit hit : hits) { + Map sourceMap = hit.getSourceAsMap(); + CaseAiMessageVo data = parseMessageFromES(sourceMap); + if (data != null) { + list.add(data); + } + } + + log.info("从 ES 中查询到 {} 条消息记录", list.size()); + + } catch (Exception e) { + log.error("从 ES 查询会话消息记录异常", e); + } + return list; + } + + private CaseAiMessageVo parseMessageFromES(Map sourceMap) { + try { + CaseAiMessageVo messageVo = new CaseAiMessageVo(); + messageVo.setQuery((String) sourceMap.get("query")); + messageVo.setAnswer((String) sourceMap.get("answer")); + + // 解析 suggestions + Object suggestionsObj = sourceMap.get("suggestions"); + if (suggestionsObj instanceof List) { + messageVo.setSuggestions((List) suggestionsObj); + } + + // 解析 caseRefer + Object caseReferObj = sourceMap.get("caseRefer"); + if (caseReferObj instanceof List) { + List caseReferList = new ArrayList<>(); + List> caseReferMaps = (List>) caseReferObj; + + for (Map caseReferMap : caseReferMaps) { + CaseReferVo caseRefer = new CaseReferVo(); + caseRefer.setCaseId((String) caseReferMap.get("caseId")); + caseRefer.setTitle((String) caseReferMap.get("title")); + caseRefer.setAuthorName((String) caseReferMap.get("authorName")); + caseRefer.setContent((String) caseReferMap.get("content")); + + // 解析 keywords + Object keywordsObj = caseReferMap.get("keywords"); + if (keywordsObj instanceof List) { + caseRefer.setKeywords((List) keywordsObj); + } + + caseReferList.add(caseRefer); + } + messageVo.setCaseRefer(caseReferList); + } + + return messageVo; + } catch (Exception e) { + log.error("解析ES消息数据异常", e); + return null; + } + } + /** * 获取ai_chat_messages索引的字段映射配置 * 根据项目中的会话消息数据结构规范定义映射