1 24 package org.riotfamily.revolt.refactor; 25 26 import org.riotfamily.revolt.Dialect; 27 import org.riotfamily.revolt.Refactoring; 28 import org.riotfamily.revolt.Script; 29 import org.riotfamily.revolt.definition.Database; 30 import org.riotfamily.revolt.definition.ForeignKey; 31 32 36 public class AddForeignKey implements Refactoring { 37 38 private String table; 39 40 private ForeignKey foreignKey; 41 42 public AddForeignKey() { 43 } 44 45 public AddForeignKey(String table, ForeignKey foreignKey) { 46 this.table = table; 47 this.foreignKey = foreignKey; 48 } 49 50 public void setTable(String table) { 51 this.table = table; 52 } 53 54 public void setForeignKey(ForeignKey foreignKey) { 55 this.foreignKey = foreignKey; 56 } 57 58 public void alterModel(Database database) { 59 database.getTable(table).addForeignKey(foreignKey); 60 } 61 62 public Script getScript(Dialect dialect) { 63 return dialect.addForeignKey(table, foreignKey); 64 } 65 66 } 67 | Popular Tags |