Add support to Foreign Key actions "ON UPDATE" and "ON DELETE"

schematic-annotations/src/main/java/net/simonvt/schematic/annotation/ForeignKeyAction.java

@@ -0,0 +1,13 @@
+package net.simonvt.schematic.annotation;
+
+/**
+ * Action triggered with foreign key ON DELETE and ON UPDATE clauses.
+ */
+public enum ForeignKeyAction {
+  NONE,
+  NO_ACTION,
+  RESTRICT,
+  SET_NULL,
+  SET_DEFAULT,
+  CASCADE,
+}

schematic-annotations/src/main/java/net/simonvt/schematic/annotation/ForeignKeyConstraint.java

@@ -21,9 +21,8 @@
     /** Column names in referenced table */
     String[] referencedColumns();
 
-    /**
-     * Defines conflict resolution algorithm.
-     * By default {@link ConflictResolutionType#NONE} is used.
-     * */
-    ConflictResolutionType onConflict() default ConflictResolutionType.NONE;
+
+    ForeignKeyAction onDelete() default ForeignKeyAction.NONE;
+
+    ForeignKeyAction onUpdate() default ForeignKeyAction.NONE;
 }

schematic-annotations/src/main/java/net/simonvt/schematic/annotation/References.java

@@ -34,4 +34,8 @@
   String table();
 
   String column();
+
+  ForeignKeyAction onDelete() default ForeignKeyAction.NONE;
+
+  ForeignKeyAction onUpdate() default ForeignKeyAction.NONE;
 }

schematic-compiler/src/main/java/net/simonvt/schematic/compiler/TableWriter.java

@@ -22,12 +22,31 @@
 import com.squareup.javapoet.JavaFile;
 import com.squareup.javapoet.MethodSpec;
 import com.squareup.javapoet.TypeSpec;
+
+import net.simonvt.schematic.annotation.AutoIncrement;
+import net.simonvt.schematic.annotation.Check;
+import net.simonvt.schematic.annotation.ConflictResolutionType;
+import net.simonvt.schematic.annotation.Constraints;
+import net.simonvt.schematic.annotation.DataType;
+import net.simonvt.schematic.annotation.DefaultValue;
+import net.simonvt.schematic.annotation.ForeignKeyAction;
+import net.simonvt.schematic.annotation.ForeignKeyConstraint;
+import net.simonvt.schematic.annotation.IfNotExists;
+import net.simonvt.schematic.annotation.NotNull;
+import net.simonvt.schematic.annotation.PrimaryKey;
+import net.simonvt.schematic.annotation.PrimaryKeyConstraint;
+import net.simonvt.schematic.annotation.References;
+import net.simonvt.schematic.annotation.Table;
+import net.simonvt.schematic.annotation.Unique;
+import net.simonvt.schematic.annotation.UniqueConstraint;
+
 import java.io.IOException;
 import java.io.Writer;
 import java.lang.reflect.Type;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+
 import javax.annotation.processing.Filer;
 import javax.annotation.processing.ProcessingEnvironment;
 import javax.lang.model.element.Element;
@@ -38,21 +57,6 @@
 import javax.lang.model.type.TypeMirror;
 import javax.tools.Diagnostic.Kind;
 import javax.tools.JavaFileObject;
-import net.simonvt.schematic.annotation.AutoIncrement;
-import net.simonvt.schematic.annotation.Check;
-import net.simonvt.schematic.annotation.ConflictResolutionType;
-import net.simonvt.schematic.annotation.Constraints;
-import net.simonvt.schematic.annotation.DataType;
-import net.simonvt.schematic.annotation.DefaultValue;
-import net.simonvt.schematic.annotation.ForeignKeyConstraint;
-import net.simonvt.schematic.annotation.IfNotExists;
-import net.simonvt.schematic.annotation.NotNull;
-import net.simonvt.schematic.annotation.PrimaryKey;
-import net.simonvt.schematic.annotation.PrimaryKeyConstraint;
-import net.simonvt.schematic.annotation.References;
-import net.simonvt.schematic.annotation.Table;
-import net.simonvt.schematic.annotation.Unique;
-import net.simonvt.schematic.annotation.UniqueConstraint;
 
 public class TableWriter {
 
@@ -251,6 +255,14 @@ public void createTable(TypeSpec.Builder databaseBuilder)
             .append("(")
             .append(references.column())
             .append(")");
+        if(references.onUpdate() != ForeignKeyAction.NONE) {
+          query.append(" ").append("ON UPDATE").append(" ");
+          writeOnActions(query, references.onUpdate());
+        }
+        if(references.onDelete() != ForeignKeyAction.NONE) {
+          query.append(" ").append("ON DELETE").append(" ");
+          writeOnActions(query, references.onDelete());
+        }
       }
     }
 
@@ -310,7 +322,14 @@ private static void writeForeignKeyConstraint(StringBuilder query, ForeignKeyCon
       query.append(foreignKey.referencedColumns()[i]);
     }
     query.append(')');
-
+    if(foreignKey.onUpdate() != ForeignKeyAction.NONE) {
+      query.append(" ").append("ON UPDATE").append(" ");
+      writeOnActions(query, foreignKey.onUpdate());
+    }
+    if(foreignKey.onDelete() != ForeignKeyAction.NONE) {
+      query.append(" ").append("ON DELETE").append(" ");
+      writeOnActions(query, foreignKey.onDelete());
+    }
   }
 
   private static void writePrimaryOrUniqueConstraint(
@@ -380,6 +399,28 @@ private static void writeOnConflict(StringBuilder query,
     }
   }
 
+  private static void writeOnActions(StringBuilder query,
+      ForeignKeyAction foreignKeyAction) {
+    switch (foreignKeyAction) {
+      case RESTRICT:
+        query.append("RESTRICT");
+        break;
+      case SET_NULL:
+        query.append("SET NULL");
+        break;
+      case SET_DEFAULT:
+        query.append("SET DEFAULT");
+        break;
+      case CASCADE:
+        query.append("CASCADE");
+        break;
+      case NO_ACTION:
+      default:
+        query.append("NO ACTION");
+        break;
+    }
+  }
+
   public void createValuesBuilder(Filer filer, String outPackage) throws IOException {
     String name = Character.toUpperCase(this.name.charAt(0)) + this.name.substring(1);
     String valuesPackage = outPackage + ".values";

schematic-sample-kotlin/src/main/java/net/simonvt/schematic/sample/database/NoteColumns.kt

@@ -16,13 +16,9 @@
 
 package net.simonvt.schematic.sample.database
 
-import net.simonvt.schematic.annotation.AutoIncrement
-import net.simonvt.schematic.annotation.Check
-import net.simonvt.schematic.annotation.DataType
+import net.simonvt.schematic.annotation.*
 import net.simonvt.schematic.annotation.DataType.Type.INTEGER
 import net.simonvt.schematic.annotation.DataType.Type.TEXT
-import net.simonvt.schematic.annotation.PrimaryKey
-import net.simonvt.schematic.annotation.References
 import net.simonvt.schematic.sample.database.NotesDatabase.Tables
 
 interface NoteColumns {
@@ -32,7 +28,8 @@ interface NoteColumns {
 
     @DataType(INTEGER) @PrimaryKey @AutoIncrement const val ID = "_id"
 
-    @DataType(INTEGER) @References(table = Tables.LISTS, column = ListColumns.ID)
+    @DataType(INTEGER) @References(table = Tables.LISTS, column = ListColumns.ID,
+            onDelete = ForeignKeyAction.CASCADE)
     const val LIST_ID = "listId"
 
     @DataType(TEXT) const val NOTE = "note"

schematic-sample-kotlin/src/main/java/net/simonvt/schematic/sample/database/TagColumns.kt

@@ -1,14 +1,8 @@
 package net.simonvt.schematic.sample.database
 
-import net.simonvt.schematic.annotation.AutoIncrement
+import net.simonvt.schematic.annotation.*
 import net.simonvt.schematic.annotation.ConflictResolutionType.REPLACE
-import net.simonvt.schematic.annotation.Constraints
-import net.simonvt.schematic.annotation.DataType
 import net.simonvt.schematic.annotation.DataType.Type
-import net.simonvt.schematic.annotation.NotNull
-import net.simonvt.schematic.annotation.PrimaryKey
-import net.simonvt.schematic.annotation.References
-import net.simonvt.schematic.annotation.UniqueConstraint
 
 @Constraints(
     unique = arrayOf(
@@ -24,7 +18,8 @@ interface TagColumns {
     @DataType(Type.INTEGER) @PrimaryKey @AutoIncrement const val ID = "_id"
 
     @DataType(Type.INTEGER) @NotNull
-    @References(table = NotesDatabase.NOTES, column = NoteColumns.ID)
+    @References(table = NotesDatabase.NOTES, column = NoteColumns.ID,
+            onDelete = ForeignKeyAction.CASCADE)
     const val NOTE_ID = "note_id"
 
     @DataType(Type.TEXT) @NotNull

schematic-sample/src/main/java/net/simonvt/schematic/sample/database/NoteColumns.java

@@ -20,6 +20,7 @@
 import net.simonvt.schematic.annotation.Check;
 import net.simonvt.schematic.annotation.DataType;
 import net.simonvt.schematic.annotation.PrimaryKey;
+import net.simonvt.schematic.annotation.ForeignKeyAction;
 import net.simonvt.schematic.annotation.References;
 import net.simonvt.schematic.sample.database.NotesDatabase.Tables;
 
@@ -32,8 +33,8 @@ public interface NoteColumns {
 
   @DataType(INTEGER) @PrimaryKey @AutoIncrement String ID = "_id";
 
-  @DataType(INTEGER) @References(table = Tables.LISTS, column = ListColumns.ID) String LIST_ID =
-      "listId";
+  @DataType(INTEGER) @References(table = Tables.LISTS, column = ListColumns.ID,
+          onDelete = ForeignKeyAction.CASCADE) String LIST_ID = "listId";
 
   @DataType(TEXT) String NOTE = "note";
 

schematic-sample/src/main/java/net/simonvt/schematic/sample/database/TagColumns.java

@@ -6,6 +6,7 @@
 import net.simonvt.schematic.annotation.DataType.Type;
 import net.simonvt.schematic.annotation.NotNull;
 import net.simonvt.schematic.annotation.PrimaryKey;
+import net.simonvt.schematic.annotation.ForeignKeyAction;
 import net.simonvt.schematic.annotation.References;
 import net.simonvt.schematic.annotation.UniqueConstraint;
 
@@ -25,7 +26,8 @@ public interface TagColumns {
 
   @DataType(Type.INTEGER)
   @NotNull
-  @References(table = NotesDatabase.NOTES, column = NoteColumns.ID)
+  @References(table = NotesDatabase.NOTES, column = NoteColumns.ID,
+          onDelete = ForeignKeyAction.CASCADE)
   String NOTE_ID = "note_id";
 
   @DataType(Type.TEXT)