欢迎来到代码驿站!

JAVA代码

当前位置:首页 > 软件编程 > JAVA代码

Java8通过Function获取字段名的步骤

时间:2021-10-30 10:32:49|栏目:JAVA代码|点击:

摘要:Java8通过Function获取字段名,解决硬编码,效果类似于mybatis-plus的LambdaQueryWrapper。

本文总共三个步骤:
1、使Function获取序列化能力;
2、通过SFunction获取字段名;
3、建一些业务代码进行测试;

使Function获取序列化能力

import java.io.Serializable;
import java.util.function.Function;

/**
 * 使Function获取序列化能力
 */
@FunctionalInterface
public interface SFunction<T, R> extends Function<T, R>, Serializable {
}

通过SFunction获取字段名

import java.lang.invoke.SerializedLambda;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

public class ColumnUtil {

  public static <T> String getName(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);

    // 从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取出字段名,可以根据实际情况调整
    TableField tableField = field.getAnnotation(TableField.class);
    if (tableField != null && tableField.value().length() > 0) {
      return tableField.value();
    } else {
      return fieldName.replaceAll("[A-Z]", "_$0").toLowerCase();
    }
  }
}

建一些业务代码进行测试

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 TableField {
  String value() default "";
}
import java.io.Serializable;

/**
 * 用户实体类。测试用
 */
public class User implements Serializable {

  private String loginName;

  @TableField("nick")
  private String nickName;

  public String getLoginName() {
    return loginName;
  }

  public void setLoginName(String loginName) {
    this.loginName = loginName;
  }

  public String getNickName() {
    return nickName;
  }

  public void setNickName(String nickName) {
    this.nickName = nickName;
  }
}
/**
 * 测试用
 */
public class Test {
  public static void main(String[] args) {
    System.out.println("字段名:" + ColumnUtil.getName(User::getLoginName));
    System.out.println("字段名:" + ColumnUtil.getName(User::getNickName));
  }
}

运行结果:

字段名:login_name
字段名:nick

上一篇:Java下SpringBoot创建定时任务详解

栏    目:JAVA代码

下一篇:基于Mybatis plus 自动代码生成器的实现代码

本文标题:Java8通过Function获取字段名的步骤

本文地址:http://www.codeinn.net/misctech/181800.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有