io.github.util.ClassUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of assistant-util Show documentation
There is a newer version: 1.1.2
package io.github.util;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.*;

/**
 *    Genius
 *   2023/07/21 09:56
 **/
public class ClassUtil {

    /**
     * 获得某一个包下的所有类
     */
    public static List getClassesInPackage(String packageName)throws IOException {
        List classNames = new ArrayList<>();

        String packagePath = packageName.replace(".", "/");
        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();

        var resources = classLoader.getResources(packagePath);
        while (resources.hasMoreElements()) {
            var resource = resources.nextElement();
            File file = new File(resource.getFile());
            if (file.isDirectory()) {
                scanClassesInDirectory(packageName, file, classNames);
            }
        }

        return classNames;
    }

    //对包名进行分割
    private static void scanClassesInDirectory(String packageName, File directory, List classNames) {
        for (File file : directory.listFiles()) {
            if (file.isFile()) {
                String className = packageName + "." + file.getName().replace(".class", "");
                classNames.add(className);
            } else if (file.isDirectory()) {
                scanClassesInDirectory(packageName + "." + file.getName(), file, classNames);
            }
        }
    }


    /**
     * 类 to Map（浅转换，不考虑继承）
     * @param obj 要转换的类
     * @return 类对应的映射map
     * input:User.class(name:xxxx,age:xx)
     * output: (name:xxxx,age:xx)
     */
    public static Map toMap(Object obj) throws IllegalAccessException {
        Map map = new HashMap<>();
        Class clazz = obj.getClass();

        for (Field field : clazz.getDeclaredFields()) {
            field.setAccessible(true);
            map.put(field.getName(), field.get(obj));
        }

        return map;
    }

    /**
     * 类 to Map（深转换，考虑继承）
     * @param obj 要转换的类
     * @return 类对应的映射map
     * input:User.class(name:xxxx,age:xx)
     * output: (name:xxxx,age:xx)
     */
    public static Map toDeepMap(Object obj) throws IllegalAccessException {
        Map map = new HashMap<>();
        Class clazz = obj.getClass();

        // 获取当前类的字段
        for (Field field : clazz.getDeclaredFields()) {
            field.setAccessible(true);
            map.put(field.getName(), field.get(obj));
        }

        // 如果有父类且不是 Object 类，递归调用获取父类的字段
        Class superClass = clazz.getSuperclass();
        if (superClass != null && !superClass.equals(Object.class)) {
            try {
                Object superObj = superClass.getDeclaredConstructor().newInstance();
                copyFields(superObj, obj, superClass);
                map.putAll(toDeepMap(superObj));
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        return map;
    }

    private static void copyFields(Object dest, Object source, Class clazz) throws IllegalAccessException {
        for (Field field : clazz.getDeclaredFields()) {
            field.setAccessible(true);
            Field destField;
            try {
                destField = dest.getClass().getDeclaredField(field.getName());
                destField.setAccessible(true);
                destField.set(dest, field.get(source));
            } catch (NoSuchFieldException e) {
                // 忽略字段不存在的异常
            }
        }
    }

}