diff --git a/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateDatabaseChangeLogLockTableGeneratorSpanner.java b/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateDatabaseChangeLogLockTableGeneratorSpanner.java index da696ad4..a3c3ba60 100644 --- a/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateDatabaseChangeLogLockTableGeneratorSpanner.java +++ b/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateDatabaseChangeLogLockTableGeneratorSpanner.java @@ -6,6 +6,7 @@ import liquibase.sqlgenerator.SqlGeneratorChain; import liquibase.sqlgenerator.core.CreateDatabaseChangeLogLockTableGenerator; import liquibase.statement.core.CreateDatabaseChangeLogLockTableStatement; +import liquibase.ext.spanner.ICloudSpanner; public class CreateDatabaseChangeLogLockTableGeneratorSpanner extends CreateDatabaseChangeLogLockTableGenerator { @@ -31,4 +32,9 @@ public Sql[] generateSql( public int getPriority() { return PRIORITY_DATABASE; } + + @Override + public boolean supports(CreateDatabaseChangeLogLockTableStatement statement, Database database) { + return (database instanceof ICloudSpanner); + } } diff --git a/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateTableGeneratorSpanner.java b/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateTableGeneratorSpanner.java index 05178d15..59bfa1d7 100644 --- a/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateTableGeneratorSpanner.java +++ b/src/main/java/liquibase/ext/spanner/sqlgenerator/CreateTableGeneratorSpanner.java @@ -21,6 +21,7 @@ import liquibase.sqlgenerator.core.CreateTableGenerator; import liquibase.statement.core.CreateTableStatement; import liquibase.structure.DatabaseObject; +import liquibase.ext.spanner.ICloudSpanner; public class CreateTableGeneratorSpanner extends CreateTableGenerator { @@ -75,4 +76,9 @@ public Sql[] generateSql( public int getPriority() { return PRIORITY_DATABASE; } + + @Override + public boolean supports(CreateTableStatement statement, Database database) { + return (database instanceof ICloudSpanner); + } } diff --git a/src/main/java/liquibase/ext/spanner/sqlgenerator/InitializeChangeLogLockTableGeneratorSpanner.java b/src/main/java/liquibase/ext/spanner/sqlgenerator/InitializeChangeLogLockTableGeneratorSpanner.java index 9621fcd0..45c0a69a 100644 --- a/src/main/java/liquibase/ext/spanner/sqlgenerator/InitializeChangeLogLockTableGeneratorSpanner.java +++ b/src/main/java/liquibase/ext/spanner/sqlgenerator/InitializeChangeLogLockTableGeneratorSpanner.java @@ -23,6 +23,7 @@ import liquibase.statement.core.DeleteStatement; import liquibase.statement.core.InitializeDatabaseChangeLogLockTableStatement; import liquibase.statement.core.InsertStatement; +import liquibase.ext.spanner.ICloudSpanner; public class InitializeChangeLogLockTableGeneratorSpanner extends AbstractSqlGenerator { @@ -62,4 +63,10 @@ public Sql[] generateSql( }, database); } -} + + @Override + public boolean supports(InitializeDatabaseChangeLogLockTableStatement statement, Database database) { + return (database instanceof ICloudSpanner); + } + +} \ No newline at end of file diff --git a/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertOrUpdateGeneratorSpanner.java b/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertOrUpdateGeneratorSpanner.java index cd69855f..1cb02c67 100644 --- a/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertOrUpdateGeneratorSpanner.java +++ b/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertOrUpdateGeneratorSpanner.java @@ -97,4 +97,5 @@ protected String getRecordCheck(InsertOrUpdateStatement insertOrUpdateStatement, protected String getElse(Database database) { return ""; } + } diff --git a/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertWithSelectGeneratorSpanner.java b/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertWithSelectGeneratorSpanner.java index 57c9144a..e7108c3a 100644 --- a/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertWithSelectGeneratorSpanner.java +++ b/src/main/java/liquibase/ext/spanner/sqlgenerator/InsertWithSelectGeneratorSpanner.java @@ -25,6 +25,7 @@ import liquibase.sqlgenerator.core.InsertGenerator; import liquibase.statement.DatabaseFunction; import liquibase.statement.core.InsertStatement; +import liquibase.ext.spanner.ICloudSpanner; /** Generator for INSERT statements in the form 'INSERT INTO FOO (..) SELECT ...'. */ public class InsertWithSelectGeneratorSpanner extends InsertGenerator { @@ -35,6 +36,11 @@ public int getPriority() { return 0; } + @Override + public boolean supports(InsertStatement statement, Database database) { + return (database instanceof ICloudSpanner); + } + @Override public Sql[] generateSql(InsertStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {