JavaPoet:在同一文件中创建两个枚举

<dependency>
    <groupId>com.squareup</groupId>
    <artifactId>javapoet</artifactId>
    <version>1.13.0</version>
</dependency>

implementation 'com.squareup:javapoet:1.13.0'

import com.squareup.javapoet.*;

import javax.lang.model.element.Modifier;
import java.io.IOException;

public class EnumGenerator {

    public static void main(String[] args) {
        // 创建一个Java文件
        JavaFile javaFile = JavaFile.builder("com.example", generateEnums())
                .build();

        // 将Java文件写入磁盘
        try {
            javaFile.writeTo(System.out);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static TypeSpec generateEnums() {
        // 创建第一个枚举
        TypeSpec enum1 = TypeSpec.enumBuilder("Enum1")
                .addModifiers(Modifier.PUBLIC)
                .addEnumConstant("VALUE1")
                .addEnumConstant("VALUE2")
                .build();

        // 创建第二个枚举
        TypeSpec enum2 = TypeSpec.enumBuilder("Enum2")
                .addModifiers(Modifier.PUBLIC)
                .addEnumConstant("VALUE3")
                .addEnumConstant("VALUE4")
                .build();

        // 创建一个类，包含两个枚举
        TypeSpec classWithEnums = TypeSpec.classBuilder("ClassWithEnums")
                .addModifiers(Modifier.PUBLIC)
                .addType(enum1)
                .addType(enum2)
                .build();

        return classWithEnums;
    }
}