feat: DI Stage3 통과

woowacourse · Sep 26, 2023 · 4f9b7ab · 4f9b7ab
1 parent a5ce3bd
commit 4f9b7ab
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 3 deletions.
diff --git a/study/src/test/java/di/stage3/context/DIContainer.java b/study/src/test/java/di/stage3/context/DIContainer.java
@@ -1,6 +1,10 @@
 package di.stage3.context;
 
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Field;
+import java.lang.reflect.InvocationTargetException;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * 스프링의 BeanFactory, ApplicationContext에 해당되는 클래스
@@ -10,11 +14,45 @@ class DIContainer {
     private final Set<Object> beans;
 
     public DIContainer(final Set<Class<?>> classes) {
-        this.beans = Set.of();
+        this.beans = classes.stream()
+                .map(this::createInstance)
+                .collect(Collectors.toSet());
+        for (Object bean : this.beans) {
+            Field[] fields = bean.getClass().getDeclaredFields();
+            for (Field field : fields) {
+                Object object = getBean(field.getType());
+                if (object == null) {
+                    return;
+                }
+                try {
+                    field.setAccessible(true);
+                    field.set(bean, object);
+                    field.setAccessible(false);
+                } catch (IllegalAccessException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+        }
+    }
+
+    private Object createInstance(Class<?> aClass) {
+        try {
+            Constructor<?> constructor = aClass.getDeclaredConstructor();
+            constructor.setAccessible(true);
+            Object instance = constructor.newInstance();
+            constructor.setAccessible(false);
+            return instance;
+        } catch (NoSuchMethodException | InstantiationException | IllegalAccessException |
+                 InvocationTargetException e) {
+            throw new RuntimeException(e);
+        }
     }
 
     @SuppressWarnings("unchecked")
     public <T> T getBean(final Class<T> aClass) {
-        return null;
+        return (T) beans.stream()
+                .filter(aClass::isInstance)
+                .findAny()
+                .orElse(null);
     }
 }
diff --git a/study/src/test/java/di/stage3/context/Stage3Test.java b/study/src/test/java/di/stage3/context/Stage3Test.java
@@ -19,7 +19,7 @@ class Stage3Test {
     void stage3() {
         final var user = new User(1L, "gugu");
 
-        final var diContainer = createDIContainer();
+        DIContainer diContainer = createDIContainer();
 
         /**
          * 제어의 역전(IoC)