将数组作为javapoet中注释参数的值传递

Question

使用javapoet我正在尝试注释一个带有 array array 的注释的班级作为参数值，即

@MyCustom(param = { Bar.class, Another.class })
class Foo { 
}

我使用AnnotationSpec.builder及其addMember()方法：

List<TypeMirror> moduleTypes = new ArrayList<>(map.keySet());
AnnotationSpec annotationSpec = AnnotationSpec.builder(MyCustom.class)
   .addMember("param", "{ $T[] } ", moduleTypes.toArray() )
   .build();
builder.addAnnotation(annotationSpec);

Answer 1

codeBlock具有一个联接收集器，您可以使用它来流式传输，执行以下操作（例如，这是枚举）。您可以为任何类型做到这一点，只有地图会更改。

AnnotationSpec.builder(MyCustom.class)
   .addMember(
      "param",
      "$L",
      moduleTypes.stream()
         .map(type -> CodeBlock.of("$T.$L", MyCustom.class, type))
         .collect(CodeBlock.joining(",", "{", "}")))
   .build()

Answer 2

也许不是最佳解决方案，但是可以通过以下方式完成javapoet的注释的数组：

List<TypeMirror> moduleTypes = new ArrayList<>(map.keySet());
CodeBlock.Builder codeBuilder  = CodeBlock.builder();
boolean arrayStart = true;
codeBuilder.add("{ ");
for (TypeMirror modType: moduleTypes)
    if (!arrayStart)
        codeBuilder.add(" , ");
    arrayStart = false;
    codeBuilder.add("$T.class", modType);
codeBuilder.add(" }");
AnnotationSpec annotationSpec = AnnotationSpec.builder(MyCustom.class)
    .addMember("param", codeBuilder.build() )
    .build();
builder.addAnnotation(annotationSpec);