Spring Boot 無縫接入 DeepSeek 和通義千問請求日志記錄及其ip黑白名單
SpringBoot版本 3.2.0 JDK 版本為17 redis 3.2.0 mybatis 3.0.3
依賴引入
關鍵依賴
<dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-openai-spring-boot-starter</artifactId>
</dependency>
完整依賴
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd"><modelVersion>4.0.0</modelVersion><groupId>com.cqie</groupId><artifactId>spring-ai</artifactId><version>0.0.1-SNAPSHOT</version><name>spring-ai</name><description>spring-ai</description><properties><java.version>17</java.version><project.build.sourceEncoding>UTF-8</project.build.sourceEncoding><project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding><spring-boot.version>3.2.0</spring-boot.version><spring-ai.version>0.8.1</spring-ai.version></properties><parent><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-parent</artifactId><version>3.2.0</version><relativePath/> <!-- lookup parent from repository --></parent><dependencies><!-- Spring Boot Web Starter --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><!-- Spring Boot Test Starter --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-test</artifactId><scope>test</scope></dependency><!-- JUnit (for testing) --><dependency><groupId>junit</groupId><artifactId>junit</artifactId><scope>test</scope></dependency><dependency><groupId>org.apache.commons</groupId><artifactId>commons-lang3</artifactId></dependency><dependency><groupId>io.swagger</groupId><artifactId>swagger-annotations</artifactId><version>1.5.21</version></dependency><dependency><groupId>com.alibaba</groupId><artifactId>fastjson</artifactId><version>2.0.40</version></dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><optional>true</optional></dependency><dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>8.0.28</version></dependency><!--hutool--><dependency><groupId>cn.hutool</groupId><artifactId>hutool-all</artifactId><version>5.8.17</version></dependency><!-- 添加官方Spring AI OpenAI依賴 --><dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-openai-spring-boot-starter</artifactId></dependency><dependency><groupId>org.mybatis.spring.boot</groupId><artifactId>mybatis-spring-boot-starter</artifactId><version>3.0.3</version></dependency></dependencies><dependencyManagement><dependencies><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-dependencies</artifactId><version>${spring-boot.version}</version><type>pom</type><scope>import</scope></dependency><dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-bom</artifactId><version>${spring-ai.version}</version><type>pom</type><scope>import</scope></dependency></dependencies></dependencyManagement><build><plugins><!-- Maven Compiler Plugin --><plugin><groupId>org.apache.maven.plugins</groupId><artifactId>maven-compiler-plugin</artifactId><version>3.8.1</version><configuration><source>${java.version}</source><target>${java.version}</target><encoding>${project.build.sourceEncoding}</encoding></configuration></plugin><!-- Spring Boot Maven Plugin --><plugin><groupId>org.springframework.boot</groupId><artifactId>spring-boot-maven-plugin</artifactId><version>${spring-boot.version}</version><configuration><mainClass>com.cqie.SpringAiApplication</mainClass></configuration><executions><execution><id>repackage</id><goals><goal>repackage</goal></goals></execution></executions></plugin></plugins></build><repositories><!-- Spring Milestones Repository --><repository><id>spring-milestones</id><name>Spring Milestones</name><url>https://repo.spring.io/milestone</url><snapshots><enabled>false</enabled></snapshots></repository><!-- Spring Snapshots Repository --><repository><id>spring-snapshots</id><name>Spring Snapshots</name><url>https://repo.spring.io/snapshot</url><releases><enabled>false</enabled></releases></repository></repositories>
</project>
建表(日志+黑名單)
CREATE TABLE `request_log` (`id` varchar(100) NOT NULL COMMENT '主鍵',`date` datetime DEFAULT NULL COMMENT '請求時間',`request_url` varchar(255) DEFAULT NULL COMMENT '請求路徑',`user_agent` varchar(255) DEFAULT NULL COMMENT 'userAgent',`status` int(11) DEFAULT NULL COMMENT '狀態碼',`ip_address` varchar(255) DEFAULT NULL COMMENT 'ip地址',`method` varchar(100) DEFAULT NULL COMMENT '方法',`error_message` varchar(255) DEFAULT NULL COMMENT '錯誤原因',PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
CREATE TABLE `black_ips` (`id` varchar(100) NOT NULL COMMENT '主鍵id',`black_ip` varchar(255) DEFAULT NULL COMMENT 'ip地址',`status` tinyint(1) DEFAULT NULL COMMENT '轉態',PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
配置文件
# DeepSeek 配置,完全兼容openai配置
# spring:
# ai:
# openai:
# base-url: https://api.deepseek.com # DeepSeek的OpenAI式端點
# api-key: sk-xxxxxxxxx
# chat.options:
# model: deepseek-chat # 指定DeepSeek的模型名稱# 通義千問配置
spring:ai:openai:base-url: https://dashscope.aliyuncs.com/compatible-mode # 通義千問api-key: sk-xxxxxxxxxxxchat.options:model: qwen-plus
配置文件示例
server:port: 8080
spring:application:name: spring-aiai:openai:base-url: https://dashscope.aliyuncs.com/compatible-modeapi-key: sk-***chat.options:model: qwen-plusdatasource:url: jdbc:mysql://ip:3306/springai?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8username: 用戶名password: 密碼driver-class-name: com.mysql.cj.jdbc.Driver# Redis配置data:redis:host: ipport: 6379password: 密碼database: 3lettuce:pool:max-active: 8max-wait: -1msmax-idle: 8min-idle: 0timeout: 10000ms
# 日志配置
logging:level:org.springframework.ai: DEBUG# mybatis配置
mybatis:mapper-locations: classpath:mapper/*.xmltype-aliases-package: com.cqie.entityconfiguration:map-underscore-to-camel-case: truecall-setters-on-nulls: truejdbc-type-for-null: 'null'
全局異常捕獲
統一返回
package com.cqie.common;import lombok.AllArgsConstructor;
import lombok.Getter;@Getter
@AllArgsConstructor
public class Result {private int code;private String message;private Object data;public Result() {this.code = 0;this.message = "success";this.data = null;}public static Result success(Object data, String message) {Result result = new Result();result.code = 200;result.message = message;result.data = data;return result;}public static Result success(Object data) {Result result = new Result();result.code = 200;result.message = "success";result.data = data;return result;}public static Result error(String errorMsg) {Result result = new Result();result.code = 500;result.message = errorMsg;result.data = null;return result;}}
定義異常
package com.cqie.common;/*** 服務異常*/
public class ServerException extends RuntimeException {public ServerException(String message) {super(message);}public ServerException(String message, Throwable cause) {super(message, cause);}
}
全局捕獲
package com.cqie.config;import com.cqie.common.CommonException;
import com.cqie.common.Result;
import com.cqie.common.ServerException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;/*** 全局異常處理器*/
@RestControllerAdvice
@Slf4j
public class GlobeExceptionHandler {// 處理全局異常@ExceptionHandler(CommonException.class)public Result CommonException(Exception e) {return Result.error(e.getMessage());}@ExceptionHandler(ServerException.class)public Result ServerException(Exception e) {return Result.error(e.getMessage());}}
基于interceptor的日志攔截器
package com.cqie.common;import com.cqie.dao.BlackIpsDao;
import com.cqie.dao.RequestLogDao;
import com.cqie.entity.BlackIps;
import com.cqie.entity.RequestLog;
import com.cqie.utils.RedisUtils;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;import java.time.LocalDateTime;
import java.util.List;
import java.util.stream.Collectors;/*** 請求檢查,請求日志記錄,黑名單處理*/
@Component
@ConditionalOnBean(RequestLogDao.class)
public class LoggingInterceptor implements HandlerInterceptor {private final RequestLogDao requestLogDao;private final BlackIpsDao blackIpsDao;private final RedisUtils redisUtils;private final String BLACK_IPS_KEY = "black_ips:";public LoggingInterceptor(RequestLogDao requestLogDao, BlackIpsDao blackIpsDao, RedisUtils redisUtils) {this.requestLogDao = requestLogDao;this.blackIpsDao = blackIpsDao;this.redisUtils = redisUtils;}@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {String ipAddress = getClientIpAddress(request);int status = response.getStatus();String errorMessage = null;// 對黑名單的ip進行處理List<String> blackIps = (List<String>) redisUtils.get(BLACK_IPS_KEY + "spring-ai");if (blackIps == null) {List<String> ipBlackList = blackIpsDao.queryByStatus(0).stream().map(BlackIps::getBlackIp).collect(Collectors.toList());// 一天的過期時間redisUtils.set(BLACK_IPS_KEY + "spring-ai", ipBlackList, 24);}if (blackIps != null && blackIps.contains(ipAddress)) {status = 500;errorMessage = "請求ip已被加入黑名單";saveRequestLog(request, ipAddress, status, errorMessage);throw new ServerException(errorMessage);}// 判斷2s請求最多請求一次,對請求頻率做限制boolean exists = redisUtils.exists(ipAddress);if (exists) {status = 500;errorMessage = "ai服務請求太頻繁";saveRequestLog(request, ipAddress, status, errorMessage);throw new ServerException(errorMessage);}// 記錄調用日志saveRequestLog(request, ipAddress, status, null);// 對請求記錄分析 限制2s請求最多請求一次redisUtils.set(ipAddress, "1", 5);return HandlerInterceptor.super.preHandle(request, response, handler);}private void saveRequestLog(HttpServletRequest request, String ipAddress, int status, String errorMessage) {LocalDateTime now = LocalDateTime.now();String method = request.getMethod();String url = request.getRequestURI();String userAgent = request.getHeader("User-Agent");RequestLog requestLog = new RequestLog();requestLog.setDate(now);requestLog.setRequestUrl(url);requestLog.setStatus(status);requestLog.setUserAgent(userAgent);requestLog.setIpAddress(ipAddress);requestLog.setMethod(method);requestLog.setErrorMessage(errorMessage);requestLogDao.insert(requestLog);}@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {HandlerInterceptor.super.afterCompletion(request, response, handler, ex);}private String getClientIpAddress(HttpServletRequest request) {String ipAddress = request.getHeader("X-Forwarded-For");if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) {ipAddress = request.getHeader("Proxy-Client-IP");}if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) {ipAddress = request.getHeader("WL-Proxy-Client-IP");}if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) {ipAddress = request.getRemoteAddr();}return ipAddress;}
}
注冊攔截器
package com.cqie.config;import com.cqie.common.LoggingInterceptor;
import com.cqie.dao.BlackIpsDao;
import com.cqie.dao.RequestLogDao;
import com.cqie.utils.RedisUtils;
import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;@Configuration
@RequiredArgsConstructor
public class WebMvcConfig implements WebMvcConfigurer {private final RequestLogDao requestLogDao;private final BlackIpsDao blackIpsDao;private final RedisUtils redisUtils;@Overridepublic void addInterceptors(InterceptorRegistry registry) {//日志攔截器registry.addInterceptor(new LoggingInterceptor(requestLogDao, blackIpsDao,redisUtils)).addPathPatterns("/**").order(0);}@Overridepublic void addResourceHandlers(ResourceHandlerRegistry registry) {// 靜態資源訪問路徑和存放路徑配置registry.addResourceHandler("/static/**").addResourceLocations("classpath:/static/", "classpath:/public/");// 新增Camunda webjar資源映射registry.addResourceHandler("/webjars/camunda/**").addResourceLocations("classpath:/META-INF/resources/webjars/camunda-webapp-ui/");// swagger訪問配置registry.addResourceHandler("/**").addResourceLocations("classpath:/META-INF/resources/", "classpath:/META-INF/resources/webjars/");}}
實現日志和日志表操作
dao server impl 這里只需要dao層就行
略
redis工具類簡單封裝
package com.cqie.utils;import jakarta.annotation.Resource;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;import java.util.concurrent.TimeUnit;@Component
public class RedisUtils {@Resourceprivate RedisTemplate<String, Object> redisTemplate;/*** 設置緩存** @param key 鍵* @param value 值*/public void set(String key, Object value) {redisTemplate.opsForValue().set(key, value);}/*** 設置緩存并設置過期時間** @param key 鍵* @param value 值* @param timeout 過期時間(秒)*/public void set(String key, Object value, long timeout) {redisTemplate.opsForValue().set(key, value, timeout, TimeUnit.HOURS);}/*** 獲取緩存** @param key 鍵* @return 值*/public Object get(String key) {return redisTemplate.opsForValue().get(key);}/*** 刪除緩存** @param key 鍵*/public void delete(String key) {redisTemplate.delete(key);}/*** 判斷key是否存在** @param key 鍵* @return true 存在 false不存在*/public boolean hasKey(String key) {return Boolean.TRUE.equals(redisTemplate.hasKey(key));}/*** 設置過期時間** @param key 鍵* @param timeout 過期時間(秒)*/public void expire(String key, long timeout) {redisTemplate.expire(key, timeout, TimeUnit.SECONDS);}/*** 獲取過期時間** @param key 鍵* @return 過期時間(秒)*/public long getExpire(String key) {return redisTemplate.getExpire(key, TimeUnit.SECONDS);}public boolean exists(String key) {return redisTemplate.hasKey(key);}
}
接口實現
package com.cqie.controller;import org.springframework.ai.chat.ChatClient;
import org.springframework.ai.chat.ChatResponse;
import org.springframework.ai.chat.StreamingChatClient;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.SystemMessage;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Flux;import java.util.ArrayList;
import java.util.List;/*** 基于DeepSeek/通義千問的聊天控制器** @author qingyuqiao*/
@RestController
@RequestMapping("/api")
public class ChatController {/*** 上下文*/private final List<Message> contextHistoryList = new ArrayList<>();private final ChatClient chatClient;private final StreamingChatClient streamingChatClient;/*** ai 初始化信息** @param chatClient* @param streamingChatClient*/public ChatController(ChatClient chatClient, StreamingChatClient streamingChatClient) {this.chatClient = chatClient;this.streamingChatClient = streamingChatClient;// 對用戶輸入進行增強contextHistoryList.add(new SystemMessage("你是一個專業的it技術顧問。"));}/*** 普通對話** @param message 問題* @return 回答結果*/@GetMapping("/chat")public ChatResponse chat(@RequestParam String message) {contextHistoryList.add(new UserMessage(message));Prompt prompt = new Prompt(contextHistoryList);ChatResponse chatResp = chatClient.call(prompt);if (chatResp.getResult() != null) {contextHistoryList.add(chatResp.getResult().getOutput());}return chatResp;}/*** 流式返回** @param message 問題* @return 流式結果*/@GetMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)public Flux<String> streamChat(@RequestParam String message) {contextHistoryList.add(new UserMessage(message));Prompt prompt = new Prompt(contextHistoryList);return streamingChatClient.stream(prompt).map(chatResponse -> {if (chatResponse.getResult() != null) {return chatResponse.getResult().getOutput().getContent();}return "";});}
}
成功請求
配合黑名單
接口限制