• 作者:老汪软件技巧
  • 发表时间:2024-09-10 07:00
  • 浏览量:

在开始写代码之前,我们需要先了解下BigDecimal有哪几种小数位的截取方式

关于“截取”那些事儿

想要了解 BigDecimal 是如何处理小数位截取的,我们需要看下 RoundingMode 这个类:

public static void main(String[] args) {
    BigDecimal bigDecimalValue = new BigDecimal("9.455");
    BigDecimal result = bigDecimalValue.setScale(2, RoundingMode.FLOOR);
    System.out.println("FLOOR: " + result);
    BigDecimal result2 = bigDecimalValue.setScale(2, RoundingMode.CEILING);
    System.out.println("CEILING: " + result2);
    BigDecimal result3 = bigDecimalValue.setScale(2, RoundingMode.HALF_UP);
    System.out.println("HALF_UP: " + result3);
    BigDecimal result4 = bigDecimalValue.setScale(2, RoundingMode.HALF_EVEN);
    System.out.println("HALF_EVEN: " + result4);
    BigDecimal result5 = bigDecimalValue.setScale(2, RoundingMode.HALF_DOWN);
    System.out.println("HALF_DOWN: " + result5);
    BigDecimal result6 = bigDecimalValue.setScale(3, RoundingMode.UNNECESSARY);
    System.out.println("UNNECESSARY: " + result6);
}

执行结果如下:

将"9.445"改为"9.455",执行结果如下:

我们可以看到,只有HALF_EVEN模式下,两次的处理逻辑是不一样的,不知道大家还其不记得有个规律叫什么“奇变偶不变”?

我们再试下

将数字改为"9.425",执行结果如下:

将数字改为"9.475",执行结果如下:

RoundingMode.UNNECESSARY 表示该值无需做截取位数的操作,这个可以用来检测位数,因为如果期望的位数和数值的实际位数不一致就会报 ArithmeticException: Rounding necessary的异常,如下图:

定义自定义注解

首先我们先新建一个自定义注解,可以直接复制如下代码:

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DecimalScale {
    int value() default 2; // 默认截取到两位小数
}

来简单介绍下上述代码的具体含义和作用(了解的同学可以跳过这p)

创建 MetaObjectHandler 接口的实现类

我们先来看看MetaObjectHandler是什么,为什么要实现这个接口

大家应该经常会碰到一些字几乎所有的表都有,而且填充方式也是一样的(例如例如创建时间、更新时间、创建人、更新人等),这些也就是我们常说的“公共字段”,而为了简化这一操作,MyBatis-Plus 就提供了 MetaObjectHandler 接口,允许用户可以定义自动填充的逻辑,以减少重复的代码逻辑,增加代码的复用性。

我们今天这篇文章正是用到了它提供的 insertFill() 这个方法,详细代码如下:

package com.imile.ops.analysis.config.myabtis;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.imile.ops.analysis.annotation.DecimalScale;
import com.imile.ops.analysis.context.RequestInfoHolder;
import com.imile.ops.analysis.context.UserContext;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
/**
 * MybatisPlus自定义填充公共字段
 *
 * @author Andrew
 * @date 2020/08/11
 */
@Component
public class MybatisMetaObjectHandler implements MetaObjectHandler {
    private final static String USER_CODE = "USER_CODE";
    private final static String USER_NAME = "USER_NAME";
    private Map getOperator() {
        Map userMap = new HashMap<>();
        UserContext userInfo = RequestInfoHolder.getLoginInfo();
        if (userInfo == null) {
            userMap.put(USER_CODE, "");
            userMap.put(USER_NAME, "");
        } else {
            userMap.put(USER_CODE, userInfo.getUserCode());
            userMap.put(USER_NAME, userInfo.getUserName());
        }
        return userMap;
    }
    @Override
    public void insertFill(MetaObject metaObject) {
        Map userMap = getOperator();
        // 如果用户不为数据迁移
        Object createUserCode = getFieldValByName("createUserCode", metaObject);
        Object id = getFieldValByName("id", metaObject);
        if (createUserCode == null) {
            setFieldValByName("createUserCode", userMap.get(USER_CODE), metaObject);
        }
        Object createUserName = getFieldValByName("createUserName", metaObject);
        if (createUserName == null) {
            setFieldValByName("createUserName", userMap.get(USER_NAME), metaObject);
        }
        Object createDate = getFieldValByName("createDate", metaObject);
        if (createDate == null) {
            setFieldValByName("createDate", LocalDateTime.now(), metaObject);
        }
        Object lastUpdUserCode = getFieldValByName("lastUpdUserCode", metaObject);
        if (lastUpdUserCode == null) {
            setFieldValByName("lastUpdUserCode", userMap.get(USER_CODE), metaObject);
        }
        Object lastUpdUserName = getFieldValByName("lastUpdUserName", metaObject);
        if (lastUpdUserName == null) {
            setFieldValByName("lastUpdUserName", userMap.get(USER_NAME), metaObject);
        }
        Object lastUpdDate = getFieldValByName("lastUpdDate", metaObject);
        if (lastUpdDate == null) {
            setFieldValByName("lastUpdDate", LocalDateTime.now(), metaObject);
        }
        // 处理带有 @DecimalScale() 注解的字段
        processDecimalFields(metaObject);
    }
    @Override
    public void updateFill(MetaObject metaObject) {
        Map userMap = getOperator();
        setFieldValByName("lastUpdUserCode", userMap.get(USER_CODE), metaObject);
        setFieldValByName("lastUpdUserName", userMap.get(USER_NAME), metaObject);
        setFieldValByName("lastUpdDate", LocalDateTime.now(), metaObject);
        processDecimalFields(metaObject);
    }
    /**
     * 处理带有 @DecimalScale() 注解的字段
     *
     * @param metaObject
     */
    private void processDecimalFields(MetaObject metaObject) {
        Arrays.stream(metaObject.getOriginalObject().getClass().getDeclaredFields())
                .filter(field -> field.isAnnotationPresent(DecimalScale.class))
                .forEach(field ->
                {
                    try {
                        field.setAccessible(true);
                        // 获取字段值
                        Object value = field.get(metaObject.getOriginalObject());
                        if (value != null) {
                            // 检查字段值是否为 BigDecimal
                            if (value instanceof BigDecimal) {
                                // 是 -> 截取(向下)指定位数小数
                                BigDecimal v = new BigDecimal(value.toString());
                                // 赋值
                                field.set(metaObject.getOriginalObject(),
                                        v.setScale(field.getAnnotation(DecimalScale.class).value(), RoundingMode.FLOOR));
                            } else {
                                throw new IllegalArgumentException("The field annotated with @DecimalScale must be a Number type.");
                            }
                        }
                    } catch (IllegalAccessException e) {
                        throw new RuntimeException(e);
                    }
                });
    }
}

可以看到上述方法用到了我们一开始介绍的截取方法,但是我遇到一个很奇怪的问题,代码中用的是RoundingMode.FLOOR 向下取整的策略,但入库的数据仍然是按照四舍五入截取的,本地debug发现这个RoundingMode.FLOOR 向下取整的策略配置并没有生效,但目前还未到原因,希望了解朋友能指点下,非常感谢》》抱拳ing~~~