如何使用JavaPoet创建Spring Beans？课程作业技术问询

完全可以通过JavaPoet生成Spring Bean，不管是直接生成带@RestController/@Component这类 stereotype 注解的类，还是生成配置类通过@Bean方法注册Bean，都能实现动态生成并加载到Spring上下文的需求。

一、你的现有代码分析与优化 #

你当前的代码已经能生成带@RestController的类，但存在一个关键问题：第一次运行时，生成的Java文件写入到src/main/java目录，此时编译后的classpath中并没有这个类的字节码，Spring启动时无法扫描到。

优化点：调整生成路径到编译输出目录 #

将生成路径改为Maven/Gradle的编译输出目录（比如Maven的target/classes），这样生成的类会直接处于classpath下，Spring启动时就能扫描到：

// 替换原有的写入路径
javaFile.writeTo(Paths.get("target/classes/"));

二、动态编译与类加载（解决首次运行无字节码问题） #

如果需要在程序运行时动态生成、编译并加载类，无需提前编译，可以结合JavaCompiler API实现：

添加动态编译与类加载的方法：

private static void compileAndLoad(JavaFile javaFile) throws Exception {
    JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
    StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);

    // 创建临时目录存放生成的Java文件
    Path tempDir = Files.createTempDirectory("javapoet_temp");
    javaFile.writeTo(tempDir);

    // 获取待编译的Java文件
    File sourceFile = tempDir.resolve(
            javaFile.packageName.replace('.', '/') + "/" + javaFile.typeSpec.name + ".java"
    ).toFile();
    Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjects(sourceFile);

    // 指定编译输出目录为当前classpath
    List<String> options = Arrays.asList("-d", System.getProperty("java.class.path"));
    JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, null, options, null, compilationUnits);

    // 执行编译
    if (!task.call()) {
        throw new RuntimeException("编译生成类失败");
    }

    // 加载编译后的类到类加载器
    URLClassLoader classLoader = URLClassLoader.newInstance(new URL[]{tempDir.toUri().toURL()});
    Class.forName(javaFile.packageName + "." + javaFile.typeSpec.name, true, classLoader);
}

然后在generateRestController方法中，写完Java文件后调用该方法即可。

三、另一种方案：生成@Bean配置类 #

除了生成带注解的业务类，还可以用JavaPoet生成Spring配置类，通过@Bean方法手动注册Bean：

private static void generateConfigClass() throws IOException, ClassNotFoundException {
    // 生成@Bean方法
    MethodSpec helloWorldBeanMethod = MethodSpec.methodBuilder("helloWorldBean")
            .addModifiers(Modifier.PUBLIC)
            .addAnnotation(Bean.class)
            .returns(Class.forName("com.example.PhotoAppDiscoveryService.HelloWorld"))
            .addStatement("return new HelloWorld()")
            .build();

    // 生成配置类
    TypeSpec configClass = TypeSpec.classBuilder("HelloWorldConfig")
            .addModifiers(Modifier.PUBLIC)
            .addAnnotation(Configuration.class)
            .addMethod(helloWorldBeanMethod)
            .build();

    JavaFile javaFile = JavaFile.builder("com.example.PhotoAppDiscoveryService", configClass)
            .build();

    javaFile.writeTo(Paths.get("target/classes/"));
}

这样生成的配置类会被Spring的组件扫描机制发现，进而将HelloWorld实例注册为Bean。

内容的提问来源于stack exchange，提问作者Sherlock Holmes