com.puresoltechnologies.purifinity.server.test.analysis.AnalysisStoreDatabaseHelper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of server.test Show documentation

The newest version!

package com.puresoltechnologies.purifinity.server.test.analysis;

import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.puresoltechnologies.ductiledb.tinkerpop.DuctileGraph;
import com.puresoltechnologies.purifinity.server.database.hbase.HBaseElementNames;

public class AnalysisStoreDatabaseHelper {

    private static final Logger logger = LoggerFactory.getLogger(AnalysisStoreDatabaseHelper.class);

    public static void cleanAnalysisStore(Connection connection, DuctileGraph graph) {
	try (PreparedStatement preparedStatement = connection.prepareStatement("DROP TABLE IF EXISTS ?")) {
	    Field[] fields = HBaseElementNames.class.getDeclaredFields();
	    for (Field field : fields) {
		String fieldName = field.getName();
		if (fieldName.startsWith("ANALYSIS_") && fieldName.endsWith("_TABLE")) {
		    String tableName = (String) field.get(null);
		    preparedStatement.setString(1, tableName);
		    preparedStatement.execute();
		}
	    }
	    connection.commit();
	} catch (IllegalArgumentException | IllegalAccessException e) {
	    throw new RuntimeException("Could not clean analysis store.", e);
	} catch (SQLException e) {
	    try {
		connection.rollback();
	    } catch (SQLException e1) {
		logger.warn("Could not rollback table deletions.", e1);
	    }
	    throw new RuntimeException("Could not clean analysis store.", e);
	}
	GraphTraversal vertices = graph.traversal().V().emit();
	while (vertices.hasNext()) {
	    Vertex vertex = vertices.next();
	    vertex.remove();
	}
	graph.tx().commit();
    }
}