Mainクラス

DIコンテナを作成して、DIコンテナからコンポーネントを取得して利用します。

public static void main(String[] args) throws Exception {
		
		DI di = new DI();
		di.init("vol");
		
		// 対象のディレクトリにあるクラスを取得する
		Product p = di.get(Product.class);
		p.use();
		
		//　サブディレクトリに存在するクラスも取得する
		User user = di.get(User.class);
		user.login();
		
		UserDao userDao = di.get(UserDao.class);
		userDao.update();
		
		UserController controller = di.get(UserController.class);
		controller.getUser();
	}

クラス一覧を取得する

特定のアノテーションが付いているクラスを取得する前に、クラスの一覧を取得します。“packageName.Class”のようなパッケージ名とクラス名さえあれば、リフレクションを利用してクラスの作成ができるので、文字列のリストでクラス一覧を作成します。

詳細については参考サイトを参照してください。

public class Finder {
	
	public static List<String> getClassList(String packageName) throws IOException{
	    final char PACKAGE_SEP = '.';
	    final char PACKAGE_RESOURCE_SEP = '/';
	    
	    String packageResource 
	        = packageName.replace(PACKAGE_SEP, PACKAGE_RESOURCE_SEP);
	    ClassLoader loader = ClassLoader.getSystemClassLoader();
	    URL resource = loader.getResource(packageResource);
	    File files = new File(resource.getFile());
	    
	    List<String> classNames = new ArrayList<>();
	    for(String file : files.list()) {
	        if (file.matches(".*\\.class$")) {
	        	String className = packageName + PACKAGE_SEP + file.replaceAll("\\.class$","");
	            classNames.add(className);
	        } else {
	        	// パッケージの場合再帰処理する
	        	String subPathName = packageName + "." + file.toString();
	        	classNames.addAll(getClassList(subPathName));
	        }
	    }
	    return classNames;
	}
}

パッケージ配下にパッケージがある場合もあるので、サブパッケージの中身は再帰処理で取得します。

@Componentアノテーションが付いているクラスのみ異種コンテナに登録する

異種コンテナを作成します。ここでは、Finderクラスから取得した、クラス一覧情報のリストから@Componentアノテーションがついてるクラスを登録します。

public class DI {
	
	// 異種コンテナ
	private Map<Class<?>, Object> container = new HashMap<>();
	
	public <T> T get(Class<T> type) {
		return type.cast(container.get(type));
	}
	
	@SuppressWarnings("unused")
	private <T> void put(Class<T> type, T instance) {
		container.put(Objects.requireNonNull(type), type.cast(instance));
	}
	
	/**
	 * DIコンテナを初期化する
	 * @Componentアノテーションがついているクラスを全てDIコンテナに登録します。
	 * 
	 * @param packageName
	 * @throws Exception
	 */
	public void init(String packageName) throws Exception {
		List<String> classNameList = Finder.getClassList(packageName);
		for(String name : classNameList) {
			//　クラス文字列からクラスオブジェクトを取得する
			Class<?> clazz = Class.forName(name);
			// アノテーション一覧を取得する
			Annotation[] anos = clazz.getAnnotations();
			// @Componentアノテーションがついているか判定する
			if(haveComponent(anos)) {
				// コンストラクタオブジェクトを取得する
				Constructor<?> constructor = clazz.getConstructor();
				// インスタンスからgetClass()をしないとクラス情報は取得できない
				// clazz.getName().getClass()ではjava.lang.classになってしまう
				Object obj = constructor.newInstance();
				container.put(obj.getClass(), obj);
			}
		}
	}
	
	// Componentアノテーションが付いてるか確認する
	private boolean haveComponent(Annotation[] annotations) {
		for(Annotation annotation : annotations) {
			if(annotation.annotationType() == Component.class) {
				return true;
			}
		}
		return false;
	}
	
}

テスト

作った機能が正しいかテストします。以下のテストは成功します。

@Test
	void test() throws Exception {
		DI di = new DI();
		di.init("vol");
		
		User user = di.get(User.class);
		assertTrue(user instanceof User);
		
		UserDao dao = di.get(UserDao.class);
		assertTrue(dao instanceof UserDao);
		
		// アノテーションがついてないクラスはインスタンス化されない
		Sample sample = di.get(Sample.class);
		assertNull(sample);
	}

アノテーションクラス

一応アノテーションについても載せます。特に何もないアノテーションです

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(value={CONSTRUCTOR, FIELD, LOCAL_VARIABLE, METHOD, PACKAGE, MODULE, PARAMETER, TYPE})
public @interface Component{

}

クラス階層は以下のようになっています。サブパッケージがあっても取得できています

参考

Java パッケージ配下のクラスを取得する

Java | アノテーションを利用して簡単なDIコンテナを作る