自己动手写spring（三） 支持注解方式

前言

本文使用注解方式来读取bean类，并解决bean之间的依赖。其中Component和Value注解直接copy自Spring，Resource注解来自javax。

扫描包中的所有类

还是以前文提到的beanA和beanB两个model类为例（省略set和get方法）。

@Component
public class BeanA {
	@Value("studentA")
	private String title;
	@Resource
	private BeanB beanB;
}

@Component
public class BeanB {
	@Value("studentB")
	private String title;    	
}

根据前文，我们可以发现，要实现bean工厂功能，首先要将bean信息加载到内存（由配置文件方式转化为“以类的方式”存储）。
在本文中，虽然配置文件没了，但思路是一致的。我们要根据注解，采集bean信息，存储在map中。注解和配置文件方式的一个不同是：配置文件是配什么bean加载什么，而注解方式，则需要我们扫描目标pacakge中的所有bean。

public class Main {
	private static Map<String,Class> beanId2Clazz = new HashMap<String,Class>();
	// 将一个字符串的首字符小写
	private static String smallCaseFirstChar(String prop){
		char ch = (char) (prop.charAt(0) + 32);
		return prop.replaceFirst(prop.charAt(0) + "", ch + "");
	}
	public static void main(String[] args) throws Exception {
		// 扫描包中包含的类（即扫描目的包所在文件的子文件）
		String packageName = "org.lqk.lspring.bean";
		String rootPath = Main3.class.getResource("/").getPath();
		File file = new File(rootPath + File.separator +  packageName.replace(".", File.separator));
		String[] fileNames = file.list();
		// 扫描
		for(String fileName : fileNames){
			String className = fileName.substring(0,fileName.length() - ".class".length());
			String fullClassName = packageName + "." + className;
			Class clazz = Class.forName(fullClassName);
			Component cop = (Component) clazz.getAnnotation(Component.class);
			if(null != cop){
				String beanId = cop.value();
				if(StringUtils.isEmpty(beanId)){
					beanId = smallCaseFirstChar(className);
				}
				beanId2Clazz.put(beanId, clazz);
			}
		}
	}
}

关于扫描一个包中所有的类，有现成的org.reflections包，此处为了减少##读者的理解曲线，就不提了。

当加载完bean的信息后，整个步骤已经跟前文很像了。

加载beanB类

public class Main {
	private static Map<String,Class> beanId2Clazz = new HashMap<String,Class>();
	public static void load() throws ClassNotFoundException {
	    // 即为上述的main类
	}
	public static Object getBean(String beanId) throws Exception{
		Class clazz = beanId2Clazz.get(beanId);
		if(null == clazz){
			return null;
		}
		Object obj = clazz.newInstance();
		Field[] fields = clazz.getDeclaredFields();
		// 处理带value注解的属性
		for(Field field : fields){
			Value v = field.getAnnotation(Value.class);
			if(null == v){
				continue;
			}
			String propertyValue = v.value();
			Method m = clazz.getMethod("set" + bigCaseFirstChar(field.getName()), String.class);
			m.invoke(obj, propertyValue);
		}
		return obj;
	}
	public static void main(String[] args) throws Exception {
		load();
		BeanB beanB = (BeanB)getBean("beanB");
		System.out.println(beanB.getTitle());
	}
}

加载beanA类

public class Main2 {
	private static Map<String,Class> beanId2Clazz = new HashMap<String,Class>();
	public static void load() throws ClassNotFoundException {
         // 不再赘述
	}
	public static Object getBean(String beanId) throws Exception{
		Class clazz = beanId2Clazz.get(beanId);
		if(null == clazz){
			return null;
		}
		Object obj = clazz.newInstance();
		Field[] fields = clazz.getDeclaredFields();
		for(Field field : fields){
			Method m = clazz.getMethod("set" + bigCaseFirstChar(field.getName()), field.getType());
			Value v = field.getAnnotation(Value.class);
			// 处理value注解
			if(null != v){
				String propertyValue = v.value();
				m.invoke(obj, propertyValue);
			}
			Resource r = field.getAnnotation(Resource.class);
			// 处理resource注解
			if(null != r){
				String propertyBeanId = r.name();
				if(StringUtils.isEmpty(propertyBeanId)){
					propertyBeanId = smallCaseFirstChar(field.getType().getSimpleName());
				}
				// 递归处理
				Object propertyObj = getBean(propertyBeanId);
				m.invoke(obj, propertyObj);
			}
		}
		return obj;
	}
	public static void main(String[] args) throws Exception {
		load();
		BeanA beanA = (BeanA)getBean("beanA");
		System.out.println(beanA.getBeanB().getTitle());
	}
}

至此，我们已经完全使用注解方式创建了一个bean工厂，下文将会尝试把注解和配置文件两种方式整合到一块儿。