通過AOP攔截Spring?Boot日志并將其存入數據庫功能實現

更新時間：2023年08月30日 08:30:20 作者：華為云開發(fā)者聯盟

本文介紹了如何使用Spring Boot和AOP技術實現攔截系統日志并保存到數據庫中的功能，包括配置數據庫連接、定義日志實體類、定義日志攔截器、使用AOP攔截日志并保存到數據庫中等步驟，感興趣的朋友一起看看吧

前言

在軟件開發(fā)中，常常需要記錄系統運行時的日志。日志記錄有助于排查系統問題、優(yōu)化系統性能、監(jiān)控操作行為等。本文將介紹如何使用Spring Boot和AOP技術實現攔截系統日志并保存到數據庫中的功能。

摘要

本文將通過以下步驟實現攔截系統日志并保存到數據庫中的功能：

配置數據庫連接
定義日志實體類
定義日志攔截器
使用AOP攔截日志并保存到數據庫中

AOP介紹

AOP，全稱是Aspect Oriented Programming，即面向切面編程。AOP的目的是將那些與業(yè)務無關，但是業(yè)務模塊都需要的功能，如日志統計、安全控制、事務處理等，封裝成可重用的組件，從而將它們從業(yè)務邏輯代碼中劃分出來，編寫成獨立的切面。這樣做，既可以保持業(yè)務邏輯的純凈和高內聚性，又可以使得系統的多個模塊都可以共享這些公共的功能。

Spring框架提供了對AOP的支持，Spring Boot自然也不例外。使用Spring Boot的AOP功能，我們可以在運行時動態(tài)地將代碼橫向切入到各個關注點（方法或者類）中。這種橫向切面的方式，比傳統的縱向切面（繼承）更加靈活。

AOP的實現

添加依賴

在pom.xml中添加以下依賴：

<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
<dependency>
<groupId>org.mybatis.spring.boot</groupId>
<artifactId>mybatis-spring-boot-starter</artifactId>
</dependency>

這樣我們就可以使用Spring Boot的AOP功能和MyBatis框架。

配置數據庫連接

首先需要在Spring Boot項目的application.properties文件中配置數據庫連接信息：

spring.datasource.url=jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf-8&useSSL=false
spring.datasource.username=root
spring.datasource.password=123456
spring.datasource.driver-class-name=com.mysql.jdbc.Driver

或者你也可以使用YAML的配置格式：

定義日志實體類

定義一個Log實體類用于保存日志信息，并使用@Entity和@Table注解指定對應的數據庫表和字段：

@Entity
@Table(name = "sys_log")
public class Log {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private String username;
private String operation;
private String method;
private String params;
private String ip;
private Date createTime;
// 省略getter和setter方法
}

定義日志攔截器

定義一個日志攔截器LogInterceptor，通過實現HandlerInterceptor接口來攔截請求并記錄日志：

@Component
public class LogInterceptor implements HandlerInterceptor {
@Autowired
private LogRepository logRepository;
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// 獲取請求的IP地址
String ip = getIpAddress(request);
// 獲取當前用戶
String username = getCurrentUsername();
// 獲取請求的方法名
String method = request.getMethod();
// 獲取請求的URL
String url = request.getRequestURI();
// 獲取請求的參數
String params = getParams(request);
// 創(chuàng)建日志實體
Log log = new Log();
log.setIp(ip);
log.setMethod(method);
log.setOperation("訪問");
log.setParams(params);
log.setUsername(username);
log.setCreateTime(new Date());
// 保存日志到數據庫中
logRepository.save(log);
return true;
}
// 省略實現HandlerInterceptor接口的其他方法
/**
* 獲取請求的IP地址
*/
private String getIpAddress(HttpServletRequest request) {
String ip = request.getHeader("X-Forwarded-For");
if (StringUtils.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("Proxy-Client-IP");
}
if (StringUtils.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("WL-Proxy-Client-IP");
}
if (StringUtils.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("HTTP_CLIENT_IP");
}
if (StringUtils.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("HTTP_X_FORWARDED_FOR");
}
if (StringUtils.isBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getRemoteAddr();
}
return ip;
}
/**
* 獲取當前用戶
*/
private String getCurrentUsername() {
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
if (authentication != null) {
return authentication.getName();
}
return null;
}
/**
* 獲取請求的參數
*/
private String getParams(HttpServletRequest request) {
Map<String, String[]> parameterMap = request.getParameterMap();
if (parameterMap == null || parameterMap.isEmpty()) {
return null;
}
StringBuilder sb = new StringBuilder();
for (Map.Entry<String, String[]> entry : parameterMap.entrySet()) {
sb.append(entry.getKey()).append("=").append(Arrays.toString(entry.getValue())).append("&");
}
return sb.toString();
}
}

使用AOP攔截日志并保存到數據庫中

使用AOP技術攔截所有Controller類中的方法，并執(zhí)行LogInterceptor中的preHandle方法，記錄日志并保存到數據庫中。

定義一個LogAspect切面類，通過實現@Aspect注解和@Before注解來實現方法攔截：

@Aspect
@Component
public class LogAspect {
@Autowired
private LogInterceptor logInterceptor;
@Pointcut("execution(public * com.example.demo.controller..*.*(..))")
public void logAspect() {}
@Before("logAspect()")
public void doBefore(JoinPoint joinPoint) {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
if (attributes == null) {
return;
}
HttpServletRequest request = attributes.getRequest();
HttpServletResponse response = attributes.getResponse();
HandlerMethod handlerMethod = (HandlerMethod) joinPoint.getSignature();
try {
logInterceptor.preHandle(request, response, handlerMethod);
} catch (Exception e) {
e.printStackTrace();
}
}
}

代碼方法介紹

LogInterceptor.preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)方法：攔截請求并記錄日志的方法。
LogInterceptor.getIpAddress(HttpServletRequest request)方法：獲取請求的IP地址。
LogInterceptor.getCurrentUsername()方法：獲取當前用戶。
LogInterceptor.getParams(HttpServletRequest request)方法：獲取請求的參數。
LogAspect.logAspect()方法：定義AOP切入點，攔截Controller類中的所有方法。
LogAspect.doBefore(JoinPoint joinPoint)方法：執(zhí)行方法攔截操作，并調用LogInterceptor.preHandle方法來記錄日志。

測試用例

可以使用Postman等工具發(fā)起請求來測試攔截器是否生效，并查看數據庫中是否保存了對應的日志信息。這里就不直接演示了，畢竟使用起來非常的簡單易上手。

全文小結

本文介紹了如何使用Spring Boot和AOP技術實現攔截系統日志并保存到數據庫中的功能，包括配置數據庫連接、定義日志實體類、定義日志攔截器、使用AOP攔截日志并保存到數據庫中等步驟。通過本文的介紹，可以更好地理解Spring Boot和AOP的應用，為開發(fā)高效、穩(wěn)定的系統提供參考。

注：

環(huán)境說明：Windows10 + Idea2021.3.2 + Jdk1.8 + SpringBoot 2.3.1.RELEASE

到此這篇關于通過AOP攔截Spring Boot日志并將其存入數據庫的文章就介紹到這了,更多相關AOP攔截Spring Boot日志內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: