一、思路
先要从当前类出发找到对应包下的所有类文件,再从这些类中筛选出类上有@MyComponent注解的类;把它们都装入Map中,同时类属性完成@MyValue的赋值操作。
二、具体实现
测试类结构:
测试类:myse、mycontor、BigStar、MyAnneTest结构都一致
@MyComponent("BigStar")
public class BigStar{
@MyValue("张三")
private String name;
@Override
public 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容器
@Test
public 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/bean
System.out.println("包的绝对地址:"+ path);
// 获得地址: /C:/Users/kk/IdeaProjects/IOCTest/target/classes/bean
DirectoryToClass(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.class
if (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='后悔'}}
五、后言
大多少自之定义注解都需要扫描包这一步,把getToClass()方法换一下就是其他的功能。ioc真正关键的还是@Autowired注解的实现,我放到下篇讲解。