spring事務(5)-----手寫SpringIOC容器框架之手寫@Service和@Resource註解
阿新 • • 發佈：2018-11-22
一，xml方式
註解方式就不寫了，主要就是dom4j的解析。
二，註解方式
2.1，我們先來看一下spring自帶的寫法
spring.xml
<beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p" xmlns:context="http://www.springframework.org/schema/context" xmlns:aop="http://www.springframework.org/schema/aop" xmlns:tx="http://www.springframework.org/schema/tx" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsdhttp://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx.xsd"> <!-- <bean id="userServiceImpl" class="com.qingruihappy2.utils.UserServiceImpl"></bean> --> <context:component-scan base-package="com.qingruihappy4" /> </beans>
package com.qingruihappy4.dao; import org.springframework.stereotype.Service; @Service public class OrderDao { public void addOrder() { System.out.println("我是OrderDao層，我在用@Service方法注入到容器中去。"); } }
package com.qingruihappy4.dao; import org.springframework.stereotype.Service; @Service public class UserDao { public void add() { System.out.println("我是UserDao層，我在用@Service方法注入到容器中去。"); } }
package com.qingruihappy4.service; public interface OrderService { public void addOrder(); }
package com.qingruihappy4.service.impl; import javax.annotation.Resource; import org.springframework.stereotype.Service; import com.qingruihappy4.dao.OrderDao; import com.qingruihappy4.service.OrderService; @Service public class OrderServiceImpl implements OrderService { // 從Spring容器中讀取bean @Resource private OrderDao orderDao; public void addOrder() { orderDao.addOrder(); System.out.println("我是OrderServiceImpl，我在通過@Resource把orderDao取出來"); } }
package com.qingruihappy4.service; public interface OrderService { public void addOrder(); }
package com.qingruihappy4.service.impl; import javax.annotation.Resource; import org.springframework.stereotype.Service; import com.qingruihappy4.dao.OrderDao; import com.qingruihappy4.service.OrderService; @Service public class OrderServiceImpl implements OrderService { // 從Spring容器中讀取bean @Resource private OrderDao orderDao; public void addOrder() { orderDao.addOrder(); System.out.println("我是OrderServiceImpl，我在通過@Resource把orderDao取出來"); } }
package com.qingruihappy4; import org.springframework.context.support.ClassPathXmlApplicationContext; import com.qingruihappy4.service.OrderService; import com.qingruihappy4.service.UserService; public class Test001 { public static void main(String[] args) throws Exception { ClassPathXmlApplicationContext app = new ClassPathXmlApplicationContext("spring.xml"); //假如我們現在放到mvc的模式中，我們會發現啟動的時候就會去載入spring的xml檔案，去掃包。注意掃包的時候並沒有把通過@Service往容器中放值 //啟動的時候就會通過另一套機制載入發現有@Service註解，面向切面的思想，就會把它放到容器中去。 //當開始訪問的時候就會通過@Resource註解，面向切面的思想就會從容器中去取對應的例項，沒有就會報空指標異常 UserService userService = (UserService) app.getBean("userServiceImpl"); userService.add(); OrderService orderService = (OrderService) app.getBean("orderServiceImpl"); orderService.addOrder(); } }
2.2,手寫
自定義註解
package com.qingruihappy3.annotation; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; //自定義註解 從Spring容器獲取bean @Target({ ElementType.FIELD }) @Retention(RetentionPolicy.RUNTIME) public @interface ExtResource { }
package com.qingruihappy3.annotation; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; // 自定義註解 注入到Spring容器 @Target({ ElementType.TYPE }) @Retention(RetentionPolicy.RUNTIME) public @interface ExtService { }
掃包工具類這個不是重點，知道它能獲取到規定package下面所有的class檔案的類的例項就行了
package com.qingruihappy3.utils; 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.util.ArrayList; import java.util.Enumeration; import java.util.List; import java.util.jar.JarEntry; import java.util.jar.JarFile; public class ClassUtil { /** * 取得某個介面下所有實現這個介面的類 */ 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(); } } } } }
dao層
package com.qingruihappy3.dao; import com.qingruihappy3.annotation.ExtService; @ExtService public class OrderDao { public void addOrder() { System.out.println("我是OrderDao層，我在用@ExtService方法注入到容器中去。"); } }
package com.qingruihappy3.dao; import com.qingruihappy3.annotation.ExtService; @ExtService public class UserDao { public void add() { System.out.println("我是UserDao層，我在用@ExtService方法注入到容器中去。"); } }
service介面層
package com.qingruihappy3.service; public interface OrderService { public void addOrder(); }
package com.qingruihappy3.service; public interface UserService { public void add(); }
service實現層
package com.qingruihappy3.service.impl; import com.qingruihappy3.annotation.ExtResource; import com.qingruihappy3.annotation.ExtService; import com.qingruihappy3.dao.OrderDao; import com.qingruihappy3.service.OrderService; @ExtService public class OrderServiceImpl implements OrderService { // 從Spring容器中讀取bean @ExtResource private OrderDao orderDao; public void addOrder() { orderDao.addOrder(); System.out.println("我是OrderServiceImpl，我在通過@ExtResource把orderDao取出來"); } }
package com.qingruihappy3.service.impl; import com.qingruihappy3.annotation.ExtResource; import com.qingruihappy3.annotation.ExtService; import com.qingruihappy3.dao.UserDao; import com.qingruihappy3.service.OrderService; import com.qingruihappy3.service.UserService; //將該類注入到spring容器裡面 @ExtService public class UserServiceImpl implements UserService { // 從Spring容器中讀取bean @ExtResource private UserDao userDao; public void add() { userDao.add(); System.out.println("我是UserServiceImpl，我在通過@ExtResource把UserDao取出來"); } }
重點來了
package com.qingruihappy3.spring; import java.lang.reflect.Field; import java.util.ArrayList; import java.util.List; import java.util.concurrent.ConcurrentHashMap; import org.apache.commons.lang.StringUtils; import com.qingruihappy3.annotation.ExtResource; import com.qingruihappy3.annotation.ExtService; import com.qingruihappy3.utils.ClassUtil; /** * 手寫Spring專題 註解版本注入bean * * @author 黃慶瑞 * */ @SuppressWarnings({ "rawtypes", "unchecked" }) public class ClassPathXmlApplicationContext { // 掃包範圍 private String packageName; public ClassPathXmlApplicationContext(String packageName) { this.packageName = packageName; } // 把帶有@service註解的物件放到map中去 public ConcurrentHashMap<String, Object> setMap() throws Exception { // 1.使用反射機制獲取該包下所有的類已經存在bean的註解類,放到一個list集合中 List<Class> listClassesAnnotation = findClassExisService(); if (listClassesAnnotation == null || listClassesAnnotation.isEmpty()) { throw new Exception("沒有需要初始化的bean"); } // 2.使用Java反射機制初始化物件，在放到一個map集合中，map的可以就是物件名稱的首字母小寫，value就是加註解的物件 ConcurrentHashMap<String, Object> initBean = initBean(listClassesAnnotation); if (initBean == null || initBean.isEmpty()) { throw new Exception("初始化bean為空!"); } return initBean; } // 使用beanID查詢物件 public Object getBean(ConcurrentHashMap<String, Object> initBean,String beanId) throws Exception { // 3.使用beanID查詢查詢對應bean物件 Object object = initBean.get(beanId); // 4.使用反射讀取類的屬性,賦值資訊,這裡是有問題的。 attriAssign(initBean,object); return object; } // 使用反射讀取類的屬性,賦值資訊 public void attriAssign(ConcurrentHashMap<String, Object> initBean,Object object) throws Exception { // 1.獲取類的屬性是否存在 獲取bean註解 Class<? extends Object> classInfo = object.getClass(); Field[] declaredFields = classInfo.getDeclaredFields(); for (Field field : declaredFields) { ExtResource annotation = field.getAnnotation(ExtResource.class); if (annotation != null) { // 屬性名稱 String name = field.getName(); // 2.使用屬性名稱查詢bean容器賦值 Object bean = initBean.get(name); if (bean != null) { // 私有訪問允許訪問 field.setAccessible(true); // 給屬性賦值，給object（userserviceimpl）的屬性field（userdao）賦值bean（從容器中拿到到userdao） field.set(object, bean); continue; } } else { throw new Exception("ExtResource沒有注入到" + field.getName() + "方法上去"); } } } // 使用反射機制獲取該包下所有的類已經存在bean的註解類 public List<Class> findClassExisService() throws Exception { // 1.使用反射機制獲取該包下所有的類 if (StringUtils.isEmpty(packageName)) { throw new Exception("掃包地址不能為空!"); } // 2.使用反射技術獲取當前包下所有的類 List<Class<?>> classesByPackageName = ClassUtil.getClasses(packageName); // 3.存放類上有bean注入註解 List<Class> exisClassesAnnotation = new ArrayList<Class>(); // 4.判斷該類上屬否存在註解 for (Class classInfo : classesByPackageName) { // 確定就是這個註解 ExtService extService = (ExtService) classInfo.getAnnotation(ExtService.class); if (extService != null) { exisClassesAnnotation.add(classInfo); continue; } } return exisClassesAnnotation; } // 初始化bean物件 public ConcurrentHashMap<String, Object> initBean(List<Class> listClassesAnnotation) throws InstantiationException, IllegalAccessException { ConcurrentHashMap concurrentHashMap = new ConcurrentHashMap<String, Object>(); for (Class classInfo : listClassesAnnotation) { // 初始化物件 Object newInstance = classInfo.newInstance(); // 轉成小寫 String beanId = toLowerCaseFirstOne(classInfo.getSimpleName()); concurrentHashMap.put(beanId, newInstance); } return concurrentHashMap; } // 首字母轉小寫 public static String toLowerCaseFirstOne(String s) { if (Character.isLowerCase(s.charAt(0))) return s; else return (new StringBuilder()).append(Character.toLowerCase(s.charAt(0))).append(s.substring(1)).toString(); } }
測試（裡面有思路）
package com.qingruihappy3; import com.qingruihappy3.service.UserService; import com.qingruihappy3.spring.ClassPathXmlApplicationContext; import java.util.concurrent.ConcurrentHashMap; import com.qingruihappy3.service.OrderService; public class Test001 { public static void main(String[] args) throws Exception { //1,掃包 ClassPathXmlApplicationContext app = new ClassPathXmlApplicationContext("com.qingruihappy3"); //2,把有service註解的物件放到map容器中來，（其實這一步可以一併放在步驟3中實現的，但是為了好理解一下，單獨拆出來了） ConcurrentHashMap<String, Object> map = app.setMap(); //3,從容器中取出值來（注意這個時候不僅要取userServiceImpl物件，還要取userdao物件的），並且給它的帶有resource註解的屬性值（userdao）賦值 UserService userService = (UserService) app.getBean(map,"userServiceImpl"); userService.add(); OrderService orderService = (OrderService) app.getBean(map,"orderServiceImpl"); orderService.addOrder(); } }