/**
* Copyright (C) 2010-2014 Morgner UG (haftungsbeschränkt)
*
* This file is part of Structr <http://structr.org>.
*
* Structr is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* Structr is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Structr. If not, see <http://www.gnu.org/licenses/>.
*/
package org.structr.core.graph;
import java.util.LinkedList;
import java.util.List;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.tooling.GlobalGraphOperations;
import org.structr.common.error.FrameworkException;
import org.structr.core.entity.AbstractNode;
//~--- JDK imports ------------------------------------------------------------
import java.util.logging.Level;
import java.util.logging.Logger;
import org.structr.common.SecurityContext;
import org.structr.core.GraphObject;
import org.structr.core.app.StructrApp;
//~--- classes ----------------------------------------------------------------
/**
* Clears database.
*
* This command takes no parameters.
*
* @author Axel Morgner
*/
public class ClearDatabase extends NodeServiceCommand {
private static final Logger logger = Logger.getLogger(ClearDatabase.class.getName());
//~--- methods --------------------------------------------------------
public void execute() throws FrameworkException {
final GraphDatabaseService graphDb = (GraphDatabaseService) arguments.get("graphDb");
final NodeFactory nodeFactory = new NodeFactory(securityContext);
if (graphDb != null) {
final List<AbstractNode> result = new LinkedList<>();
try (final Tx tx = StructrApp.getInstance().tx()) {
result.addAll(nodeFactory.bulkInstantiate(GlobalGraphOperations.at(graphDb).getAllNodes()));
}
long deletedNodes = bulkGraphOperation(securityContext, result, 1000, "ClearDatabase", new BulkGraphOperation<AbstractNode>() {
@Override
public void handleGraphObject(SecurityContext securityContext, AbstractNode node) {
// Delete only "our" nodes
if (node.getProperty(GraphObject.id) != null) {
try {
StructrApp.getInstance(securityContext).delete(node);
} catch (FrameworkException fex) {
logger.log(Level.WARNING, "Unable to delete node {0}: {1}", new Object[] { node.getUuid(), fex.getMessage() } );
}
}
}
@Override
public void handleThrowable(SecurityContext securityContext, Throwable t, AbstractNode node) {
logger.log(Level.WARNING, "Unable to delete node {0}: {1}", new Object[] { node.getUuid(), t.getMessage() } );
}
@Override
public void handleTransactionFailure(SecurityContext securityContext, Throwable t) {
logger.log(Level.WARNING, "Unable to clear database: {0}", t.getMessage() );
}
});
logger.log(Level.INFO, "Finished deleting {0} nodes", deletedNodes);
}
}
}