本人对IOC怎么实现的很好奇 思路完全按照 http://www.54sb.org/博主实现
首先是包结构
讲解一下思路
首先通过@initBean注解注入bean,然后定义一个初始化类的方法,用classUtil扫描包,扫描出包下有initBean注解的类 放入beanContainer容器里面
然后在初始化类的方法里再定义一个方法,该方法用classUtil扫描包,扫描包下有writeBean的字段,再从beanContainer容器里面取出来,然后把取出来的对象给该字段引用。
就是这么简单~但是写代码的严谨性要有
下面上代码
@initBean 相当于@component @service @controller 这些注入bean
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)//class
public @interface InitBean{
String beanName() default "";
}
@writeBean相当于@AutoWire@Resource
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)//字段
public @interface WriteBean {
String beanName() default "";
}
注入两个Service
import annotation.InitBean;
import annotation.WriteBean;
import model.User;
import java.util.List;
@InitBean
public class OrderService {
@WriteBean
private UserService userService;
public void writeUsers(){
List<User> users=userService.getUsers();
for (User user:users) {
System.out.println("用户信息:"+user.toString());
}
}
}
import java.util.ArrayList;
import java.util.List;
@InitBean
public class UserService {
public List<User> getUsers(){
List<User> list=new ArrayList<User>();
User user=new User();
user.setUserName("张三");
user.setPassWord("zhangsan");
list.add(user);
user=new User();
user.setUserName("李四");
user.setPassWord("lisi");
list.add(user);
user=new User();
user.setUserName("王五");
user.setPassWord("wangwu");
list.add(user);
user=new User();
user.setUserName("马六");
user.setPassWord("maliu");
list.add(user);
return list;
}
}
容器
public class BeanContainer {
public static Map<String,Object> beanMap=new HashMap<>();
public static Object getBean(Class cla){
String beanName = cla.getName();
return beanMap.get(beanName);
}
public static Object getBean(String beanName){
return beanMap.get(beanName);
}
}
初始化类
package init;
import annotation.InitBean;
import annotation.WriteBean;
import container.BeanContainer;
import util.ClassUtil;
import util.StringUtils;
import java.lang.reflect.Field;
import java.util.List;
import java.util.Map;
public class InitRun {
//扫描包下有Initbean的类
public void initClass() throws Exception {
List<Class<?>> clas=ClassUtil.getClasses("bean");
for (Class cla:clas) {
InitBean initBean=(InitBean) cla.getAnnotation(InitBean.class);
if(StringUtils.isNullOrEmpty(initBean)){
continue;
}
String beanName=initBean.beanName();
if(StringUtils.isNullOrEmpty(beanName)){
beanName=cla.getName();
}
if(BeanContainer.beanMap.containsKey(beanName)){
throw new Exception("存在重复的beanName:"+beanName);
}
BeanContainer.beanMap.put(beanName, cla.newInstance());
}
}
//扫描有初始化的类下的writeBean的字段
public void initField() throws IllegalAccessException, InstantiationException {
List<Class<?>> clas=ClassUtil.getClasses("bean");
Map<String, Object> currMap=BeanContainer.beanMap;
for (String key:currMap.keySet()) {
Object obj=currMap.get(key);
Field[] fields=obj.getClass().getDeclaredFields();
if(StringUtils.isNullOrEmpty(fields)){
continue;
}
for (Field field:fields) {
WriteBean writeBean=field.getAnnotation(WriteBean.class);
if(StringUtils.isNullOrEmpty(writeBean)){
continue;
}
String beanName=writeBean.beanName();
if(StringUtils.isNullOrEmpty(beanName)){
beanName=field.getType().getName();
}
Object writeValue=null;
field.setAccessible(true);
if(!BeanContainer.beanMap.containsKey(beanName)){
writeValue=field.getClass().newInstance();
BeanContainer.beanMap.put(beanName,writeValue );
}
writeValue=BeanContainer.beanMap.get(beanName);
field.set(obj, writeValue);
}
}
}
}
model 我们用的service操作的实体
package model;
public class User {
private String userName;
private String passWord;
public String getUserName() {
return userName;
}
public void setUserName(String userName) {
this.userName = userName;
}
public String getPassWord() {
return passWord;
}
public void setPassWord(String passWord) {
this.passWord = passWord;
}
@Override
public String toString() {
return "User [userName=" + userName + ", passWord=" + passWord + "]";
}
}
测试类
package test;
import container.BeanContainer;
import bean.OrderService;
import init.InitRun;
public class TestIOC {
public static void main(String[] args) {
InitRun run=new InitRun();
try {
//初始化需要注入的类
run.initClass();
//初始化需要注入的字段
run.initField();
//从对象池获得对象
OrderService orderService=(OrderService) BeanContainer.getBean(OrderService.class);
//调用方法
orderService.writeUsers();
} catch (Exception e) {
e.printStackTrace();
}
}
}
测试结果
StringUtil
package util; import java.util.Collection; import java.util.Map; //import oracle.sql.CLOB; public class StringUtils { public static boolean isNullOrEmpty(Object obj) { try { if (obj == null) return true; if (obj instanceof CharSequence) { return ((CharSequence) obj).length() == 0; } if (obj instanceof Collection) { return ((Collection) obj).isEmpty(); } if (obj instanceof Map) { return ((Map) obj).isEmpty(); } if (obj instanceof Object[]) { Object[] object = (Object[]) obj; if (object.length == 0) { return true; } boolean empty = true; for (int i = 0; i < object.length; i++) { if (!isNullOrEmpty(object[i])) { empty = false; break; } } return empty; } return false; } catch (Exception e) { return true; } } public static void main(String[] args) { String a = ""; System.out.println(isNullOrEmpty(a)); } }
ClassUtil比较复杂 看要用到的方法就行 博主说百度一大堆
package util; import java.io.File; import java.io.FileFilter; import java.io.IOException; import java.net.JarURLConnection; import java.net.URL; import java.net.URLDecoder; import java.text.MessageFormat; import java.util.ArrayList; import java.util.Enumeration; import java.util.HashSet; import java.util.List; import java.util.Set; import java.util.jar.JarEntry; import java.util.jar.JarFile; /** * 类相关的工具类 * * @author <a href="mailto:ohergal@gmail.com">ohergal</a> * */ public class ClassUtil { private static boolean checkStr(String str) { String[] record = splitString(String.valueOf(str)); Set checkSet = new HashSet(); for (String tmp : record) { checkSet.add(tmp); } if (checkSet.size() == record.length) { return true; } return false; } public static String[] splitString(String str) { if (str==null||"".equals(str)) { return null; } String[] finalStrs = new String[str.length()]; for (int i = 0; i < str.length(); i++) { finalStrs[i] = str.substring(i, i + 1); } return finalStrs; } /** * 取得某个接口下所有实现这个接口的类 * */ public static List<Class> getAllClassByInterface(Class c) { List<Class> returnClassList = null; if (c.isInterface()) { // 获取当前的包名 String packageName = c.getPackage().getName(); // 获取当前包下以及子包下所以的类 List<Class<?>> allClass = getClasses(packageName); if (allClass != null) { returnClassList = new ArrayList<Class>(); for (Class classes : allClass) { // 判断是否是同一个接口 if (c.isAssignableFrom(classes)) { // 本身不加入进去 if (!c.equals(classes)) { returnClassList.add(classes); } } } } } return returnClassList; } /* * 取得某一类所在包的所有类名 不含迭代 */ public static String[] getPackageAllClassName(String classLocation, String packageName) { // 将packageName分解 String[] packagePathSplit = packageName.split("[.]"); String realClassLocation = classLocation; int packageLength = packagePathSplit.length; for (int i = 0; i < packageLength; i++) { realClassLocation = realClassLocation + File.separator + packagePathSplit[i]; } File packeageDir = new File(realClassLocation); if (packeageDir.isDirectory()) { String[] allClassName = packeageDir.list(); return allClassName; } return null; } /** * 从包package中获取所有的Class * * @param pack * @return */ public static List<Class<?>> getClasses(String packageName) { // 第一个class类的集合 List<Class<?>> classes = new ArrayList<Class<?>>(); // 是否循环迭代 boolean recursive = true; // 获取包的名字 并进行替换 String packageDirName = packageName.replace('.', '/'); // 定义一个枚举的集合 并进行循环来处理这个目录下的things Enumeration<URL> dirs; try { dirs = Thread.currentThread().getContextClassLoader() .getResources(packageDirName); // 循环迭代下去 while (dirs.hasMoreElements()) { // 获取下一个元素 URL url = dirs.nextElement(); // 得到协议的名称 String protocol = url.getProtocol(); // 如果是以文件的形式保存在服务器上 if ("file".equals(protocol)) { // 获取包的物理路径 String filePath = URLDecoder.decode(url.getFile(), "UTF-8"); // 以文件的方式扫描整个包下的文件 并添加到集合中 findAndAddClassesInPackageByFile(packageName, filePath, recursive, classes); } else if ("jar".equals(protocol)) { // 如果是jar包文件 // 定义一个JarFile JarFile jar; try { // 获取jar jar = ((JarURLConnection) url.openConnection()) .getJarFile(); // 从此jar包 得到一个枚举类 Enumeration<JarEntry> entries = jar.entries(); // 同样的进行循环迭代 while (entries.hasMoreElements()) { // 获取jar里的一个实体 可以是目录 和一些jar包里的其他文件 如META-INF等文件 JarEntry entry = entries.nextElement(); String name = entry.getName(); // 如果是以/开头的 if (name.charAt(0) == '/') { // 获取后面的字符串 name = name.substring(1); } // 如果前半部分和定义的包名相同 if (name.startsWith(packageDirName)) { int idx = name.lastIndexOf('/'); // 如果以"/"结尾 是一个包 if (idx != -1) { // 获取包名 把"/"替换成"." packageName = name.substring(0, idx) .replace('/', '.'); } // 如果可以迭代下去 并且是一个包 if ((idx != -1) || recursive) { // 如果是一个.class文件 而且不是目录 if (name.endsWith(".class") && !entry.isDirectory()) { // 去掉后面的".class" 获取真正的类名 String className = name.substring( packageName.length() + 1, name.length() - 6); try { // 添加到classes classes.add(Class .forName(packageName + '.' + className)); } catch (ClassNotFoundException e) { e.printStackTrace(); } } } } } } catch (IOException e) { e.printStackTrace(); } } } } catch (IOException e) { e.printStackTrace(); } return classes; } /** * 以文件的形式来获取包下的所有Class * * @param packageName * @param packagePath * @param recursive * @param classes */ public static void findAndAddClassesInPackageByFile(String packageName, String packagePath, final boolean recursive, List<Class<?>> classes) { // 获取此包的目录 建立一个File File dir = new File(packagePath); // 如果不存在或者 也不是目录就直接返回 if (!dir.exists() || !dir.isDirectory()) { return; } // 如果存在 就获取包下的所有文件 包括目录 File[] dirfiles = dir.listFiles(new FileFilter() { // 自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件) public boolean accept(File file) { return (recursive && file.isDirectory()) || (file.getName().endsWith(".class")); } }); // 循环所有文件 for (File file : dirfiles) { // 如果是目录 则继续扫描 if (file.isDirectory()) { findAndAddClassesInPackageByFile( packageName + "." + file.getName(), file.getAbsolutePath(), recursive, classes); } else { // 如果是java类文件 去掉后面的.class 只留下类名 String className = file.getName().substring(0, file.getName().length() - 6); try { // 添加到集合中去 classes.add(Class.forName(packageName + '.' + className)); } catch (ClassNotFoundException e) { e.printStackTrace(); } } } } }