模拟Spring扫描包

悠悠 2023-07-17 15:58 91阅读 0赞

## 模拟Spring扫描包 ##

`转载的是老师的博客,只是写了一些注释,便于理解。原博客地址: https://blog.csdn.net/kongfanyu/article/details/103930414`

通过自定义注解扫描包及子包下的所有类,并通过反射创建对象.

### 定义注解 ###

package com.hanker.domain12;
    
    import java.lang.annotation.ElementType;
    import java.lang.annotation.Retention;
    import java.lang.annotation.RetentionPolicy;
    import java.lang.annotation.Target;
    
    /** * @Retention 表示定义的注解何时有效, RUNTIME运行时有效 * @Target 表示定义的注解运行在哪里, TYPE应用在类上, METHOD应用在方法上, FIELD应用在类上 */
    @Target(ElementType.TYPE)
    @Retention(RetentionPolicy.RUNTIME)
    @interface Component { 
    
    }
    //========================
    
    
    /** * {ElementType.TYPE}的花括号可以省略，如:ElementType.TYPE,因为只有一个值 * {ElementType.TYPE,ElementType.METHOD,ElementType.FIELD} */
    @Retention(RetentionPolicy.RUNTIME)
    @Target({ ElementType.TYPE})
    @interface Controller { 
    
    }
    
    @Retention(RetentionPolicy.RUNTIME)
    @Target({ ElementType.TYPE})
    @interface Service { 
    
    }

### 业务类 ###

package com.hanker.domain12;
    
    public interface UserService { 
        void saveUser();
    }
    
    
    @Service
    class UserServiceImpl implements UserService { 
        @Override
        public void saveUser() { 
            System.out.println("添加用户");
        }
    }
    // 通过Controller注解来创建UserController 实例
    @Controller
    class UserController { 
        public void login() { 
            System.out.println("用户登录控制器");
        }
    }

### 包扫描 ###

package com.hanker.domain12;
    
    import java.io.File;
    import java.io.FileFilter;
    import java.net.URL;
    import java.util.HashMap;
    import java.util.Map;
    
    public class AnnotationApplicationContext { 
    
        private Map<String, Object> factory = new HashMap<>();
    
        public AnnotationApplicationContext() { 
            //包名
            String pkg = "com.hanker.domain12";
            scanPackage(pkg);
        }
    
        //包扫描方法
        private void scanPackage(final String pkg) { 
            /* * replaceAll() 是用来替换字符串的,第一个参数都是 匹配字符串的正则表达式 第二个参数是用来替换每个匹配项的字符串 * 成功则返回替换的字符床,识别则返回原始字符串 * \\.实际上被转义为两次，\\在java中被转换为一个'\'字符，然后'\.'被传给正则，\.表示对点字符进行转义，使.就表示字符'.',而不使用它在正则中的特殊意义 * */
            String pkgDir = pkg.replaceAll("\\.", "/");
            URL url = getClass().getClassLoader().getResource(pkgDir);
            // url 的值 是"file:/D:/work/Idea-work/jdbc_01/out/production/jdbc_01/com/hanker/domain12"
            System.out.println("--url.getFile()-" + url.getFile());
            //url.getFile 的值 是"/D:/work/Idea-work/jdbc_01/out/production/jdbc_01/com/hanker/domain12"
            File file = new File(url.getFile());
            // file 的值 D:\work\Idea-work\jdbc_01\out\production\jdbc_01\com\hanker\domain12
            /*java.io.File.listFiles(FileFilter filter) 返回抽象路径名数组，表示在目录中此抽象路径名表示，满足指定过滤器的文件和目录。 * * */
            //listFiles()方法是返回某个目录下所有文件和目录的绝对路径，返回的是File数组
            //过滤
            File fs[] = file.listFiles(new FileFilter() { 
                @Override
                public boolean accept(File file) { 
                    String fname = file.getName();
                    //isDirectory()是检查一个对象是否是文件夹。返回值是boolean类型的。如果是则返回true，否则返回false。
                    //是目录则继续调用自身
                    if (file.isDirectory()) { 
                        scanPackage(pkg + "." + fname);
                    } else { 
                        //判断文件名后缀是否.class
                        if (fname.endsWith(".class")) { 
                            return true;
                        }
                    }
                    return false;
                }
            });
            //循环所有class类
            for (File f : fs) { 
                String fname = f.getName();
                System.out.println("f.getName=========" + f.getName());
                //去除.class以后的文件名
                fname = fname.substring(0, fname.lastIndexOf("."));
                //将名字的第一字母转换为小写(用它作为key存储map)
                //将下标为0的字符转换为小写 在 + 从下标为1 开始 组成一个新的字符串
                String key = String.valueOf(fname.charAt(0)).toLowerCase() + fname.substring(1);
                //构建一个全类名(包名.类名)
                String pkgCls = pkg + "." + fname;
                try { 
                    //通过全类名反射创建对象
                    Class<?> c = Class.forName(pkgCls);
                    //判定类上是否有注解isAnnotationPresent()
                    if (c.isAnnotationPresent(Controller.class) ||
                            c.isAnnotationPresent(Service.class) ||
                            c.isAnnotationPresent(Component.class)) { 
                        Object obj = c.newInstance();
                        //将对象放到map容器
                        factory.put(key, obj);
                    }
                } catch (Exception e) { 
                    throw new RuntimeException(e);
                }
            }
        }
    
        //通过键值(小写开头的类名)获取对象
        public Object getBean(String key) { 
            return factory.get(key);
        }
    
        //初始化map
        public void close() { 
            //clear是清空整个map
            factory.clear();
            factory = null;
        }
    }

### 测试类 ###

package com.hanker.domain12;
    
    import org.junit.Test;
    
    public class TestApp { 
        @Test
        public void Test() { 
            AnnotationApplicationContext ctx = new AnnotationApplicationContext();
            UserController obj = (UserController) ctx.getBean("userController");
            System.out.println(obj);
            obj.login();
            UserServiceImpl obj2 = (UserServiceImpl) ctx.getBean("userServiceImpl");
            System.out.println(obj2);
            obj2.saveUser();
            ctx.close();
        }
    }

### 执行效果 ###

![在这里插入图片描述][20200323105513494.png_pic_center]

可以看到已经创建了对象,并且调用了其方法

[20200323105513494.png_pic_center]: /images/20230528/5a3ae2e4240c4b4fa39f3963a6d4d3af.png