复盘:返回不同模板的数据(注解、反射、单例模式、工厂模式、递归)
工作中一个接口需要动态加载不同类型的mapper类,mapper获取实体类,返回给前端。
自定义注解 @Subscribe,Mapper添加@Subscribe(desc = "xxxxxx"),维护不同类型与Mapper的对应关系。
image.png
获取所有Mapper。所有需要用的mapper实现自定义的TemplateMapper接口,通过反射获取所有该接口的实现类,(要求:接口与实现类在同一目录下)
image.png
获取Mapper实例。基于以上1,2,维护Map<String,Class>类,通过前端传递type,获取对应class,然后通过ApplicationContext.getBean(Class)返回对应mapper实例。
public class MapperFactory { public static Map<String, Class> typeClassMap; public static Fruits selectByIdAndType(String type, Long instanceId) { Class aClass = MapperFactory.getTypeClassMap().get(type); TemplateMapper mapper = (TemplateMapper) Application.applicationContext.getBean(aClass); return (Fruits) mapper.selectById(instanceId); } /** * typeClassMap 单例-懒加载。 * * @return */ public static synchronized Map<String, Class> getTypeClassMap() { if (typeClassMap == null) { typeClassMap = new HashMap<>(); // 获取所有TemplateMapper接口所有实现类 List<Class> classList = getAllAchieveClass(Fruits.class, true); for (Class aClass : classList) { Subscribe annotation = (Subscribe) aClass.getAnnotation(Subscribe.class); typeClassMap.putIfAbsent(annotation.desc(), aClass); } } return typeClassMap; } /** * 获取所有clazz接口的实现类. * * @param isAbstract 是否包含抽象类(抽象类和接口),true 包含,false 不包含 * @return */ private static List<Class> getAllAchieveClass(Class clazz, Boolean isAbstract) { ArrayList<Class> list = new ArrayList<>(); //判断是否是接口 if (clazz.isInterface()) { // 获取clazz接口类所在路径下的所有实现类 ArrayList<Class> allClass = getAllClassByPath(clazz.getPackage().getName()); // 循环判allClass否实现了指定的接口,并且排除接口类自己 for (int i = 0; i < allClass.size(); i++) { if (!isAbstract) { if (Modifier.isAbstract(allClass.get(i).getModifiers())) { continue; } } //判断是不是同一个接口 if (clazz.isAssignableFrom(allClass.get(i))) { if (!clazz.equals(allClass.get(i))) { list.add(allClass.get(i)); } } } } return list; } /** * 从指定路径下获取所有类 * * @return */ private static ArrayList<Class> getAllClassByPath(String packageName) { ArrayList<Class> list = new ArrayList<>(); ClassLoader classLoader = Thread.currentThread().getContextClassLoader(); String path = packageName.replace('.', '/'); try { ArrayList<File> fileList = new ArrayList<>(); Enumeration<URL> enumeration = classLoader.getResources(path); while (enumeration.hasMoreElements()) { URL url = enumeration.nextElement(); fileList.add(new File(url.getFile())); } for (int i = 0; i < fileList.size(); i++) { list.addAll(findClass(fileList.get(i), packageName)); } } catch (IOException e) { e.printStackTrace(); } return list; } /** * 如果file是文件夹,则递归调用findClass方法 * 如果file本身是类文件,则加入list中进行保存,并返回 * * @param file * @param packageName * @return */ private static ArrayList<Class> findClass(File file, String packageName) { ArrayList<Class> list = new ArrayList<>(); if (!file.exists()) { return list; } File[] files = file.listFiles(); for (File file2 : files) { if (file2.isDirectory()) { //添加断言用于判断 assert !file2.getName().contains("."); ArrayList<Class> arrayList = findClass(file2, packageName + "." + file2.getName()); list.addAll(arrayList); } else if (file2.getName().endsWith(".class")) { try { //保存的类文件不需要后缀.class list.add(Class.forName(packageName + '.' + file2.getName().substring(0, file2.getName().length() - 6))); } catch (ClassNotFoundException e) { e.printStackTrace(); } } } return list; }}
以上解决方案并不完美,有缺陷,getAllClassByPath获取指定路径下所有类,在jar包运行环境下失效。
改进
自定义注解 @Subscribe,Mapper添加@Subscribe(desc = "xxxxxx"),维护不同类型与Mapper的对应关系。
获取@Subscribe所有class类。使用反射框架Reflections,可以解决。
<dependency> <groupId>org.reflections</groupId> <artifactId>reflections</artifactId> <version>0.9.10</version></dependency>
示例代码
Map<String, Class> typeClassMap = new HashMap<>(); Reflections reflections = new Reflections("com.xxxxx.xxxxxx"); Set<Class<?>> typesAnnotatedWith = reflections.getTypesAnnotatedWith(Subscribe.class); typesAnnotatedWith.stream().forEach(aClass -> { Subscribe annotation = (Subscribe) aClass.getAnnotation(Subscribe.class); typeClassMap.putIfAbsent(annotation.desc(), aClass); });
获取Mapper实例。基于以上1,2,维护Map<String,Class>类,通过前端传递type,获取对应class,然后通过ApplicationContext.getBean(Class)返回对应mapper实例。
复盘,成长,加油!
作者:lconcise
链接:https://www.jianshu.com/p/55ee3e6966c7