From 1baa21c114f634b8ecc3af64efcfaab6a57a571e Mon Sep 17 00:00:00 2001
From: yanderong <307678080@qq.com>
Date: 星期六, 11 七月 2026 14:33:25 +0800
Subject: [PATCH] feat: 增加余额变更记录
---
DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java | 63 +++++++++++++++++++------------
1 files changed, 39 insertions(+), 24 deletions(-)
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
index 16f4d2a..f755c86 100644
--- 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
@@ -15,10 +15,7 @@
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;
+import java.util.*;
/**
* 用户余额变更拦截器
@@ -59,13 +56,22 @@
}
// 关键:先执行前查旧值(需要拿 userId)
- Long userId = extractUserId(param);
+ Map<String, Long> mapValues = extractUserId(param);
+ Long userId = mapValues.get("userId");
+ if(Objects.isNull(userId)) {
+ return invocation.proceed();
+ }
+
// 查询旧的用户
UserInfo oldUser = getMapper().selectById(userId);
// 执行更新
Object result = invocation.proceed();
// 查新的用户
UserInfo newUser = getMapper().selectById(userId);
+ Long balanceChangeType = mapValues.get("balanceChangeType");
+ if(Objects.nonNull(balanceChangeType)) {
+ newUser.setBalanceChangeType(balanceChangeType.intValue());
+ }
// 新增余额的变更记录(所有余额的变更记录都需要存)
if(Objects.nonNull(oldUser) && Objects.nonNull(newUser)) {
@@ -88,28 +94,30 @@
/**
* 获取用户id
*/
- private Long extractUserId(Object param) {
-
+ private Map<String, Long> extractUserId(Object param) {
+ Map<String, Long> mapValues = new HashMap<>();
if (param == null) {
- return null;
+ return mapValues;
}
// 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");
- }
+ Object balanceChangeType = map.get("balanceChangeType");
+ if (Objects.nonNull(balanceChangeType)) {
+ mapValues.put("balanceChangeType", Long.valueOf(balanceChangeType.toString()));
+ }
+ Object id = map.get("id");
+ if (Objects.nonNull(id)) {
+ mapValues.put("userId", Long.valueOf(id.toString()));
+ return mapValues;
+ }
+ } catch (Exception e) {
+ // log.error("sql中没有id");
+ }
// 第二种:MyBatis-Plus update(entity, wrapper)
Object entity = map.get("param1");
@@ -118,24 +126,31 @@
}
if (entity != null) {
- return extractId(entity);
+ return extractId(entity, mapValues);
}
}
// 直接就是实体
- return extractId(param);
+ return extractId(param, mapValues);
}
- private Long extractId(Object obj) {
+ private Map<String, Long> extractId(Object obj, Map<String, Long> mapValues) {
Field field = ReflectionUtils.findField(obj.getClass(), "id");
if (field == null) {
- return null;
+ return mapValues;
}
-
ReflectionUtils.makeAccessible(field);
Object value = ReflectionUtils.getField(field, obj);
+ mapValues.put("userId", value == null ? null : Long.valueOf(value.toString()));
- return value == null ? null : Long.valueOf(value.toString());
+ Field fieldType = ReflectionUtils.findField(obj.getClass(), "balanceChangeType");
+ if (fieldType == null) {
+ return mapValues;
+ }
+ ReflectionUtils.makeAccessible(fieldType);
+ Object valueType = ReflectionUtils.getField(fieldType, obj);
+ mapValues.put("balanceChangeType", valueType == null ? null : Long.valueOf(valueType.toString()));
+ return mapValues;
}
}
--
Gitblit v1.7.1