富贵不能淫;贫贱不能移;威武不能屈;此之谓大丈夫。——《孟子·滕文公下》
前两天看了点mybatis-plus
源码,又看了点手工耿的视频,就有感而发,写下这个工具类
介绍:
平时我们返回给前端返回值就像这样
我们每次都要写这个魔法值key
这样一两个都还好,但久而久之可能会写错
如果我们如果遇到比较长的字段名,可能就要写很长很长,并且没有编译器的提示
所以我照着mybatis-plus
源码,顺便用了它一点方法,写了下面这个工具类
这个工具类里就一个方法和一个接口,以及一个测试的主函数,可以通过Lambda
表达式获取到它的属性名
package com.ruben.utils;/**
* @ClassName: FunctionUtils
* @Date: 2020/11/9 0009 23:43
* @Description:
*/
import com.baomidou.mybatisplus.core.toolkit.ExceptionUtils;
import com.baomidou.mybatisplus.core.toolkit.support.SerializedLambda;
import com.ruben.pojo.User;
import java.io.*;
import java.util.HashMap;
import java.util.Map;
/**
* @ClassName: FunctionUtils
* @Description: 我还没有写描述
* @Date: 2020/11/9 0009 23:43
* *
* @author: <achao1441470436@gmail.com>
* @version: 1.0
* @since: JDK 1.8
*/
public class FunctionUtils {
@FunctionalInterface
public interface IFunction<T, R> extends Serializable {
R apply(T t);
}
public static String getAttributeName(IFunction func) {
if (!func.getClass().isSynthetic()) {
throw ExceptionUtils.mpe("该方法仅能传入 lambda 表达式产生的合成类");
}
SerializedLambda lambda;
try {
ByteArrayOutputStream baos = new ByteArrayOutputStream(1024);
ObjectOutputStream oos = new ObjectOutputStream(baos);
oos.writeObject(func);
oos.flush();
ObjectInputStream objIn = new ObjectInputStream(new ByteArrayInputStream(baos.toByteArray())) {
@Override
protected Class<?> resolveClass(ObjectStreamClass objectStreamClass) throws IOException, ClassNotFoundException {
Class<?> clazz = super.resolveClass(objectStreamClass);
return clazz == java.lang.invoke.SerializedLambda.class ? SerializedLambda.class : clazz;
}
};
lambda = (SerializedLambda) objIn.readObject();
} catch (IOException | ClassNotFoundException e) {
throw new RuntimeException("这不可能发生!", e);
}
String methodName = lambda.getImplMethodName();
return methodName.substring(3, 4).toLowerCase() methodName.substring(4);
}
public static void main(String[] args) {
User user = new User();
Map<String, Object> map = new HashMap<>(1 << 4);
IFunction<User, String> getUsername = User::getUsername;
map.put(FunctionUtils.getAttributeName(getUsername), user.getUsername());
}
}
之后我们就可以使用这种方式去获取属性名,再也不怕写错了
怎么样,是不是很没用?!