io.ebean.querybean.generator.KotlinLangAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-querybean-generator Show documentation
Generates Kotlin querybeans
There is a newer version: 15.8.0
Show newest version
package io.ebean.querybean.generator;

class KotlinLangAdapter implements LangAdapter {

  @Override
  public void alias(Append writer, String shortName, String fullName) {
    writer.append("  companion object {").eol();
    writer.append("    /**").eol();
    writer.append("     * shared 'Alias' instance used to provide").eol();
    writer.append("     * properties to select and fetch clauses").eol();
    writer.append("     */").eol();
    writer.append("    val _alias = Q").append(shortName).append("(true)").eol();
    writer.eol();
    writer.append("    /**").eol();
    writer.append("     * Return a query bean used to build a FetchGroup.").eol();
    writer.append("     */").eol();
    writer.append("    fun forFetchGroup(): Q%s {", shortName).eol();
    writer.append("      return Q%s(io.ebean.FetchGroup.queryFor(%s::class.java));", shortName, fullName).eol();
    writer.append("    }").eol();
    writer.append("  }").eol().eol();
  }

  @Override
  public void assocBeanConstructor(Append writer, String shortName) {
    writer.append("  constructor(name: String, root: R) : super(name, root)").eol();
    writer.eol();
    writer.append("  constructor(name: String, root: R, prefix: String) : super(name, root, prefix)").eol();
  }

  @Override
  public void fetch(Append writer, String origShortName) {
    writeAssocBeanFetch(writer, origShortName, "", "Eagerly fetch this association loading the specified properties.");
    writeAssocBeanFetch(writer, origShortName, "Query", "Eagerly fetch this association using a 'query join' loading the specified properties.");
    writeAssocBeanFetch(writer, origShortName, "Cache", "Eagerly fetch this association using L2 cache.");
    writeAssocBeanFetch(writer, origShortName, "Lazy", "Use lazy loading for this association loading the specified properties.");
  }

  private void writeAssocBeanFetch(Append writer, String origShortName, String fetchType, String comment) {
//    fun fetch(vararg properties: TQProperty): R {
//      return fetchProperties(*properties)
//    }
    writer.append("  /**").eol();
    writer.append("   * ").append(comment).eol();
    writer.append("   */").eol();
    writer.append("  fun fetch%s(vararg properties: TQProperty) : R {", fetchType, origShortName).eol();
    writer.append("    return fetch%sProperties(*properties)", fetchType).eol();
    writer.append("  }").eol();
    writer.eol();
  }

  @Override
  public void rootBeanConstructor(Append writer, String shortName, String dbName, String fullName) {
    String name = (dbName == null) ? "default" : dbName;
    writer.append("  /**").eol();
    writer.append("   * Construct using the %s Database.", name).eol();
    writer.append("   */").eol();
    if (dbName == null) {
      writer.append("  constructor() : super(%s::class.java)", fullName).eol().eol();
    } else {
      writer.append("  constructor() : super(%s::class.java, io.ebean.DB.byName(\"%s\"))", fullName, dbName).eol().eol();
    }

    writer.append("  /**").eol();
    writer.append("   * @deprecated migrate to query.usingTransaction()", name).eol();
    writer.append("   */").eol();
    writer.append("   @Deprecated(message=\"migrate to query.usingTransaction()\")").eol();
    if (dbName == null) {
      writer.append("  constructor(transaction: io.ebean.Transaction) : super(%s::class.java, transaction)", fullName).eol().eol();
    } else {
      writer.append("  constructor(transaction: io.ebean.Transaction) : super(%s::class.java, io.ebean.DB.byName(\"%s\"), transaction)", fullName, dbName).eol().eol();
    }

    writer.eol();
    writer.append("  /**").eol();
    writer.append("   * Construct with a given Database.").eol();
    writer.append("   */").eol();
    writer.append("  constructor(database: io.ebean.Database) : super(%s::class.java, database)", fullName).eol().eol();

    writer.append("  /**").eol();
    writer.append("   * Construct for Alias.").eol();
    writer.append("   */").eol();
    writer.append("  private constructor(dummy: Boolean) : super(dummy)").eol().eol();

    writer.append("  /**").eol();
    writer.append("   * Private constructor for FetchGroup building.").eol();
    writer.append("   */").eol();
    writer.append("  private constructor(fetchGroupQuery: io.ebean.Query<%s>) : super(fetchGroupQuery)", fullName).eol();

    writer.eol();
    writer.append("  /** Return a copy of the query. */").eol();
    writer.append("  override fun copy() : Q%s {", shortName).eol();
    writer.append("    return Q%s(query().copy())", shortName).eol();
    writer.append("  }").eol();
    writer.eol();
  }

  @Override
  public void fieldDefn(Append writer, String propertyName, String typeDefn)  {
    writer.append("  lateinit var %s: ", propertyName);
    if (typeDefn.endsWith(",Integer>")) {
      typeDefn = typeDefn.replace(",Integer>", ",Int>");
    }
    writer.append(typeDefn);
  }

}