package io.lumify.web.devTools;
import io.lumify.miniweb.HandlerChain;
import com.google.inject.Inject;
import io.lumify.core.config.Configuration;
import io.lumify.core.model.user.UserRepository;
import io.lumify.core.model.workQueue.WorkQueueRepository;
import io.lumify.core.model.workspace.WorkspaceRepository;
import io.lumify.core.util.LumifyLogger;
import io.lumify.core.util.LumifyLoggerFactory;
import io.lumify.web.BaseRequestHandler;
import org.securegraph.Authorizations;
import org.securegraph.Graph;
import org.securegraph.Property;
import org.securegraph.Vertex;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public class QueueVertices extends BaseRequestHandler {
private static final LumifyLogger LOGGER = LumifyLoggerFactory.getLogger(QueueVertices.class);
private final Graph graph;
private final WorkQueueRepository workQueueRepository;
@Inject
public QueueVertices(
UserRepository userRepository,
WorkspaceRepository workspaceRepository,
Configuration configuration,
Graph graph,
WorkQueueRepository workQueueRepository) {
super(userRepository, workspaceRepository, configuration);
this.graph = graph;
this.workQueueRepository = workQueueRepository;
}
@Override
public void handle(HttpServletRequest request, HttpServletResponse response, HandlerChain chain) throws Exception {
String propertyName = getOptionalParameter(request, "propertyName");
if (propertyName != null && propertyName.trim().length() == 0) {
propertyName = null;
}
final Authorizations authorizations = getUserRepository().getAuthorizations(getUserRepository().getSystemUser());
final String finalPropertyName = propertyName;
Thread t = new Thread(new Runnable() {
@Override
public void run() {
LOGGER.info("requeue all vertices (property: %s)", finalPropertyName);
Iterable<Vertex> vertices = graph.getVertices(authorizations);
for (Vertex vertex : vertices) {
if (finalPropertyName == null) {
workQueueRepository.pushElement(vertex);
} else {
Iterable<Property> properties = vertex.getProperties(finalPropertyName);
for (Property property : properties) {
workQueueRepository.pushGraphPropertyQueue(vertex, property);
}
}
}
workQueueRepository.flush();
LOGGER.info("requeue all vertices complete");
}
});
t.setName("requeue-vertices");
t.start();
respondWithHtml(response, "Started requeue thread");
}
}