java获取类注解内容和类字段注解内容

最新推荐文章于 2024-04-02 14:04:52 发布

原创最新推荐文章于 2024-04-02 14:04:52 发布 · 1.4k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#annotation #java

本文介绍了一个Java工具类，该工具类利用反射技术从指定类中提取字段信息，并将这些信息转换为便于操作的数据结构。主要实现了从实体类中获取字段名称及表注解的功能。

package com.dev.demo.util;

import com.baomidou.mybatisplus.annotation.TableField;
import com.dev.demo.model.TbUser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.persistence.Column;
import javax.persistence.Table;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class BeanUtil {
    static Logger log = LoggerFactory.getLogger(BeanUtil.class);

    /**
     * 获取table中的columns和类中的字段集合
     *
     * @param clazz
     * @return
     */
    public static List<DbFiled> getFieldName(Class clazz) {
        List<DbFiled> list = new ArrayList<>();
        Field[] fields = clazz.getDeclaredFields();
        for (int j = 0; j < fields.length; j++) {
            Field field = fields[j];
            if (field.isAnnotationPresent(Column.class)) {
                if (field.isAnnotationPresent(TableField.class)) {
                    try {
                        //获取SignalsRuleExportDTO字段上的ExcelProperty注解实例
                        TableField column = field.getAnnotation(TableField.class);
                        //获取 ExcelProperty 这个代理实例所持有的 InvocationHandler
                        InvocationHandler ch = Proxy.getInvocationHandler(column);
                        // 获取 AnnotationInvocationHandler 的 memberValues 字段
                        Field fv = ch.getClass().getDeclaredField("memberValues");
                        // 因为这个字段事 private final 修饰，所以要打开权限
                        fv.setAccessible(true);
                        // 获取 memberValues
                        Map kvs = (Map) fv.get(ch);
                        // 修改 value 属性值
                        Object columnName = kvs.get("value");

                        list.add(new DbFiled(columnName.toString(), field.getName()));
                    } catch (Exception e) {
                        log.error("getFieldName", e);
                    }
                } else {
                    list.add(new DbFiled(field.getName(), field.getName()));
                }
            }
        }
        return list;
    }

    /**
     * 获取table中的columns和类中的字段集合(包括表名前提model配置@Table注解且注解包含name属性)
     *
     * @param clazz
     * @return
     */
    public static TableClassBean getTableClassBean(Class clazz) {
        List<DbFiled> list = new ArrayList<>();
        Field[] fields = clazz.getDeclaredFields();
        for (int j = 0; j < fields.length; j++) {
            Field field = fields[j];
            if (field.isAnnotationPresent(Column.class)) {
                if (field.isAnnotationPresent(TableField.class)) {
                    try {
                        //获取SignalsRuleExportDTO字段上的ExcelProperty注解实例
                        TableField column = field.getAnnotation(TableField.class);
                        //获取 ExcelProperty 这个代理实例所持有的 InvocationHandler
                        InvocationHandler ch = Proxy.getInvocationHandler(column);
                        // 获取 AnnotationInvocationHandler 的 memberValues 字段
                        Field fv = ch.getClass().getDeclaredField("memberValues");
                        // 因为这个字段事 private final 修饰，所以要打开权限
                        fv.setAccessible(true);
                        // 获取 memberValues
                        Map kvs = (Map) fv.get(ch);
                        // 修改 value 属性值
                        Object columnName = kvs.get("value");

                        list.add(new DbFiled(columnName.toString(), field.getName()));
                    } catch (Exception e) {
                        log.error("getFieldName", e);
                    }
                } else {
                    list.add(new DbFiled(field.getName(), field.getName()));
                }
            }
        }
        Table[] annotationsByType = TbUser.class.getAnnotationsByType(Table.class);
        if (annotationsByType != null && annotationsByType.length >= 1) {
            Table table = annotationsByType[0];
            if (table.name() != null && !"".equals(table.name().trim())) {
                return new TableClassBean(table.name(), list);
            }
        }
        log.info("no annotation or javax.persistence.Table");
        return new TableClassBean(TbUser.class.getSimpleName(), list);
    }

    public static void main(String[] args) {
        System.out.println(getFieldName(TbUser.class));
        System.out.println(getTableClassBean(TbUser.class));
    }
}

详细参看：https://blog.youkuaiyun.com/CTLLIN/article/details/110958293