基于注解手写Spring的IOC（上）

一、思路

先要从当前类出发找到对应包下的所有类文件，再从这些类中筛选出类上有@MyComponent注解的类；把它们都装入Map中，同时类属性完成@MyValue的赋值操作。

二、具体实现

测试类结构：

测试类：myse、mycontor、BigStar、MyAnneTest结构都一致

@MyComponent("BigStar")
public class BigStar{@MyValue("张三")private String name;@Overridepublic String toString() {return "BigStar{" +"name='" + name + '\'' +'}';}
}

自定义注解类（MyComponent--扫描类）

@Target({ElementType.TYPE}) // 作用范围：类上
@Retention(RetentionPolicy.RUNTIME) // 运行时候生效
public @interface MyComponent {String value() default "";
}

数据注入（MyValue）

@Target({ ElementType.FIELD}) // 使用的范围 字段上
@Retention(RetentionPolicy.RUNTIME) // 生效时候 运行时候
public @interface MyValue {String value() default "";
}

三、代码编写

首先处理获得项目的包路径：也是IOC启动方法

在junit4 中编写：

MyIOC（）方法

   private Map<Object,Object> IOC=new HashMap<>(); // 创建 IOC容器@Testpublic void MyIOC() throws IOException, ClassNotFoundException, InvocationTargetException, NoSuchMethodException, InstantiationException, IllegalAccessException {ClassLoader contextClassLoader =Thread.currentThread().getContextClassLoader(); //当前线程中获得类加载器// System.out.println("包路径："+ BigStar.class.getPackage()); // 获得包数据// 获得包资源// BigStar.class.getPackage().getName()获得包 名字// contextClassLoader.getResources() 获得资源相关的路径Enumeration<URL> resources = contextClassLoader.getResources( BigStar.class.getPackage().getName());// 如果有存在没有被读取过的--包只有一个，所有数据只有一条while (resources.hasMoreElements()){// 获得数据URL url = resources.nextElement();String path = url.getPath(); // 获取包的绝对路径// 包的绝对地址：/C:/Users/kk/IdeaProjects/IOCTest/target/classes/beanSystem.out.println("包的绝对地址："+ path);// 获得地址： /C:/Users/kk/IdeaProjects/IOCTest/target/classes/beanDirectoryToClass(path); // 把包路径传入System.out.println("容器中的数据："+IOC);}}

对路径进行处理：

只需要 “包名.类名字” 即可

DirectoryToClass（）方法

    // 处理类路径--获得 包名字+类名字private  void DirectoryToClass(String Mypath) throws ClassNotFoundException, InvocationTargetException, NoSuchMethodException, InstantiationException, IllegalAccessException {// 获得类路径//  String myclassPath=myclass.getPath();List<Object> list=new ArrayList<>();String dgtoPath=""; //递归路径File files=new File(Mypath); //打开文件if (files.isDirectory()) { //如果是目录for (File myclass : files.listFiles()) { // 遍历目录中的文件String  directoryPath=myclass.getPath(); //获得单个文件路径路径// 路径例如： C:\Users\kk\IdeaProjects\IOCTest\target\classes\bean\Star.class// System.out.println("类文件路径：" + directoryPath);dgtoPath=directoryPath;String path1 = directoryPath.replace("\\", "."); // 把 \替换成 .String path = path1.replace("/", "."); // 把 /替换成 .String[] split = path.split("classes\\."); // 根据classes 切割字符串String classpath = split[split.length - 1]; // 获得包路径路径 com.xx.cc.classif (classpath.contains(".class")) { //  是 .class后缀String[] name = classpath.split("\\.class");//  根据.class切割 ---类加载器只需要包名字+类名字//例如： bean.BigStar 或者bean.test.test2.myse 等System.out.println("包名.类名字："+name[0]);getToClass(name[0]);  //参数传入：获得对象} // TODO 如果是目录的话？递归else {DirectoryToClass(dgtoPath); // 如果是目录就--递归}}}}

难点已经解决了，剩下的就是根据"包名.类名"使用反射创建对象。

getToClass()方法

 // 获得类对象属性private void getToClass(String classPath) throws ClassNotFoundException, NoSuchMethodException, InvocationTargetException, InstantiationException, IllegalAccessException {Class<?> clazz = Class.forName(classPath); // 获得类加载器// System.out.println("创建成功的类："+clazz);try { // 异常处理 即使clazz.getDeclaredConstructor().newInstance(); 遇到接口不能创建对象--程序也不会停止Object bean = clazz.getDeclaredConstructor().newInstance(); // 创建对象--暴力创建MyComponent beanName = clazz.getAnnotation(MyComponent.class); // 获得类上注解if(!Objects.isNull(beanName)){ // 类上有注解的时候String key=beanName.value(); // 获得注解中的value值--类名字//    System.out.println("key名字："+key);Field[] declaredFields = clazz.getDeclaredFields(); // 获得字段for (Field field:declaredFields){  // 遍历字段MyValue value = field.getAnnotation(MyValue.class); // 获得字段上的注解field.setAccessible(true); // 打开权限 暴力注入 狠狠地注入field.set(bean,value.value()); // 设置当前bean对象的字段}IOC.put(key,bean); // 存储到IOC中}}catch (Exception e){System.out.println("不是类："+e); // 如果遇到其他类型文件、接口等}}

四、运行结果

包的绝对地址：/C:/Users/kk/IdeaProjects/IOCTest/target/classes/bean
包名.类名字：bean.BigStar
包名.类名字：bean.MyAnneTest
包名.类名字：bean.MyComponent
不是类：java.lang.NoSuchMethodException: bean.MyComponent.<init>()
包名.类名字：bean.MyValue
不是类：java.lang.NoSuchMethodException: bean.MyValue.<init>()
包名.类名字：bean.test.mycontor
包名.类名字：bean.test.test2.myse
容器中的数据：{mycontor=mycontor{hobb='李四'}, MyAnneTest=MyAnneTest{name='张三', hobby='打篮球'}, BigStar=BigStar{name='张三'}, myse=myse{name='后悔'}}