io.ebeaninternal.dbmigration.ddlgeneration.platform.DdlIdentity Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ebean-ddlgen Show documentation
DDL and DB Migration generation
The newest version!
package io.ebeaninternal.dbmigration.ddlgeneration.platform;

import io.ebean.annotation.IdentityGenerated;
import io.ebean.config.dbplatform.IdType;
import io.ebeaninternal.server.deploy.IdentityMode;

public class DdlIdentity {

  public static final DdlIdentity NONE = new DdlIdentity();

  private final IdType idType;
  private final IdentityMode identityMode;
  private final String sequenceName;

  public DdlIdentity(IdType idType, IdentityMode identityMode, String sequenceName) {
    this.idType = idType;
    this.identityMode = identityMode;
    this.sequenceName = sequenceName;
  }

  private DdlIdentity() {
    this.idType = null;
    this.identityMode = null;
    this.sequenceName = null;
  }

  public boolean useSequence() {
    return idType == IdType.SEQUENCE;
  }

  public boolean useIdentity() {
    return idType == IdType.IDENTITY;
  }

  public String getSequenceName() {
    return sequenceName;
  }

  private String generatedBy() {
    return isAlways() ? "always" : "by default";
  }

  private boolean isAlways() {
    return IdentityGenerated.ALWAYS == identityMode.getGenerated();
  }

  public String optionGenerated() {
    return " generated " + generatedBy() +" as identity";
  }

  public String identityOptions(String startWith, String incrementBy, String cache) {
    return options(true, startWith, incrementBy, cache);
  }

  public String sequenceOptions(String startWith, String incrementBy, String cache) {
    return options(false, startWith, incrementBy, cache);
  }

  private String options(boolean brackets, String startWith, String incrementBy, String cache) {
    if (!identityMode.hasOptions()) {
      return "";
    }
    StringBuilder sb = new StringBuilder(40);
    if (brackets) {
      sb.append(" (");
    } else {
      sb.append(" ");
    }
    optionFor(sb, startWith, identityMode.getStart());
    optionFor(sb, incrementBy, identityMode.getIncrement());
    optionFor(sb, cache, identityMode.getCache());
    if (brackets) {
      sb.append(")");
    }
    return sb.toString();
  }

  private void optionFor(StringBuilder sb, String prefix, int val) {
    if (val > 0 && prefix != null) {
      if (sb.length() > 2) {
        sb.append(" ");
      }
      sb.append(prefix).append(" ").append(val);
    }
  }

  public int getStart() {
    return identityMode.getStart();
  }

  public int getIncrement() {
    return identityMode.getIncrement();
  }

  public int getCache() {
    return identityMode.getCache();
  }
}