add bean Ioc support

monsoon · monsoon · commit 8a41e5370461 · 2017-04-18T12:31:51.000+08:00
diff --git a/src/main/java/simple/xfj/framework/helper/BeanHelper.java b/src/main/java/simple/xfj/framework/helper/BeanHelper.java
@@ -0,0 +1,39 @@
+package simple.xfj.framework.helper;
+
+import simple.xfj.framework.util.ReflectionUtil;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * Created by asus on 2017/4/18.
+ */
+public class BeanHelper {
+
+    private static final Map<Class<?>,Object> beanMap = new HashMap<Class<?>, Object>();
+
+    static {
+        Set<Class<?>> beanSet = ClassHelper.getBeanSet();
+        if(null != beanSet && beanSet.size()>0){
+            for(Class clazz : beanSet){
+                Object obj = ReflectionUtil.newInstance(clazz);
+                beanMap.put(clazz,obj);
+            }
+        }
+    }
+
+    public static Map<Class<?>,Object> getBeanMap(){
+        return beanMap;
+    }
+
+    public static <T> T getClassBean(Class<T> clazz){
+        if(null != clazz){
+            if(!beanMap.containsKey(clazz)){
+                throw new RuntimeException("can not get bean by class:"+clazz.getName());
+            }
+            return (T)beanMap.get(clazz);
+        }
+        return null;
+    }
+}
diff --git a/src/main/java/simple/xfj/framework/helper/IocHelper.java b/src/main/java/simple/xfj/framework/helper/IocHelper.java
@@ -0,0 +1,38 @@
+package simple.xfj.framework.helper;
+
+/**
+ * Created by asus on 2017/4/18.
+ */
+
+import simple.xfj.framework.annotation.Autowired;
+import simple.xfj.framework.util.ReflectionUtil;
+
+import java.lang.reflect.Field;
+import java.util.Map;
+
+/**
+ * 将带有@Autowired的filed字段通过反射注入
+ */
+public class IocHelper {
+    static {
+        Map<Class<?>, Object> beanMap = BeanHelper.getBeanMap();
+        if(null != beanMap && beanMap.size() > 0){
+            for(Map.Entry entry: beanMap.entrySet()){
+                Class clazz = (Class) entry.getKey();
+                Object instance = entry.getValue();
+                Field[] fields = clazz.getDeclaredFields();
+                if(null != fields && fields.length > 0){
+                    for(Field field:fields){
+                        if(field.isAnnotationPresent(Autowired.class)){
+                            Class<?> clazzType = field.getType();
+                            Object beanInject = BeanHelper.getClassBean(clazzType);
+                            if(null != beanInject )
+                                //通过反射设置beanfield的值
+                                ReflectionUtil.setField(instance,field,beanInject);
+                        }
+                    }
+                }
+            }
+        }
+    }
+}
