From 002d689c7c587291da9fc005f9cfed2a9461593f Mon Sep 17 00:00:00 2001
From: yanderong <307678080@qq.com>
Date: 星期二, 07 七月 2026 10:38:59 +0800
Subject: [PATCH] feat: 放心行增加余额变更记录
---
DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/config/web/MybatisConfig.java | 21 ++++
DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java | 141 ++++++++++++++++++++++++++++
DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/model/vo/BalanceModifyRecordVo.java | 84 ++++++++++++++++
DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/dao/UserInfoMapper.java | 3
4 files changed, 249 insertions(+), 0 deletions(-)
diff --git a/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/config/web/MybatisConfig.java b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/config/web/MybatisConfig.java
new file mode 100644
index 0000000..7849d33
--- /dev/null
+++ b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/config/web/MybatisConfig.java
@@ -0,0 +1,21 @@
+package com.stylefeng.guns.config.web;
+
+import com.stylefeng.guns.core.intercept.BalanceUpdateInterceptor;
+import org.apache.ibatis.plugin.Interceptor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+
+/**
+ * @author Lux
+ * @version 1.0.0
+ * @since 2026/7/3 14:33
+ */
+@Configuration
+public class MybatisConfig {
+
+ @Bean
+ public Interceptor balanceUpdateInterceptor() {
+ return new BalanceUpdateInterceptor();
+ }
+}
diff --git a/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java
new file mode 100644
index 0000000..16f4d2a
--- /dev/null
+++ b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java
@@ -0,0 +1,141 @@
+package com.stylefeng.guns.core.intercept;
+
+import com.stylefeng.guns.modular.system.dao.UserInfoMapper;
+import com.stylefeng.guns.modular.system.model.UserInfo;
+import com.stylefeng.guns.modular.system.model.vo.BalanceModifyRecordVo;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.ibatis.executor.Executor;
+import org.apache.ibatis.mapping.BoundSql;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.apache.ibatis.mapping.SqlCommandType;
+import org.apache.ibatis.plugin.*;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
+import org.springframework.stereotype.Component;
+import org.springframework.util.ReflectionUtils;
+
+import java.lang.reflect.Field;
+import java.util.Map;
+import java.util.Objects;
+import java.util.Optional;
+import java.util.Properties;
+
+/**
+ * 用户余额变更拦截器
+ * @author sky
+ * @version 1.0.0
+ * @since 2026/7/3 12:03
+ */
+@Slf4j
+@Component
+@Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})})
+public class BalanceUpdateInterceptor implements Interceptor {
+
+ @Autowired
+ private ApplicationContext context;
+
+ private UserInfoMapper getMapper() {
+ return context.getBean(UserInfoMapper.class);
+ }
+
+ @Override
+ public Object intercept(Invocation invocation) throws Throwable {
+
+ MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
+ Object param = invocation.getArgs()[1];
+
+ // 获取SQL类型
+ SqlCommandType sqlCommandType = ms.getSqlCommandType();
+
+ if (sqlCommandType != SqlCommandType.UPDATE) {
+ return invocation.proceed();
+ }
+ BoundSql boundSql = ms.getBoundSql(param);
+ String sql = boundSql.getSql().toLowerCase();
+
+ // 只处理 user 表
+ if (!sql.contains("t_user")) {
+ return invocation.proceed();
+ }
+
+ // 关键：先执行前查旧值（需要拿 userId）
+ Long userId = extractUserId(param);
+ // 查询旧的用户
+ UserInfo oldUser = getMapper().selectById(userId);
+ // 执行更新
+ Object result = invocation.proceed();
+ // 查新的用户
+ UserInfo newUser = getMapper().selectById(userId);
+
+ // 新增余额的变更记录（所有余额的变更记录都需要存）
+ if(Objects.nonNull(oldUser) && Objects.nonNull(newUser)) {
+ BalanceModifyRecordVo recordVo = BalanceModifyRecordVo.convert(oldUser, newUser);
+ Optional.ofNullable(recordVo).ifPresent(getMapper()::insertBalanceModifyRecord);
+ }
+ return result;
+ }
+
+ @Override
+ public Object plugin(Object target) {
+ return Plugin.wrap(target, this);
+ }
+
+ @Override
+ public void setProperties(Properties properties) {
+
+ }
+
+ /**
+ * 获取用户id
+ */
+ private Long extractUserId(Object param) {
+
+ if (param == null) {
+ return null;
+ }
+
+ // MyBatis ParamMap
+ if (param instanceof Map) {
+
+ Map<?, ?> map = (Map<?, ?>) param;
+
+ // 第一种：直接传id
+ try{
+ Object id = map.get("id");
+ if (id != null) {
+ return Long.valueOf(id.toString());
+ }
+ }
+ catch (Exception e){
+ log.error("sql中没有id");
+ }
+
+
+ // 第二种：MyBatis-Plus update(entity, wrapper)
+ Object entity = map.get("param1");
+ if (entity == null) {
+ entity = map.get("et");
+ }
+
+ if (entity != null) {
+ return extractId(entity);
+ }
+ }
+
+ // 直接就是实体
+ return extractId(param);
+ }
+
+ private Long extractId(Object obj) {
+ Field field = ReflectionUtils.findField(obj.getClass(), "id");
+ if (field == null) {
+ return null;
+ }
+
+ ReflectionUtils.makeAccessible(field);
+ Object value = ReflectionUtils.getField(field, obj);
+
+ return value == null ? null : Long.valueOf(value.toString());
+ }
+
+}
diff --git a/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/dao/UserInfoMapper.java b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/dao/UserInfoMapper.java
index f110bb6..536bdc7 100644
--- a/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/dao/UserInfoMapper.java
+++ b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/dao/UserInfoMapper.java
@@ -2,6 +2,7 @@
import com.baomidou.mybatisplus.mapper.BaseMapper;
import com.stylefeng.guns.modular.system.model.UserInfo;
+import com.stylefeng.guns.modular.system.model.vo.BalanceModifyRecordVo;
import org.apache.ibatis.annotations.Param;
import java.util.Map;
@@ -51,4 +52,6 @@
* @return
*/
Map<String, Object> queryRealName(@Param("uid") Integer uid);
+
+ void insertBalanceModifyRecord(@Param("recordVo") BalanceModifyRecordVo recordVo);
}
diff --git a/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/model/vo/BalanceModifyRecordVo.java b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/model/vo/BalanceModifyRecordVo.java
new file mode 100644
index 0000000..c24b731
--- /dev/null
+++ b/DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/model/vo/BalanceModifyRecordVo.java
@@ -0,0 +1,84 @@
+package com.stylefeng.guns.modular.system.model.vo;
+
+import com.stylefeng.guns.modular.system.model.UserInfo;
+import lombok.Data;
+
+import java.math.BigDecimal;
+import java.util.Date;
+import java.util.Random;
+
+/**
+ * 余额修改记录
+ *
+ * @author Lux
+ * @version 1.0.0
+ * @since 2026/6/23 15:06
+ */
+@Data
+public class BalanceModifyRecordVo {
+
+ private Integer id;
+
+ /**
+ * 流水ID
+ */
+ private String recordID;
+
+ /**
+ * 用户名称
+ */
+ private String userName;
+
+ /**
+ * 用户手机号
+ */
+ private String phone;
+
+ /**
+ * 用户id
+ */
+ private Integer userId;
+
+ /**
+ * 1=增加，2=减少
+ */
+ private Integer type;
+
+ /**
+ * '添加时间'
+ */
+ private Date createTime;
+
+ /**
+ * '金额'
+ */
+ private BigDecimal money;
+
+ public static BalanceModifyRecordVo convert(UserInfo oldUser, UserInfo newUser) {
+ Random random = new Random();
+ StringBuilder result = new StringBuilder();
+ for (int i = 0; i < 8; i++) {
+ result.append(random.nextInt(10));
+ }
+ BalanceModifyRecordVo recordVo = new BalanceModifyRecordVo();
+ recordVo.setRecordID(result.toString());
+ recordVo.setUserName(newUser.getNickName());
+ recordVo.setPhone(newUser.getPhone());
+ recordVo.setUserId(newUser.getId());
+ recordVo.setCreateTime(new Date());
+ BigDecimal newBalance = newUser.getBalance() != null ? BigDecimal.valueOf(newUser.getBalance()) : BigDecimal.ZERO;
+ BigDecimal oldBalance = oldUser.getBalance() != null ? BigDecimal.valueOf(oldUser.getBalance()) : BigDecimal.ZERO;
+ BigDecimal amount = newBalance.subtract(oldBalance);
+ // 判断方向
+ if (amount.compareTo(BigDecimal.ZERO) == 0){
+ return null;
+ } else if (amount.compareTo(BigDecimal.ZERO) > 0) {
+ recordVo.setType(1); // 增加金额(充值)
+ recordVo.setMoney(amount);
+ } else {
+ recordVo.setType(2); // 减少金额，转为正数（消费）
+ recordVo.setMoney(amount.abs());
+ }
+ return recordVo;
+ }
+}
--
Gitblit v1.7.1