用spring-webmvc包實現AI（Deepseek）事件流（SSE）推送

?前后端：? Spring Boot + Angular

spring-webmvc-5.2.2包

代碼片段如下：

控制層：

@GetMapping(value = "/realtime/page/ai/sse", produces = MediaType.TEXT_EVENT_STREAM_VALUE)@ApiOperation(value = "獲取告警記錄進行AI分析")public SseEmitter getRealTimeAlarmAi(AlarmRecordQueryParam queryParam) {final DateTimeFormatter timeFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm");IPage<AlarmRecord> page = alarmRecordService.findRealTimeByParam(queryParam);StringBuilder alarmInfo = new StringBuilder();try {// 根據狀態設置前綴String prefix = queryParam.getStatus() == 1 ?"最近十條歷史告警記錄：" : "當前十條實時告警信息：";String emptyMessage = queryParam.getStatus() == 1 ?"暫無歷史告警" : "當前無實時告警";if (page.getRecords() != null && !page.getRecords().isEmpty()) {alarmInfo.append(prefix);sseService.buildAlarmContent(page, alarmInfo, timeFormatter);} else {alarmInfo.append(emptyMessage);}sseService.validatePromptLength(alarmInfo, maxPromptLength);} catch (Exception e) {log.error("告警信息處理異常", e);}return sseService.createStreamConnection(alarmInfo.toString(), "告警");}

    @ApiOperation("查詢圖表數據用AI分析數據詳情")@GetMapping("/chart/ai/sse")@OpLog(inputExpression = "開始時間:{#queryParam.startTime},結束時間:{#queryParam.endTime},圖表ID:{#queryParam.chartId}",outputExpression = "{#code}")public SseEmitter chartAiSSEData(@Validated ChartDataQueryParam queryParam) throws Exception {String ChartAi = "報表";ChartInstance chart = Optional.ofNullable(chartService.getById(queryParam.getChartId())).orElseThrow(() -> new Exception("找不到：" + queryParam.getChartId() + "的圖表定義"));List<ChartDeviceSensor> deviceSensors = ChartInstance.toChartDeviceSensorList(chart);String endTime = DataQueryParam.endTime(queryParam.getEndTime());DataQueryParam dataQueryParam = new DataQueryParam(queryParam.getStartTime(), endTime, deviceSensors);IChartDataService chartDataService = chartDataServiceManager.getInstance(chart.getChartTypeId());List dataList = chartDataService.getChartData(dataQueryParam);List<String> times = dataQueryParam.getDateType().getTimes(dataQueryParam.getStartTime(), dataQueryParam.getEndTime());ChartData chartData = new ChartData<>(chart.getId(), chart.getName(), chart.getChartFormat(), chart.getChartTypeId(), chart.getShowType(), chart.getCategoryId(), times, dataList);// 將 ChartData 轉換為壓縮字符串String csvData = ChartDataFormatter.formatChartData(chartData);log.info("當前請求字符長度：" + csvData.length());try {if (csvData.length() > maxPromptLength) {OpLogAspect.setCode(400); // 設置錯誤碼throw new IllegalArgumentException("數據長度超過限制，最大允許長度：" + maxPromptLength);}OpLogAspect.setCode(200);return sseService.createStreamConnection(csvData,ChartAi);} catch (IllegalArgumentException e) {OpLogAspect.setCode(400); // 參數錯誤throw e;} catch (Exception e) {OpLogAspect.setCode(500); // 系統錯誤throw new RuntimeException("處理請求失敗", e);}}

業務層代碼：

package com.keydak.project.core.chart.ai.service.impl;import cn.hutool.core.lang.UUID;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.keydak.project.core.alarm.data.vo.AlarmRecord;
import com.keydak.project.core.chart.ai.dto.KeydakAiConfigDTO;
import com.keydak.project.core.chart.ai.exception.BalanceException;
import com.keydak.project.core.chart.ai.service.SSEService;
import com.keydak.repository.core.enums.SystemGlobalConfigEnum;
import com.keydak.repository.core.service.ISystemGlobalConfigService;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import com.fasterxml.jackson.core.type.TypeReference;import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;/*** AI服務實現** @author xyt*/
@Service
@Slf4j
public class SSEServiceImpl implements SSEService {@Autowiredprivate ISystemGlobalConfigService systemGlobalConfigService;private final ObjectMapper objectMapper = new ObjectMapper();private RateLimiter rateLimiter;@PostConstructpublic void init() {try {// 初始化限流器時動態獲取配置KeydakAiConfigDTO initialConfig = getConfig();rateLimiter = new RateLimiter(initialConfig.getRateLimit());} catch (Exception e) {throw new RuntimeException("初始化失敗，無法獲取Keydak AI配置", e);}}// 線程池配置private static final int CORE_POOL_SIZE = 5; // 核心線程數private static final int MAX_POOL_SIZE = 8; // 最大線程數private static final long KEEP_ALIVE_TIME = 30;  // 線程空閑時間private static final int QUEUE_CAPACITY = 30; //隊列private final ExecutorService executor = new ThreadPoolExecutor(CORE_POOL_SIZE,MAX_POOL_SIZE,KEEP_ALIVE_TIME,TimeUnit.SECONDS,new LinkedBlockingQueue<>(QUEUE_CAPACITY),new ThreadPoolExecutor.AbortPolicy() // 使用 AbortPolicy 直接拒絕任務而不執行);/*** 刷新限流器配置*/@Overridepublic synchronized void refreshRateLimiter(Integer rate) {try {if (rateLimiter == null) {rateLimiter = new RateLimiter(rate);} else {rateLimiter.updateRate(rate);}log.info("限流器已更新，新速率限制: {}", rate);} catch (Exception e) {log.error("刷新限流器配置失敗", e);}}@PreDestroypublic void destroy() {executor.shutdown();try {if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {executor.shutdownNow();}} catch (InterruptedException e) {executor.shutdownNow();Thread.currentThread().interrupt();}}/*** 獲取Keydak AI配置信息。** @return 返回Keydak AI配置信息的DTO對象*/private KeydakAiConfigDTO getConfig() throws Exception {KeydakAiConfigDTO config = systemGlobalConfigService.getTag(SystemGlobalConfigEnum.KEYDAK_AI_CONFIG,KeydakAiConfigDTO.class);if (config == null) {throw new Exception("Keydak AI配置不存在");}return config;}@Overridepublic SseEmitter createStreamConnection(String message, String aiType) {SseEmitter emitter = new SseEmitter(120_000L); // 2分鐘超時try {KeydakAiConfigDTO config = getConfig();double balance = getBalance();log.info("當前余額: {} 元", balance);log.warn("當前可用令牌數: {}", rateLimiter.tokens.get());if (!rateLimiter.tryAcquire()) {log.warn("請求被限流 | 當前允許的QPS:{}", config.getRateLimit());handleRateLimitError(emitter);return emitter;}} catch (BalanceException e) {handleBalanceError(emitter, e.getMessage());return emitter;} catch (Exception e) {handleBalanceError(emitter, "系統錯誤: " + e.getMessage());return emitter;}// 保持原有事件監聽emitter.onCompletion(() -> log.info("SSE連接完成"));emitter.onTimeout(() -> {log.warn("SSE連接超時");rateLimiter.refill(); // 超時請求返還令牌});emitter.onError(e -> log.error("SSE連接錯誤", e));// 保持原有線程池處理executor.execute(() -> {try {processSSEStream(message, aiType, emitter);} catch (Exception e) {emitter.completeWithError(e);}});return emitter;}/*** 新增限流錯誤處理方法** @param emitter 事件發射器* @throws IOException 如果發送失敗*/private void handleRateLimitError(SseEmitter emitter) {try {Map<String, Object> error = new LinkedHashMap<>();error.put("error", "rate_limit_exceeded");error.put("message", "請求過于頻繁，請稍后再試");error.put("timestamp", System.currentTimeMillis());emitter.send(SseEmitter.event().data(objectMapper.writeValueAsString(error)).name("rate-limit-error").reconnectTime(5000L));emitter.complete();} catch (IOException e) {log.error("發送限流錯誤失敗", e);}}private void handleBalanceError(SseEmitter emitter, String errorMsg) {try {JSONObject error = new JSONObject();error.put("error", "balance_insufficient");error.put("message", errorMsg);emitter.send(SseEmitter.event().data(error.toJSONString()).name("balance-error"));emitter.complete();} catch (Exception e) {log.error("發送余額錯誤信息失敗", e);}}private void processSSEStream(String message, String aiType, SseEmitter emitter) throws Exception {HttpURLConnection connection = null;try {connection = createConnection();String jsonBody = buildRequestBody(message, aiType);log.info("發送AI請求數據: {}", jsonBody); // 記錄請求體sendRequestData(connection, jsonBody);validateResponse(connection);try (InputStream inputStream = connection.getInputStream();BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) {String line;while ((line = reader.readLine()) != null) {if (Thread.currentThread().isInterrupted()) {throw new InterruptedException("處理被中斷");}if (line.startsWith("data: ")) {String jsonData = line.substring(6).trim();log.debug("AI響應數據: {}", jsonData);if ("[DONE]".equals(jsonData)) {log.info("收到流結束標記");sendCompletionEvent(emitter);  // 發送完成事件break;  // 結束循環}try {processStreamData(emitter, jsonData);} catch (Exception e) {log.error("數據處理失敗，終止連接", e);emitter.completeWithError(e);break;}}}}} catch (Exception e) {log.error("SSE處理發生異常", e);throw e;} finally {if (connection != null) connection.disconnect();}}private void processStreamData(SseEmitter emitter, String jsonData) throws Exception {try {Map<String, Object> apiResponse = objectMapper.readValue(jsonData,new TypeReference<Map<String, Object>>() {});List<Map<String, Object>> choices = (List<Map<String, Object>>) apiResponse.get("choices");if (choices == null || choices.isEmpty()) return;Map<String, Object> choice = choices.get(0);Map<String, Object> delta = (Map<String, Object>) choice.get("delta");Map<String, Object> chunk = new LinkedHashMap<>();chunk.put("timestamp", System.currentTimeMillis());chunk.put("messageId", UUID.randomUUID().toString());// 處理思考過程if (delta.containsKey("reasoning_content")) {String reasoning = (String) delta.get("reasoning_content");if (reasoning != null && !reasoning.trim().isEmpty()) {chunk.put("type", "reasoning");chunk.put("content", reasoning);sendChunk(emitter, chunk);}}// 處理正式回答if (delta.containsKey("content")) {String content = (String) delta.get("content");if (content != null) {chunk.put("type", "answer");chunk.put("content", content);sendChunk(emitter, chunk);}}} catch (JsonProcessingException e) {log.error("JSON解析失敗 | 原始數據: {} | 錯誤: {}", jsonData, e.getMessage());throw new IOException("Failed to process stream data", e);} catch (ClassCastException e) {log.error("數據結構類型錯誤 | 原始數據: {} | 錯誤: {}", jsonData, e.getMessage());throw new IllegalStateException("Invalid data structure", e);} catch (Exception e) {log.error("處理數據塊時發生未知錯誤 | 原始數據: {}", jsonData, e);throw e;}}private void sendChunk(SseEmitter emitter, Map<String, Object> chunk) throws IOException {String chunkJson = objectMapper.writeValueAsString(chunk);log.debug("發送數據塊: {}", chunkJson);SseEmitter.SseEventBuilder event = SseEmitter.event().data(chunkJson).id(UUID.randomUUID().toString()).name("ai-message").reconnectTime(5000L);emitter.send(event);}private void sendCompletionEvent(SseEmitter emitter) {try {Map<String, Object> completionEvent = new LinkedHashMap<>();completionEvent.put("event", "done");completionEvent.put("timestamp", System.currentTimeMillis());completionEvent.put("messageId", UUID.randomUUID().toString());String eventJson = objectMapper.writeValueAsString(completionEvent);emitter.send(SseEmitter.event().data(eventJson).id("COMPLETION_EVENT").name("stream-end").reconnectTime(0L));  // 停止重連log.info("已發送流結束事件");} catch (IOException e) {log.error("發送完成事件失敗", e);} finally {emitter.complete();log.info("SSE連接已關閉");}}private HttpURLConnection createConnection() throws Exception {KeydakAiConfigDTO config = getConfig();HttpURLConnection connection = (HttpURLConnection) new URL(config.getUrl()).openConnection();connection.setRequestMethod("POST");connection.setDoOutput(true);connection.setRequestProperty("Content-Type", "application/json");connection.setRequestProperty("Authorization", "Bearer " + config.getKey());connection.setRequestProperty("Accept", "text/event-stream");connection.setConnectTimeout(30_000);connection.setReadTimeout(120_000);return connection;}private void sendRequestData(HttpURLConnection connection, String jsonBody) throws Exception {try (OutputStream os = connection.getOutputStream()) {os.write(jsonBody.getBytes(StandardCharsets.UTF_8));os.flush();}}private void validateResponse(HttpURLConnection connection) throws Exception {if (connection.getResponseCode() != 200) {String errorMsg = readErrorStream(connection);throw new RuntimeException("API請求失敗: " + connection.getResponseCode() + " - " + errorMsg);}}private String readErrorStream(HttpURLConnection connection) throws IOException {try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getErrorStream(), StandardCharsets.UTF_8))) {StringBuilder response = new StringBuilder();String line;while ((line = reader.readLine()) != null) {response.append(line);}return response.toString();}}private String buildRequestBody(String userMessage, String aiType) throws IOException {KeydakAiConfigDTO config = null;try {config = getConfig();} catch (Exception e) {e.printStackTrace();}Map<String, Object> request = new HashMap<>();request.put("model", config.getModelType());request.put("stream", true);List<Map<String, String>> messages = new ArrayList<>();Map<String, String> message = new HashMap<>();message.put("role", "user");if ("報表".equals(aiType)) {//報表提問詞message.put("content", buildPrompt(config.getPrompt(), userMessage));} else {//告警提問詞message.put("content", buildPrompt(config.getPromptAlarm(), userMessage));}messages.add(message);request.put("messages", messages);return objectMapper.writeValueAsString(request);}private String buildPrompt(String basePrompt, String userMessage) {return String.format("%s\n%s\n", basePrompt, userMessage);}/*** 查詢當前余額** @return 當前余額* @throws IOException 如果請求失敗*/@Override@SneakyThrowspublic double getBalance() {HttpURLConnection connection = null;try {KeydakAiConfigDTO config = getConfig();URL url = new URL(config.getBalanceUrl());connection = (HttpURLConnection) url.openConnection();connection.setRequestMethod("GET");connection.setRequestProperty("Authorization", "Bearer " + config.getKey());connection.setConnectTimeout(5000);connection.setReadTimeout(5000);int responseCode = connection.getResponseCode();if (responseCode != 200) {String errorBody = readErrorStream(connection); // 復用已有的錯誤流讀取方法throw new IOException("HTTP Error: " + responseCode + " - " + errorBody);}try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream(), StandardCharsets.UTF_8))) {StringBuilder response = new StringBuilder();String line;while ((line = reader.readLine()) != null) {response.append(line);}JSONObject jsonObject = JSON.parseObject(response.toString());// 以下解析邏輯保持原樣if (!jsonObject.containsKey("is_available") || !jsonObject.containsKey("balance_infos")) {throw new IOException("Invalid balance response format");}JSONArray balanceInfos = jsonObject.getJSONArray("balance_infos");if (jsonObject.getBoolean("is_available") && balanceInfos != null && !balanceInfos.isEmpty()) {JSONObject balanceInfo = balanceInfos.getJSONObject(0);if (!balanceInfo.containsKey("total_balance")) {throw new IOException("Missing total_balance field");}return balanceInfo.getDouble("total_balance");} else {throw new IOException("Balance information is not available");}}} finally {if (connection != null) {connection.disconnect();}}}/*** 限流器實現**/private static class RateLimiter {private volatile int capacity;private final AtomicInteger tokens;private volatile long lastRefillTime;private final Object lock = new Object();RateLimiter(int rate) {this.capacity = rate;this.tokens = new AtomicInteger(rate);this.lastRefillTime = System.currentTimeMillis();}public void refill() {synchronized (lock) {long now = System.currentTimeMillis();long elapsed = now - lastRefillTime;if (elapsed >= 1000) {tokens.set(capacity); // 直接重置為最大容量lastRefillTime = now;}}}public boolean tryAcquire() {synchronized (lock) {refill();if (tokens.get() > 0) {tokens.decrementAndGet();return true;}return false;}}public void updateRate(int newRate) {synchronized (lock) {this.capacity = newRate;tokens.set(Math.min(tokens.get(), newRate));lastRefillTime = System.currentTimeMillis();}}}/*** 告警內容構建方法**/@Overridepublic void buildAlarmContent(IPage<AlarmRecord> page,StringBuilder alarmInfo,DateTimeFormatter formatter) {page.getRecords().forEach(record -> {// 時間格式化（使用首次告警時間）String time = Optional.ofNullable(record.getFirstTime()).map(t -> t.format(formatter)).orElse("時間未知");// 設備名稱空值處理String device = StringUtils.defaultString(record.getDeviceName(), "未知設備");// 狀態/數值處理邏輯String state = resolveStateValue(record);// 告警描述處理String desc = StringUtils.defaultString(record.getContent(), "未知告警類型");// 按規范格式拼接alarmInfo.append(String.format("%s %s %s %s;", time, device, state, desc));});}/*** 狀態值解析方法*/private String resolveStateValue(AlarmRecord record) {if (record.getValue() != null) {return record.getValue().stripTrailingZeros().toPlainString();}return record.getStatus() != null ?(record.getStatus() ? "1" : "0") : "狀態未知";}/*** 長度校驗方法**/@Overridepublic void validatePromptLength(StringBuilder content, int maxLength) {if (content.length() > maxLength) {throw new IllegalArgumentException("告警數據過長，請縮小查詢范圍");}}}

前端代碼：

<div class="modal-area"><form name="formNg" novalidate><div class="modal-header"><h3 class="modal-title" style="color: #FFFFFF">AI分析</h3></div><div class="modal-body"><div class="form"><!-- 加載狀態 - 修改為動態效果 --><div ng-if="connectionStatus === 'connecting'" class="loading"><div class="ai-thinking-container"><span>AI思考中</span><div class="ai-typing-indicator"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div></div></div></div><!-- 思考過程 --><div class="thinking-panel" ng-if="thinkingContent"><div class="thinking-header"><i class="fa fa-brain"></i> 思考過程<!-- 總用時顯示（完成后保留） --><span ng-if="thinkingTime">（{{thinkingTime}}秒）</span></div><div class="thinking-content"ng-bind-html="thinkingContent"scroll-to-bottom="thinkingContent"></div></div><!-- 正式回答 --><div class="answer-panel" ng-if="answerContent"><div class="answer-header"><i class="fa fa-comment"></i> 以下是AI的分析</div><div class="answer-content"ng-bind-html="answerContent"scroll-to-bottom="answerContent"></div></div><!-- 錯誤提示 --><div ng-if="connectionStatus === 'error'" class="alert alert-danger"><i class="fa fa-exclamation-triangle"></i> 連接異常，請嘗試重新分析</div></div></div><div class="modal-footer"><button ng-click="retry()"class="btn btn-warning"ng-disabled="connectionStatus === 'connecting'"><i class="fa fa-redo"></i> 重新分析</button><button ng-click="cancel()" class="btn btn-danger"><i class="fa fa-times"></i> 關閉</button></div></form>
</div><style>.thinking-header span {margin-left: 5px;font-size: 0.9em;opacity: 0.8;color: #a0c4ff;}.modal-body {height: 6rem; /* 設置固定高度 */overflow-y: auto; /* 內容超出時顯示滾動條 */padding: 15px;background: #1B448A; /* 背景改為藍色 */border-radius: 4px;font-family: 'Consolas', monospace;color: #FFFFFF;}/* 思考過程樣式 */.thinking-panel {margin-bottom: 20px;border-left: 3px solid #4a90e2;padding-left: 15px;}.thinking-header {color: #4a90e2;font-size: 16px;margin-bottom: 10px;}.thinking-content {background: rgba(255, 255, 255, 0.05);padding: 12px;border-radius: 4px;color: #e0e0e0;line-height: 1.6;}/* 正式回答樣式 */.answer-panel {margin-top: 25px;border-top: 1px solid #00c85333;padding-top: 15px;}.answer-header {color: #00c853;font-size: 16px;margin-bottom: 10px;}.answer-content {background: rgba(255, 255, 255, 0.05);padding: 12px;border-radius: 4px;color: #ffffff;line-height: 1.6;}/* 圖標樣式 */.fa-brain {color: #4a90e2;margin-right: 8px;}.fa-comment {color: #00c853;margin-right: 8px;}/* 新的加載動畫樣式 */.loading {color: #FFF;text-align: left;padding: 15px;font-size: 16px;}.ai-thinking-container {display: flex;align-items: center;gap: 8px;}.ai-typing-indicator {display: flex;align-items: center;gap: 4px;height: 20px;}.typing-dot {width: 8px;height: 8px;background-color: #FFFFFF;border-radius: 50%;opacity: 0.4;animation: typing-animation 1.4s infinite ease-in-out;}.typing-dot:nth-child(1) {animation-delay: 0s;}.typing-dot:nth-child(2) {animation-delay: 0.2s;}.typing-dot:nth-child(3) {animation-delay: 0.4s;}@keyframes typing-animation {0%, 60%, 100% {transform: translateY(0);opacity: 0.4;}30% {transform: translateY(-5px);opacity: 1;}}
</style>

// 報表分析
UI.Controllers.controller("AiTipsCtrl", ["$scope", "$sce", "$uibModalInstance", "parent", "SSEService", "$timeout",function($scope, $sce, $uibModalInstance, parent, SSEService, $timeout) {// 狀態管理$scope.connectionStatus = 'connecting'; // connecting | connected | error | completed$scope.thinkingContent = null;$scope.answerContent = null;$scope.thinkingTime = null; // 新增：思考時間變量$scope.startTime = null; // 新增：開始時間戳let eventSource = null;let thinkingBuffer = "";let answerBuffer = "";// 自動滾動指令$scope.scrollToBottom = function() {$timeout(() => {const container = document.querySelector('.modal-body');if (container) {container.scrollTop = container.scrollHeight + 120;}}, 50);};// 內容更新方法function processChunkData(data) {if (data.type === 'reasoning') {// 如果是第一條思考內容，記錄開始時間if (!thinkingBuffer && !$scope.startTime) {$scope.startTime = new Date().getTime();}thinkingBuffer += data.content;$scope.thinkingContent = $sce.trustAsHtml(thinkingBuffer.replace(/\n/g, '<br/>').replace(/ {2}/g, ' &nbsp;'));// 更新思考時間updateThinkingTime();}else if (data.type === 'answer') {answerBuffer += data.content;$scope.answerContent = $sce.trustAsHtml(answerBuffer.replace(/\n/g, '<br/>').replace(/ {2}/g, ' &nbsp;'));}$scope.scrollToBottom();}function updateThinkingTime() {if ($scope.startTime) {const currentTime = new Date().getTime();$scope.thinkingTime = ((currentTime - $scope.startTime) / 1000).toFixed(1);}}// 初始化SSE連接function initSSE() {const url = '/data/chart/ai/sse?' + $.param(parent.queryParam);eventSource = new EventSource(url);eventSource.onopen = () => {$scope.$apply(() => {$scope.connectionStatus = 'connected';});};// 處理消息事件eventSource.addEventListener('ai-message', e => {$scope.$apply(() => {try {const data = JSON.parse(e.data);processChunkData(data);} catch (err) {console.error('消息解析錯誤:', err);$scope.answerContent = $sce.trustAsHtml('<div class="text-danger">數據格式錯誤</div>');}});});// 處理結束事件eventSource.addEventListener('stream-end', () => {$scope.$apply(() => {$scope.connectionStatus = 'completed';//最終更新一次思考時間updateThinkingTime();safeClose();});});// 錯誤處理eventSource.onerror = (err) => {$scope.$apply(() => {console.error('SSE連接錯誤:', err);$scope.connectionStatus = 'error';safeClose();});};}// 安全關閉連接function safeClose() {if (eventSource) {eventSource.close();eventSource = null;}}// 重新嘗試$scope.retry = () => {safeClose();thinkingBuffer = "";answerBuffer = "";$scope.thinkingContent = null;$scope.answerContent = null;$scope.thinkingTime = null; //重置思考時間$scope.startTime = null; //重置開始時間$scope.connectionStatus = 'connecting';initSSE();};// 關閉模態框$scope.cancel = () => {safeClose();$uibModalInstance.dismiss();};// 初始化initSSE();// 清理$scope.$on('$destroy', () => {safeClose();});}
]);
// 告警分析
UI.Controllers.controller("AiAlarmTipsCtrl", ["$scope", "$sce", "$uibModalInstance", "parent", "SSEService", "$timeout",function($scope, $sce, $uibModalInstance, parent, SSEService, $timeout) {// 狀態管理$scope.connectionStatus = 'connecting'; // connecting | connected | error | completed$scope.thinkingContent = null;$scope.answerContent = null;$scope.thinkingTime = null; // 新增：思考時間變量$scope.startTime = null; // 新增：開始時間戳let eventSource = null;let thinkingBuffer = "";let answerBuffer = "";// 自動滾動指令$scope.scrollToBottom = function() {$timeout(() => {const container = document.querySelector('.modal-body');if (container) {container.scrollTop = container.scrollHeight + 120;}}, 50);};// 內容更新方法function processChunkData(data) {if (data.type === 'reasoning') {// 如果是第一條思考內容，記錄開始時間if (!thinkingBuffer && !$scope.startTime) {$scope.startTime = new Date().getTime();}thinkingBuffer += data.content;$scope.thinkingContent = $sce.trustAsHtml(thinkingBuffer.replace(/\n/g, '<br/>').replace(/ {2}/g, ' &nbsp;'));// 更新思考時間updateThinkingTime();}else if (data.type === 'answer') {answerBuffer += data.content;$scope.answerContent = $sce.trustAsHtml(answerBuffer.replace(/\n/g, '<br/>').replace(/ {2}/g, ' &nbsp;'));}$scope.scrollToBottom();}// 更新思考時間function updateThinkingTime() {if ($scope.startTime) {const currentTime = new Date().getTime();$scope.thinkingTime = ((currentTime - $scope.startTime) / 1000).toFixed(1);}}// 初始化SSE連接function initSSE() {const url = '/alarm/record/realtime/page/ai/sse?' + $.param(parent.queryParam);eventSource = new EventSource(url);eventSource.onopen = () => {$scope.$apply(() => {$scope.connectionStatus = 'connected';});};// 處理消息事件eventSource.addEventListener('ai-message', e => {$scope.$apply(() => {try {const data = JSON.parse(e.data);processChunkData(data);} catch (err) {console.error('消息解析錯誤:', err);$scope.answerContent = $sce.trustAsHtml('<div class="text-danger">數據格式錯誤</div>');}});});// 處理結束事件eventSource.addEventListener('stream-end', () => {$scope.$apply(() => {$scope.connectionStatus = 'completed';// 最終更新一次思考時間updateThinkingTime();safeClose();});});// 錯誤處理eventSource.onerror = (err) => {$scope.$apply(() => {console.error('SSE連接錯誤:', err);$scope.connectionStatus = 'error';safeClose();});};}// 安全關閉連接function safeClose() {if (eventSource) {eventSource.close();eventSource = null;}}// 重新嘗試$scope.retry = () => {safeClose();thinkingBuffer = "";answerBuffer = "";$scope.thinkingContent = null;$scope.answerContent = null;$scope.thinkingTime = null; // 重置思考時間$scope.startTime = null; // 重置開始時間$scope.connectionStatus = 'connecting';initSSE();};// 關閉模態框$scope.cancel = () => {safeClose();$uibModalInstance.dismiss();};// 初始化initSSE();// 清理$scope.$on('$destroy', () => {safeClose();});}
]);showAiTips: function (resolve) {this.showDialog("Template/AiTips.html", "AiTipsCtrl", resolve, 600);},showAiAlarmTips: function (resolve) {this.showDialog("Template/AiAlarmTips.html", "AiAlarmTipsCtrl", resolve, 600);}

數據庫結構：

INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'balanceUrl', 'https://api.deepseek.com/user/balance', '余額查詢');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'enable', 'true', '啟用AI報表助手');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'key', '', 'API密鑰');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'modelType', 'deepseek-reasoner', 'deepseek模型類型');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'prompt', '', 'AI提問詞');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'promptAlarm', '', 'AI提問詞');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'rateLimit', '3', '限制每秒多少次請求');
INSERT INTO `system_global_config`(`tag`, `key`, `value`, `description`) VALUES ('keydak_ai_config', 'url', 'https://api.deepseek.com/v1/chat/completions', 'API接口');

實體類(使用AES加密密鑰)：

package com.keydak.project.core.chart.ai.dto;import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.util.*;/*** AI配置信息** @author xyt*/
@Data
@AllArgsConstructor
@NoArgsConstructor
public class KeydakAiConfigDTO {private Boolean enable;/*** API_URL地址**/private String url;/*** 查詢余額地址**/private String balanceUrl;/*** API密鑰**/private String key;/*** 限流次數**/private Integer rateLimit;/*** AI提問詞(報表)**/private String prompt;/*** AI提問詞(告警)**/private String promptAlarm;/*** 模型類型**/private String modelType;private static final String SALT = ""; // 16 bytes for AES-128private static final String ALGORITHM = "AES/ECB/PKCS5Padding";public void validate() {List<String> missingFields = new ArrayList<>();if (url == null) missingFields.add("API_URL地址");if (balanceUrl == null) missingFields.add("查詢余額地址");if (key == null) missingFields.add("API密鑰");if (rateLimit == null) missingFields.add("限流次數");if (prompt == null) missingFields.add("AI提問詞");if (!missingFields.isEmpty()) {throw new IllegalStateException("參數不能為空: " + String.join(", ", missingFields));}}/*** 判斷密鑰是否已經加密*/public boolean isEncryptedKey(String key) {try {// 嘗試解密，如果能成功解密則認為已經是加密過的decryptKey(key);return true;} catch (Exception e) {return false;}}/*** 加密密鑰**/public String encryptKey(String key) throws Exception {SecretKeySpec secretKey = new SecretKeySpec(SALT.getBytes(StandardCharsets.UTF_8), "AES");Cipher cipher = Cipher.getInstance(ALGORITHM);cipher.init(Cipher.ENCRYPT_MODE, secretKey);byte[] encryptedKey = cipher.doFinal(key.getBytes(StandardCharsets.UTF_8));return Base64.getEncoder().encodeToString(encryptedKey);}/*** 解密密鑰*/public String decryptKey(String encryptedKey) throws Exception {SecretKeySpec secretKey = new SecretKeySpec(SALT.getBytes(StandardCharsets.UTF_8), "AES");Cipher cipher = Cipher.getInstance(ALGORITHM);cipher.init(Cipher.DECRYPT_MODE, secretKey);byte[] decryptedKey = cipher.doFinal(Base64.getDecoder().decode(encryptedKey));return new String(decryptedKey, StandardCharsets.UTF_8);}
}