Springの小話 - @RestControllerをJavaConfigでBean登録したい！

皆さん@RestControllerをJavaConfigでBean登録したいと思ったことはありませんか？私は共通コンポーネント的なものを作ることが多いため、これはよく思ったりします。今回のSpringの小話はそんなお悩みに対するTipsの紹介となります。

皆さんもご存じのとおり@RestControllerは@Componentを含むメタアノテーションのため、@RestControllerのクラスがコンポーネントスキャンのパス上に存在する場合、いやおうなくBean登録されてしまいます。これがアプリのクラスであれば問題になることはないですが、共通コンポーネントとして作成する場合は問題となります。

共通コンポーネントとする場合、ライブラリに含めるがそれを使うか、つまりBeanとして登録するかはJavaConfigで利用者側で選択できるようにしたいことが多々ありますが、その一方で、SpringにRestControllerとして認識してもらうには@Componentを含む@RestControllerを付けるしかなく、次のようなJavaConfigによる登録はできません。

@Bean
@RestController // ← これはできない！
JavaConfigurableController controller() {
    return new JavaConfigurableController();
}

できないものはしょうがないと「Bean登録したくないときはアプリで@ComponentScanの指定で除外してください」というのもエレガントさに欠けるためできれば避けたいです。

そんな@Componentを内包したメタアノテーションを付けたクラスをコンポーネントスキャンのBean登録の対象にせず、どうやってJavaConfigでBean登録可能にするかですが、結論からいうとその方法は「Conditionで@Componentの作用を常に無効化する」です。

例えば次のようなRESTのControllerがあったとします。

@RestController
public class JavaConfigurableController {
    @GetMapping("/javaconfig")
    public String hello() {
        return "called javaconfig";
    }
}

このクラスがコンポーネントスキャンの対象に入っている場合、このままではBean登録されてしまうため、常にBean登録を無効化する次のConditionクラスを用意します。

public class AlwaysFalseCondition implements Condition {
    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        return false; // 常に条件に合致しない⇒Bean登録しない
    }
}
---
@Target({ ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Conditional(AlwaysFalseBootCondition.class)
public @interface SkipRegistration {
}

用意した無効化ConditionをControllerクラスに付けます。こうすることで常にコンポーネントスキャンによるBean登録は無効化されます。

@RestController
@SkipRegistration
public class JavaConfigurableController {
    @GetMapping("/javaconfig")
    public String hello() {
        return "called javaconfig";
    }
}

これで@RestControllerはついているがコンポーネントスキャンはされないBeanクラスができたので、あとはこのBeanクラスを使いたいときにJavaConfigで登録するだけです。

@Bean
JavaConfigurableController controller() {
    return new JavaConfigurableController();
}

public class AlwaysFalseBootCondition extends SpringBootCondition {
    @Override
    public ConditionOutcome getMatchOutcome(ConditionContext context, AnnotatedTypeMetadata metadata) {
        ConditionMessage message = ConditionMessage
                .forCondition("AlwaysFalseCondition")
                .because("Condition always returns false");
        return ConditionOutcome.noMatch(message);
    }
}

同様なことは他にも例えば@Profileの機能を使ってもできたりしますが、このためだけにプロファイルを増やすことには抵抗があります。ですので、@Profileを使うよりも今回の無効化作戦の方がエレガントだとは思っていたりします。ただ、その一方で強引に無効化する若干ハックまがいな手段であることは筆者自身も感じたりはしていますが、それなりに使っていても問題になることはないため、これが正解！と思うようになってきました。（あと他に妙案もないし・・）

とはいえ、本音はコンポーネントを意味する@ComponentとRESTのコントローラーであることを意味する@RestControllerは別々に定義できるようにしてくれればいいのにSpringさん、、と思うところではあります。