Генерация аннотаций с использованием JavaPoet

Я пишу генератор кода с использованием JavaPoet и мне нужно добавить аннотацию к классу

Например:

package some.package

import org.hibernate.annotations.CacheConcurrencyStrategy;
import javax.persistence.Entity;
import javax.persistence.Cache

@Entity
@Cache(usage = CacheConcurrencyStrategy.NONSTRICT_READ_WRITE)
public class SomeClass {
}

Мой код выглядит так:

TypeSpec spec = TypeSpec
  .classBuilder("SomeClass")
  .addAnnotation(Entity.class)
  .addAnnotation(AnnotationSpec.builder(Cache.class)
     .addMember("usage", "$L", CacheConcurrencyStrategy.NONSTRICT_READ_WRITE)
     .build())
  .build()

Этот код генерирует класс, но в результирующем коде отсутствует оператор импорта для CacheConcurrencyStrategy. Как я могу сгенерировать код для вывода всего необходимого кода?

1 ответ

Решение

Попробуй это:

TypeSpec spec = TypeSpec
  .classBuilder("SomeClass")
  .addAnnotation(Entity.class)
  .addAnnotation(AnnotationSpec.builder(Cache.class)
      .addMember("usage", "$T.$L", CacheConcurrencyStrategy.class,
          CacheConcurrencyStrategy.NONSTRICT_READ_WRITE.name())
      .build())
  .build()

$T идентифицирует класс перечисления и $L константа перечисления.

Другие вопросы по тегам