org.umlg.sqlg.test.tp3.SqlgAbstractGraphProvider Maven / Gradle / Ivy
package org.umlg.sqlg.test.tp3;
import org.apache.commons.configuration.Configuration;
import org.apache.tinkerpop.gremlin.AbstractGraphProvider;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.DefaultGraphTraversal;
import org.apache.tinkerpop.gremlin.structure.Element;
import org.apache.tinkerpop.gremlin.structure.Graph;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.umlg.sqlg.SqlgPlugin;
import org.umlg.sqlg.sql.dialect.SqlDialect;
import org.umlg.sqlg.structure.*;
import org.umlg.sqlg.structure.SqlgDataSourceFactory.SqlgDataSource;
import java.beans.PropertyVetoException;
import java.sql.*;
import java.util.HashSet;
import java.util.Set;
/**
* Created by pieter on 2015/12/13.
*/
public abstract class SqlgAbstractGraphProvider extends AbstractGraphProvider {
private static Logger logger = LoggerFactory.getLogger(SqlgGraph.class.getName());
private static final Set IMPLEMENTATIONS = new HashSet() {{
add(SqlgEdge.class);
add(SqlgElement.class);
add(SqlgGraph.class);
add(SqlgProperty.class);
add(SqlgVertex.class);
add(SqlgVertexProperty.class);
add(DefaultGraphTraversal.class);
}};
@Override
public void clear(final Graph g, final Configuration configuration) throws Exception {
logger.debug("clearing datasource " + configuration.getString("jdbc.url"));
SqlgDataSource sqlgDataSource = null;
if (null != g) {
if (g.features().graph().supportsTransactions() && g.tx().isOpen()) {
g.tx().rollback();
}
g.close();
}
SqlgPlugin plugin = getSqlgPlugin();
SqlDialect sqlDialect = plugin.instantiateDialect();
try {
sqlgDataSource = SqlgGraph.createDataSourceFactory(configuration).setup(plugin.getDriverFor(configuration.getString("jdbc.url")), configuration);
try (Connection conn = sqlgDataSource.getDatasource().getConnection()) {
DatabaseMetaData metadata = conn.getMetaData();
if (sqlDialect.supportsCascade()) {
String tableNamePattern = "%";
String[] types = {"TABLE"};
ResultSet resultSet = metadata.getTables(null, null, tableNamePattern, types);
while (resultSet.next()) {
String schema = resultSet.getString(2);
if (!sqlDialect.getGisSchemas().contains(schema)) {
StringBuilder sql = new StringBuilder("DROP TABLE ");
sql.append(sqlDialect.maybeWrapInQoutes(resultSet.getString(2)));
sql.append(".");
sql.append(sqlDialect.maybeWrapInQoutes(resultSet.getString(3)));
sql.append(" CASCADE");
if (sqlDialect.needsSemicolon()) {
sql.append(";");
}
try (PreparedStatement preparedStatement = conn.prepareStatement(sql.toString())) {
preparedStatement.executeUpdate();
}
}
}
resultSet = metadata.getSchemas(null, null);
while (resultSet.next()) {
String schema = resultSet.getString(1);
if (!sqlDialect.getDefaultSchemas().contains(schema) && !sqlDialect.getGisSchemas().contains(schema)) {
StringBuilder sql = new StringBuilder("DROP SCHEMA ");
sql.append(sqlDialect.maybeWrapInQoutes(schema));
if (sqlDialect.needsSchemaDropCascade()) {
sql.append(" CASCADE");
}
if (sqlDialect.needsSemicolon()) {
sql.append(";");
}
try (PreparedStatement preparedStatement = conn.prepareStatement(sql.toString())) {
preparedStatement.executeUpdate();
}
}
}
}
} catch (SQLException e) {
throw new RuntimeException(e);
}
} catch (PropertyVetoException e) {
throw new RuntimeException(e);
} finally {
if (sqlgDataSource != null) {
sqlgDataSource.close();
}
}
}
@Override
public Set getImplementations() {
return IMPLEMENTATIONS;
}
@Override
public Object convertId(final Object id, final Class c) {
return "jippo.jippo" + SchemaManager.LABEL_SEPARATOR + id.toString();
}
public abstract SqlgPlugin getSqlgPlugin();
}