/**
* 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.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.helpers.collection.Iterables;
import org.neo4j.tooling.GlobalGraphOperations;
import org.structr.common.SecurityContext;
import org.structr.common.StructrAndSpatialPredicate;
import org.structr.common.error.ErrorBuffer;
import org.structr.common.error.FrameworkException;
import org.structr.core.Result;
import org.structr.core.app.StructrApp;
import org.structr.core.entity.AbstractNode;
import org.structr.core.entity.AbstractRelationship;
import org.structr.schema.SchemaHelper;
//~--- classes ----------------------------------------------------------------
/**
* Rebuild index for nodes or relationships of given type.
*
* Use 'type' argument for node type, and 'relType' for relationship type.
*
* @author Axel Morgner
*/
public class BulkRebuildIndexCommand extends NodeServiceCommand implements MaintenanceCommand, TransactionPostProcess {
private static final Logger logger = Logger.getLogger(BulkRebuildIndexCommand.class.getName());
//~--- methods --------------------------------------------------------
@Override
public void execute(Map<String, Object> attributes) throws FrameworkException {
final String mode = (String) attributes.get("mode");
final String entityType = (String) attributes.get("type");
final String relType = (String) attributes.get("relType");
final GraphDatabaseService graphDb = (GraphDatabaseService) arguments.get("graphDb");
final SecurityContext superUserContext = SecurityContext.getSuperUserInstance();
final NodeFactory nodeFactory = new NodeFactory(superUserContext);
final RelationshipFactory relFactory = new RelationshipFactory(superUserContext);
Class type = null;
if (entityType != null) {
type = SchemaHelper.getEntityClassForRawType(entityType);
}
// final Result<AbstractNode> result = StructrApp.getInstance(securityContext).command(SearchNodeCommand.class).execute(true, false, Search.andExactType(type.getSimpleName()));
if (mode == null || "nodesOnly".equals(mode)) {
final List<AbstractNode> nodes = new LinkedList<>();
// instantiate all nodes in a single list
try (final Tx tx = StructrApp.getInstance().tx()) {
final Result<AbstractNode> result = nodeFactory.instantiateAll(Iterables.filter(new StructrAndSpatialPredicate(true, false, false), GlobalGraphOperations.at(graphDb).getAllNodes()));
for (AbstractNode node : result.getResults()) {
if (type == null || node.getClass().equals(type)) {
nodes.add(node);
}
}
tx.success();
}
if (type == null) {
logger.log(Level.INFO, "Node type not set or no entity class found. Starting (re-)indexing all nodes");
} else {
logger.log(Level.INFO, "Starting (re-)indexing all nodes of type {0}", new Object[]{type.getSimpleName()});
}
long count = bulkGraphOperation(securityContext, nodes, 100, "RebuildNodeIndex", new BulkGraphOperation<AbstractNode>() {
@Override
public void handleGraphObject(SecurityContext securityContext, AbstractNode node) {
try {
// Set type to update labels
final String type = node.getProperty(NodeInterface.type);
node.setProperty(NodeInterface.type, null);
node.setProperty(NodeInterface.type, type);
} catch (FrameworkException ex) {
ex.printStackTrace();
}
node.updateInIndex();
}
@Override
public void handleThrowable(SecurityContext securityContext, Throwable t, AbstractNode node) {
logger.log(Level.WARNING, "Unable to index node {0}: {1}", new Object[]{node, t.getMessage()});
}
@Override
public void handleTransactionFailure(SecurityContext securityContext, Throwable t) {
logger.log(Level.WARNING, "Unable to index node: {0}", t.getMessage());
}
});
logger.log(Level.INFO, "Done with (re-)indexing {0} nodes", count);
}
if (mode == null || "relsOnly".equals(mode)) {
final List<AbstractRelationship> rels = new LinkedList<>();
long count = 0;
// instantiate all relationships in a single list
try (final Tx tx = StructrApp.getInstance().tx()) {
final List<AbstractRelationship> unfilteredRels = relFactory.instantiate(Iterables.filter(new StructrAndSpatialPredicate(true, false, false), GlobalGraphOperations.at(graphDb).getAllRelationships()));
for (AbstractRelationship rel : unfilteredRels) {
if (relType == null || rel.getType().equals(relType)) {
rels.add(rel);
}
}
tx.success();
}
if (relType == null) {
logger.log(Level.INFO, "Relationship type not set, starting (re-)indexing all relationships");
} else {
logger.log(Level.INFO, "Starting (re-)indexing all relationships of type {0}", new Object[]{relType});
}
count = bulkGraphOperation(securityContext, rels, 100, "RebuildRelIndex", new BulkGraphOperation<AbstractRelationship>() {
@Override
public void handleGraphObject(SecurityContext securityContext, AbstractRelationship rel) {
rel.updateInIndex();
}
@Override
public void handleThrowable(SecurityContext securityContext, Throwable t, AbstractRelationship rel) {
logger.log(Level.WARNING, "Unable to index relationship {0}: {1}", new Object[]{rel, t.getMessage()});
}
@Override
public void handleTransactionFailure(SecurityContext securityContext, Throwable t) {
logger.log(Level.WARNING, "Unable to index relationship: {0}", t.getMessage());
}
});
logger.log(Level.INFO, "Done with (re-)indexing {0} relationships", count);
}
}
// ----- interface TransactionPostProcess -----
@Override
public boolean execute(SecurityContext securityContext, ErrorBuffer errorBuffer) throws FrameworkException {
execute(Collections.EMPTY_MAP);
return true;
}
@Override
public boolean requiresEnclosingTransaction() {
return false;
}
}