com.anwen.mongo.toolkit.AnnotationUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongo-plus-core Show documentation

mongo-plus核心代码

There is a newer version: 2.1.6.1

package com.anwen.mongo.toolkit;

import com.anwen.mongo.annotation.ID;
import com.anwen.mongo.enums.IdTypeEnum;
import com.anwen.mongo.model.BaseModelID;

import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.Map;

/**
 * @author JiaChaoYang
 * 注解操作
 * @since 2023-02-13 13:59
 **/
public class AnnotationUtil {

    public static Map getFieldAnnotation(Object object) {
        Field[] fields = object.getClass().getDeclaredFields();
        Map resultMap = new HashMap<>();
        Class superclass = object.getClass().getSuperclass();
        IdTypeEnum idTypeEnum = null;
        String fieldName = "id";
        Class fieldType = String.class;
        for (Field field : fields) {
            if (field.isAnnotationPresent(ID.class)){
                idTypeEnum = field.getAnnotation(ID.class).type();
                fieldName = field.getName();
                fieldType = field.getType();
            }
        }
        if (superclass == BaseModelID.class){
            try {
                idTypeEnum = superclass.getField("id").getAnnotation(ID.class).type();
            } catch (NoSuchFieldException e) {
                throw new RuntimeException(e);
            }
        }
        if (idTypeEnum != null){
            resultMap.put("fieldName",fieldName);
            resultMap.put("fieldType",fieldType);
            resultMap.put("generateType", idTypeEnum);
        }
        return resultMap;
    }
}