使用JavaPoet生成注釋

Question

我正在使用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的import語句。 如何生成代碼以便輸出所有必需的代碼？

Answer 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標識枚舉常量。