001package io.ebean.querybean.generator;
002
003
004import java.util.Set;
005
006/**
007 * Enum property type.
008 */
009public class PropertyTypeEnum extends PropertyType {
010
011  private final String enumClass;
012
013  private final String enumShortName;
014
015  PropertyTypeEnum(String enumClass, String enumShortName) {
016    super("PEnum");
017    this.enumClass = enumClass;
018    this.enumShortName = enumShortName;
019  }
020
021  @Override
022  public String getTypeDefn(String shortName, boolean assoc) {
023    if (assoc) {
024      return "PEnum<R," + enumShortName + ">";
025
026    } else {
027      return "PEnum<Q" + shortName + "," + enumShortName + ">";
028    }
029  }
030
031  @Override
032  public void addImports(Set<String> allImports) {
033    super.addImports(allImports);
034    allImports.add(enumClass);
035  }
036
037}