migratedb.v1.commandline.driversupport.DerbyDriverSupport Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of migratedb-commandline Show documentation

There is a newer version: 1.2.0

Show newest version

package migratedb.v1.commandline.driversupport;

import com.google.auto.service.AutoService;
import migratedb.v1.commandline.DriverSupport;
import migratedb.v1.core.api.logging.Log;

import java.sql.Driver;
import java.sql.SQLException;
import java.util.Properties;

@AutoService(DriverSupport.class)
public class DerbyDriverSupport implements DriverSupport {
    private static final Log LOG = Log.getLog(DerbyDriverSupport.class);

    @Override
    public String getName() {
        return "Derby";
    }

    @Override
    public boolean handlesJdbcUrl(String url) {
        return url.startsWith("jdbc:derby:") || url.startsWith("jdbc:p6spy:derby:");
    }

    @Override
    public String getDriverClass(String url, ClassLoader classLoader) {
        if (url.startsWith("jdbc:p6spy:derby:")) {
            return "com.p6spy.engine.spy.P6SpyDriver";
        }
        if (url.startsWith("jdbc:derby://")) {
            return "org.apache.derby.jdbc.ClientDriver";
        }
        return "org.apache.derby.jdbc.EmbeddedDriver";
    }


    @Override
    public void shutdownDatabase(String url, Driver driver) {
        // only do this on the embedded version
        if (!url.startsWith("jdbc:derby://")) {
            try {
                int i = url.indexOf(";");
                String shutdownUrl = (i < 0 ? url : url.substring(0, i)) + ";shutdown=true";
                driver.connect(shutdownUrl, new Properties()).close();
            } catch (SQLException e) {
                LOG.debug("Unexpected error on Derby Embedded Database shutdown: " + e.getMessage());
            }
        }
    }
}