实现功能
1.我们看到@ComponentScan注解一个开始定义就是需要支持,扫描多个包,将多个包的类名获取到。现在就实现这个功能。
实现思路
根据传入的字符串数组,获得多个包下的类全限制名。
实现步骤
--基于之前的代码--
1.在PackageUtils.增加一个扫描多个包的方法
/**
* 支持同时扫描多个包的路径下的类全限制名以及其子包的所有类的全限制名 包名直接使用逗号分割
*
* @param packageNames:传入多个包名
* @param isRecursion:是否扫描包的子包
* @return 返回所以扫描到类全限制名
*/
public static Set<String> getClassNames(String[] packageNames, boolean isRecursion) {
//1.创建一个Set集合将每个包下的类放在一起。
Set<String> classNamesSet=new HashSet<String>();
// 数值的长度
int length = packageNames.length;
for (int i = 0; i < length; i++) {
Set<String> classNameSet = PackageUtils.getClassName(packageNames[i], isRecursion);
//每次都将包放在总的Set集合中
classNamesSet.addAll(classNameSet);
}
return classNamesSet;
}
2.修改AbstractApplicationContext的构造函数,标红处
public AbstractApplicationContext(Class<?> classType) {
//判断配置类是否有Configuration注解
Configuration annotation = classType.getAnnotation(Configuration.class);
if(annotation!=null){
//获得组件扫描注解
ComponentScan componentScan = classType.getDeclaredAnnotation(ComponentScan.class);
//获得包名
this.basePackage = componentScan.basePackages();
//根据包名获得类全限制名
//Set<String> classNames = PackageUtils.getClassName(this.basePackage[0], true);
11 //将扫描一个包,修改为多个包
12 Set<String> classNames = PackageUtils.getClassNames(this.basePackage, true);
//通过类名创建对象
Iterator<String> iteratorClassName = classNames.iterator();
while(iteratorClassName.hasNext()){
String className = iteratorClassName.next();
try {
//通过类全名创建对象
Object instance = Class.forName(className).newInstance();
//将对象加到容器中,对象名就类全名
this.getContext().addObject(instance.getClass().getSimpleName(),instance);
} catch (InstantiationException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
}
}
测试代码
1.Config类-,扫描ioc.core.test.config包和ioc.core.test.service包
package ioc.core.test.config; import ioc.core.annotation.ComponentScan;
import ioc.core.annotation.Configuration; //使用定义@Configuration定义该类是一个配置类
@Configuration
//使用ComponentScan设置扫描包的路径
@ComponentScan(basePackages={"ioc.core.test.config","ioc.core.test.service"})
public class Config { }
2.测试类,输出容器中的对象
package ioc.core.test; import org.junit.Test; import ioc.core.impl.AnntationApplicationContext;
import ioc.core.test.config.Config;
public class AnntationApplicationContextTest { @Test
public void login(){
try {
AnntationApplicationContext context=new AnntationApplicationContext(Config.class);
System.out.println(context.getContext().getObjects()); } catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
} }
3.测试结果,将两个包的类的对象都放在了容器里面。成功。