简单实现IOC

本人对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();
            }
         }
      }
   }
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值