/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.ngrinder.infra.init;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import liquibase.Liquibase;
import liquibase.changelog.ChangeLogIterator;
import liquibase.changelog.DatabaseChangeLog;
import liquibase.changelog.filter.ContextChangeSetFilter;
import liquibase.changelog.filter.DbmsChangeSetFilter;
import liquibase.changelog.filter.ShouldRunChangeSetFilter;
import liquibase.changelog.visitor.UpdateVisitor;
import liquibase.database.Database;
import liquibase.exception.DatabaseException;
import liquibase.exception.LiquibaseException;
import liquibase.exception.LockException;
import liquibase.lockservice.LockServiceEx;
import liquibase.parser.ChangeLogParserFactory;
import liquibase.resource.ResourceAccessor;
import liquibase.util.StringUtils;
/**
* {@link Liquibase} extension to use {@link TypeConverter} for lock acquire.
*
* @author JunHo Yoon
* @since 3.0
*/
public class LiquibaseEx extends Liquibase {
private static final Logger LOGGER = LoggerFactory.getLogger(LiquibaseEx.class);
private final String changeLogFile;
/**
* Constructor.
*
* @param changeLogFile changeLogFile
* @param resourceAccessor resource accessor
* @param database database to be used
* @throws LiquibaseException occurs when initialization is failed.
*/
public LiquibaseEx(String changeLogFile, ResourceAccessor resourceAccessor, Database database)
throws LiquibaseException {
super(changeLogFile, resourceAccessor, database);
this.changeLogFile = changeLogFile;
}
/*
* (non-Javadoc)
*
* @see liquibase.Liquibase#update(java.lang.String)
*/
@Override
public void update(String contexts) throws LiquibaseException {
contexts = StringUtils.trimToNull(contexts);
LockServiceEx lockService = LockServiceEx.getInstance(database);
lockService.waitForLock();
try {
getChangeLogParameters().setContexts(StringUtils.splitAndTrim(contexts, ","));
DatabaseChangeLog changeLog = ChangeLogParserFactory.getInstance()
.getParser(this.changeLogFile, getFileOpener())
.parse(this.changeLogFile, getChangeLogParameters(), getFileOpener());
checkDatabaseChangeLogTable(true, changeLog, contexts);
changeLog.validate(database, contexts);
ChangeLogIterator changeLogIterator = getStandardChangelogIterator(contexts, changeLog);
changeLogIterator.run(new UpdateVisitor(database), database);
} finally {
try {
lockService.releaseLock();
} catch (LockException e) {
LOGGER.error("Error while releasing db lock", e);
}
}
}
private ChangeLogIterator getStandardChangelogIterator(String contexts, DatabaseChangeLog changeLog)
throws DatabaseException {
return new ChangeLogIterator(changeLog, new ShouldRunChangeSetFilter(database), new ContextChangeSetFilter(
contexts), new DbmsChangeSetFilter(database));
}
}