Merge pull request #6 from etemplar/autoincrementpk

Autoincrement field addition.
jrvansuita · Jun 7, 2017 · 7f84ccd · 7f84ccd
2 parents 984c849 + 3373da8
commit 7f84ccd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/sqliteparser/src/main/java/com/vansuita/sqliteparser/Create.java b/sqliteparser/src/main/java/com/vansuita/sqliteparser/Create.java
@@ -48,6 +48,17 @@ public Create pk(String fieldName) {
         return this;
     }
 
+    /**
+     * Sets the primary key autoincrement field name.
+     *
+     * @param fieldName The name of the field
+     * @return the same instance of {@link Create} class.
+     */
+    public Create pkai(String fieldName) {
+        fields.put(fieldName, AUTOINCREMENT);
+        return this;
+    }
+
     /**
      * Sets a String field name.
      *

diff --git a/sqliteparser/src/main/java/com/vansuita/sqliteparser/SqlParser.java b/sqliteparser/src/main/java/com/vansuita/sqliteparser/SqlParser.java
@@ -64,6 +64,7 @@ public String build() {
     protected static final String CREATE_TABLE = "CREATE TABLE %s (%s)";
     protected static final String INTEGER = "INTEGER";
     protected static final String PK = INTEGER + " PRIMARY KEY";
+    protected static final String AUTOINCREMENT = PK + " AUTOINCREMENT";
     protected static final String TEXT = "TEXT";
     protected static final String FLOAT = "FLOAT";