通过方法引用获取方法名

川长思鸟来 2023-09-30 14:35 18阅读 0赞

在学习Mybatis-plus时，可以通过getter方法的方法引用来获取到对应的字段名

LambdaQueryWrapper<PurchaseOrder> wrapper = new LambdaQueryWrapper<>();
    wrapper.like(PurchaseOrder::getBillTypeName, order.getBillTypeName());

这种情况避免了魔法值的出现，也不需要手动写死，避免因需求改变而导致的错误

QueryWrapper<OrderDataDTO> wrapper = new QueryWrapper<>();
    wrapper.like("bill_type_name", dto.getBillTypeName())

lambda表达式不仅可以通过方法引用简化代码，也可以通过getter/setter的方法引用获取到对应的属性名，避免出现bug

### 代码实现 ###

#### 函数式接口 ####

> 首先需要定义一个函数式接口来接受我们的方法引用，该接口必须要继承 **Serializable** 才能获取到方法信息

@FunctionalInterface
    public interface SFunction<T> extends Serializable {
        /**
         * 获取对象属性信息
         *
         * @param source
         * @return
         */
        Object get(T source);
    }

**函数式接口**：如果一个接口中只有一个抽象方法（可以包含多个默认方法或多个static方法），那么该接口就是函数式接口。本质上是将函数的实现直接转换为了一个声明语句的定义，极大简化了原有的实现方式

**@FunctionalInterface**：用来指定某个接口必须是函数式接口，所以该注解只能修饰接口，加上标注, 则会触发JavaCompiler的检查。对于符合函数接口的接口，加不加都无关紧要，但是加上则会提供一层编译检查的保障。如果不符合，则会报错。

#### 工具类 ####

解析Function获取字段属性名

public class ConvertUtil {
    
        public static final String GET = "get";
    
        public static final String IS = "is";
    
        private static final Map<Class<?>, SerializedLambda> CLASS_LAMBDA_CACHE = new ConcurrentHashMap<>();
    
        /**
         * 转换方法引用为属性名
         *
         * @param fn
         * @param <T>
         * @return
         */
        public static <T> String convertToFieldName(SFunction<T> fn) {
            SerializedLambda lambda = getSerializedLambda(fn);
            String methodName = lambda.getImplMethodName();
            if (methodName.startsWith(GET)) {
                methodName = methodName.substring(3);
            } else if (methodName.startsWith(IS)) {
                methodName = methodName.substring(2);
            } else {
                throw new IllegalArgumentException("无效的getter方法：" + methodName);
            }
            return StringUtil.firstToLowerCase(methodName);
        }
    
        public static SerializedLambda getSerializedLambda(Serializable fn) {
            SerializedLambda lambda = CLASS_LAMBDA_CACHE.get(fn.getClass());
            // 先检查缓存中是否存在
            if (lambda == null) {
                try {
                    // 提取SerializedLambda并缓存
                    Method method = fn.getClass().getDeclaredMethod("writeReplace");
                    method.setAccessible(Boolean.TRUE);
                    lambda = (SerializedLambda) method.invoke(fn);
                    CLASS_LAMBDA_CACHE.put(fn.getClass(), lambda);
                } catch (NoSuchMethodException | InvocationTargetException | IllegalAccessException e) {
                    throw new RuntimeException(e);
                }
            }
            return lambda;
        }
    }

/**
     * 首字母转换小写
     *
     * @param str 需要转换的字符串
     * @return 转换好的字符串
     */
    public static String firstToLowerCase(final String str) {
        if (isBlank(str)) {
            return EMPTY;
        }
        return str.substring(0, 1).toLowerCase() + str.substring(1);
    }

#### 测试 ####

public static void main(String[] args) {
        String firstNameFieldNameStyle = ConvertUtil.convertToFieldName(User::getFirstName);
        String firstNameUnderlineStyle = ConvertUtil.convertToUnderline(User::getFirstName);
        System.out.println("firstNameFieldNameStyle：" + firstNameFieldNameStyle);
        System.out.println("firstNameUnderlineStyle：" + firstNameUnderlineStyle);
    }

![188bc3815281489ea1ba9bc8845eabaf.png][]

想要深入研究的可以看这个链接：[Lambda表达式获取传入的方法引用的方法名\_社恐Coder的博客-CSDN博客][Lambda_Coder_-CSDN]

[188bc3815281489ea1ba9bc8845eabaf.png]: https://img-blog.csdnimg.cn/188bc3815281489ea1ba9bc8845eabaf.png
[Lambda_Coder_-CSDN]: https://blog.csdn.net/u012503481/article/details/100896507