diff --git a/http-generator-helidon/src/main/java/io/avaje/http/generator/helidon/nima/ControllerWriter.java b/http-generator-helidon/src/main/java/io/avaje/http/generator/helidon/nima/ControllerWriter.java index 409f6842..09b161e8 100644 --- a/http-generator-helidon/src/main/java/io/avaje/http/generator/helidon/nima/ControllerWriter.java +++ b/http-generator-helidon/src/main/java/io/avaje/http/generator/helidon/nima/ControllerWriter.java @@ -118,6 +118,7 @@ private void writeRoutes(List methods) { private void writeClassStart() { writer.append(AT_GENERATED).eol(); writer.append(diAnnotation()).eol(); + writer.append("@io.avaje.inject.BeanTypes({%s$Route.class, HttpFeature.class})", shortName).eol(); writer.append("public final class %s$Route implements HttpFeature {", shortName).eol().eol(); var controllerName = "controller"; diff --git a/http-generator-jex/src/main/java/io/avaje/http/generator/jex/ControllerWriter.java b/http-generator-jex/src/main/java/io/avaje/http/generator/jex/ControllerWriter.java index 0138d8f9..5e9faec0 100644 --- a/http-generator-jex/src/main/java/io/avaje/http/generator/jex/ControllerWriter.java +++ b/http-generator-jex/src/main/java/io/avaje/http/generator/jex/ControllerWriter.java @@ -82,7 +82,8 @@ private void writeRouting(MethodReader method) { private void writeClassStart() { writer.append(AT_GENERATED).eol(); writer.append(diAnnotation()).eol(); - writer.append("public final class ").append(shortName).append("$Route implements Routing.HttpService {").eol().eol(); + writer.append("@io.avaje.inject.BeanTypes({%s$Route.class, Routing.HttpService.class})", shortName).eol(); + writer.append("public final class %s$Route implements Routing.HttpService {", shortName).eol().eol(); String controllerName = "controller"; String controllerType = shortName;