代码

package com.shunneng.platform.web.rest.util;import javax.persistence.Column;
import java.io.Serializable;
import java.lang.invoke.SerializedLambda;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.function.Function;public class ColumnUtil {/*** 使Function获取序列化能力*/@FunctionalInterfacepublic interface SFunction<T, R> extends Function<T, R>, Serializable {}//默认配置static String defaultSplit = "";static Integer defaultToType = 0;/*** 获取实体类的字段名称(实体声明的字段名称)*/public static <T> String getFieldName(SFunction<T, ?> fn) {return getFieldName(fn, defaultSplit);}public static <T> String getEntityFieldName(SFunction<T, ?> fn) {return getEntityFieldName(fn, defaultSplit);}/*** 获取实体类的字段名称** @param split 分隔符，多个字母自定义分隔符*/public static <T> String getFieldName(SFunction<T, ?> fn, String split) {return getFieldName(fn, split, defaultToType,true);}public static <T> String getEntityFieldName(SFunction<T, ?> fn, String split) {return getFieldName(fn, split, defaultToType,false);}/*** 获取实体类的字段名称** @param split  分隔符，多个字母自定义分隔符* @param toType 转换方式，多个字母以大小写方式返回 0.不做转换 1.大写 2.小写*/public static <T> String getFieldName(SFunction<T, ?> fn, String split, Integer toType, Boolean isAnon) {SerializedLambda serializedLambda = getSerializedLambda(fn);// 从lambda信息取出method、field、class等String fieldName = serializedLambda.getImplMethodName().substring("get".length());fieldName = fieldName.replaceFirst(fieldName.charAt(0) + "", (fieldName.charAt(0) + "").toLowerCase());Field field;try {field = Class.forName(serializedLambda.getImplClass().replace("/", ".")).getDeclaredField(fieldName);} catch (ClassNotFoundException | NoSuchFieldException e) {throw new RuntimeException(e);}// 从field取出字段名if(isAnon){Column tableField = field.getAnnotation(Column.class);if (tableField != null && tableField.name().length() > 0) {return tableField.name();}}//0.不做转换 1.大写 2.小写switch (toType) {case 1:return fieldName.replaceAll("[A-Z]", split + "$0").toUpperCase();case 2:return fieldName.replaceAll("[A-Z]", split + "$0").toLowerCase();default:return fieldName.replaceAll("[A-Z]", split + "$0");}}private static <T> SerializedLambda getSerializedLambda(SFunction<T, ?> fn) {// 从function取出序列化方法Method writeReplaceMethod;try {writeReplaceMethod = fn.getClass().getDeclaredMethod("writeReplace");} catch (NoSuchMethodException e) {throw new RuntimeException(e);}// 从序列化方法取出序列化的lambda信息boolean isAccessible = writeReplaceMethod.isAccessible();writeReplaceMethod.setAccessible(true);SerializedLambda serializedLambda;try {serializedLambda = (SerializedLambda) writeReplaceMethod.invoke(fn);} catch (IllegalAccessException | InvocationTargetException e) {throw new RuntimeException(e);}writeReplaceMethod.setAccessible(isAccessible);return serializedLambda;}
}

使用方式

ColumnUtil.getEntityFieldName(PlatformConfig::getHosId)