feat: 放心行增加余额变更记录

yanderong

2026-07-07 002d689c7c587291da9fc005f9cfed2a9461593f

feat: 放心行增加余额变更记录

 DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/config/web/MybatisConfig.java

New file
@@ -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();
    }
}

 DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/core/intercept/BalanceUpdateInterceptor.java

New file
@@ -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());
    }

}

 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);
}

 DriverYKDTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/model/vo/BalanceModifyRecordVo.java

New file
@@ -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;
    }
}