Support for @Identity - use identity options with sequence

src/main/java/io/ebeaninternal/dbmigration/ddlgeneration/platform/BaseTableDdl.java

@@ -248,18 +248,19 @@ public class BaseTableDdl implements TableDdl {
     List<Column> columns = createTable.getColumn();
     List<Column> pk = determinePrimaryKeyColumns(columns);
 
-    DdlIdentity idMode = DdlIdentity.NONE;
+    DdlIdentity identity = DdlIdentity.NONE;
     if ((pk.size() == 1)) {
       final IdentityMode identityMode = MTableIdentity.fromCreateTable(createTable);
       IdType idType = platformDdl.useIdentityType(identityMode.getIdType());
-      idMode = new DdlIdentity(idType, identityMode);
+      String sequenceName = (IdType.SEQUENCE != idType) ? null : sequenceName(createTable, pk, identity);
+      identity = new DdlIdentity(idType, identityMode, sequenceName);
     }
 
     String partitionMode = createTable.getPartitionMode();
 
     DdlBuffer apply = writer.apply();
     apply.append(platformDdl.getCreateTableCommandPrefix()).append(" ").append(tableName).append(" (");
-    writeTableColumns(apply, columns, idMode);
+    writeTableColumns(apply, columns, identity);
     writeUniqueConstraints(apply, createTable);
     writeCompoundUniqueConstraints(apply, createTable);
     if (!pk.isEmpty()) {
@@ -294,9 +295,8 @@ public class BaseTableDdl implements TableDdl {
     // we drop the related sequence (if sequences are used)
     dropTable(writer.dropAll(), tableName);
 
-    if (idMode.useSequence()) {
-      String pkCol = pk.get(0).getName();
-      writeSequence(writer, createTable, pkCol);
+    if (identity.useSequence()) {
+      writeSequence(writer, identity);
     }
 
     // add blank line for a bit of whitespace between tables
@@ -308,6 +308,14 @@ public class BaseTableDdl implements TableDdl {
     }
   }
 
+  private String sequenceName(CreateTable createTable, List<Column> pk, DdlIdentity identity) {
+    String seqName = identity.getSequenceName();
+    if (seqName == null || seqName.isEmpty()) {
+      seqName = namingConvention.getSequenceName(createTable.getName(), pk.get(0).getName());
+    }
+    return seqName;
+  }
+
   /**
    * Add table and column comments (separate from the create table statement).
    */
@@ -377,18 +385,9 @@ public class BaseTableDdl implements TableDdl {
     }
   }
 
-  protected void writeSequence(DdlWrite writer, CreateTable createTable, String pk) throws IOException {
-    // explicit sequence use or platform decides
-    String explicitSequenceName = createTable.getSequenceName();
-    int initial = toInt(createTable.getSequenceInitial());
-    int allocate = toInt(createTable.getSequenceAllocate());
-
-    String seqName = explicitSequenceName;
-    if (seqName == null || seqName.isEmpty()) {
-      seqName = namingConvention.getSequenceName(createTable.getName(), pk);
-    }
-
-    String createSeq = platformDdl.createSequence(seqName, initial, allocate);
+  protected void writeSequence(DdlWrite writer, DdlIdentity identity) throws IOException {
+    String seqName = identity.getSequenceName();
+    String createSeq = platformDdl.createSequence(seqName, identity);
     if (hasValue(createSeq)) {
       writer.apply().append(createSeq).newLine();
       writer.dropAll().appendStatement(platformDdl.dropSequence(seqName));

src/main/java/io/ebeaninternal/dbmigration/ddlgeneration/platform/DdlIdentity.java

@@ -10,15 +10,18 @@ public class DdlIdentity {
 
   private final IdType idType;
   private final IdentityMode identityMode;
+  private final String sequenceName;
 
-  public DdlIdentity(IdType idType, IdentityMode identityMode) {
+  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() {
@@ -29,6 +32,10 @@ public class DdlIdentity {
     return idType == IdType.IDENTITY;
   }
 
+  public String getSequenceName() {
+    return sequenceName;
+  }
+
   private String generatedBy() {
     return isAlways() ? "always" : "by default";
   }
@@ -41,16 +48,28 @@ public class DdlIdentity {
     return " generated " + generatedBy() +" as identity";
   }
 
-  public String options(String startWith, String incrementBy, String cache) {
+  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);
-    sb.append(" (");
+    if (brackets) {
+      sb.append(" (");
+    }
     optionFor(sb, startWith, identityMode.getStart());
     optionFor(sb, incrementBy, identityMode.getIncrement());
     optionFor(sb, cache, identityMode.getCache());
-    sb.append(")");
+    if (brackets) {
+      sb.append(")");
+    }
     return sb.toString();
   }
 
@@ -63,4 +82,15 @@ public class DdlIdentity {
     }
   }
 
+  public int getStart() {
+    return identityMode.getStart();
+  }
+
+  public int getIncrement() {
+    return identityMode.getIncrement();
+  }
+
+  public int getCache() {
+    return identityMode.getCache();
+  }
 }

src/main/java/io/ebeaninternal/dbmigration/ddlgeneration/platform/PlatformDdl.java

@@ -68,6 +68,9 @@ public class PlatformDdl {
   protected String identityStartWith = "start with";
   protected String identityIncrementBy = "increment by";
   protected String identityCache = "cache";
+  protected String sequenceStartWith = "start with";
+  protected String sequenceIncrementBy = "increment by";
+  protected String sequenceCache = "cache";
 
   protected String alterTableIfExists = "";
 
@@ -195,7 +198,7 @@ public class PlatformDdl {
 
     StringBuilder sb = new StringBuilder(columnDefn.length() + 60);
     sb.append(columnDefn).append(identity.optionGenerated());
-    sb.append(identity.options(identityStartWith, identityIncrementBy, identityCache));
+    sb.append(identity.identityOptions(identityStartWith, identityIncrementBy, identityCache));
     return sb.toString();
   }
 
@@ -362,16 +365,10 @@ public class PlatformDdl {
   /**
    * Generate and return the create sequence DDL.
    */
-  public String createSequence(String sequenceName, int initialValue, int allocationSize) {
-
+  public String createSequence(String sequenceName, DdlIdentity identity) {
     StringBuilder sb = new StringBuilder("create sequence ");
     sb.append(sequenceName);
-    if (initialValue > 1) {
-      sb.append(" start with ").append(initialValue);
-    }
-    if (allocationSize > 1) {
-      sb.append(" increment by ").append(allocationSize);
-    }
+    identity.sequenceOptions(sequenceStartWith, sequenceIncrementBy, sequenceCache);
     sb.append(";");
     return sb.toString();
   }

src/main/java/io/ebeaninternal/dbmigration/ddlgeneration/platform/SqlServerDdl.java

@@ -114,18 +114,22 @@ public class SqlServerDdl extends PlatformDdl {
    * Generate and return the create sequence DDL.
    */
   @Override
-  public String createSequence(String sequenceName, int initialValue, int allocationSize) {
-
-    StringBuilder sb = new StringBuilder("create sequence ");
-    sb.append(sequenceName);
-    sb.append(" as bigint ");
-    if (initialValue > 1) {
-      sb.append(" start with ").append(initialValue);
+  public String createSequence(String sequenceName, DdlIdentity identity) {
+    StringBuilder sb = new StringBuilder(80);
+    sb.append("create sequence ").append(sequenceName).append(" as bigint");
+    final int start = identity.getStart();
+    if (start > 1) {
+      sb.append(" start with ").append(start);
     } else {
-      sb.append(" start with 1 ");
+      sb.append(" start with 1");
     }
-    if (allocationSize > 1) {
-      sb.append(" increment by ").append(allocationSize);
+    final int increment = identity.getIncrement();
+    if (increment > 1) {
+      sb.append(" increment by ").append(increment);
+    }
+    final int cache = identity.getCache();
+    if (cache > 1) {
+      sb.append(" cache ").append(increment);
     }
     sb.append(";");
     return sb.toString();