ArangoDBBatchVertex.java

//////////////////////////////////////////////////////////////////////////////////////////
//
// Implementation of the Blueprints Interface for ArangoDB by triAGENS GmbH Cologne.
//
// Copyright triAGENS GmbH Cologne.
//
//////////////////////////////////////////////////////////////////////////////////////////

package com.arangodb.blueprints.batch;

import java.util.HashMap;
import java.util.Map;

import org.apache.log4j.Logger;

import com.arangodb.blueprints.client.ArangoDBBaseDocument;
import com.arangodb.blueprints.client.ArangoDBException;
import com.arangodb.blueprints.client.ArangoDBSimpleVertex;
import com.tinkerpop.blueprints.Direction;
import com.tinkerpop.blueprints.Edge;
import com.tinkerpop.blueprints.Vertex;
import com.tinkerpop.blueprints.VertexQuery;
import com.tinkerpop.blueprints.util.ExceptionFactory;
import com.tinkerpop.blueprints.util.StringFactory;

/**
 * The arangodb batch vertex class
 * 
 * @author Achim Brandt (http://www.triagens.de)
 * @author Johannes Gocke (http://www.triagens.de)
 * @author Guido Schwab (http://www.triagens.de)
 */

public class ArangoDBBatchVertex extends ArangoDBBatchElement implements Vertex {

	/**
	 * the logger
	 */
	private static final Logger logger = Logger.getLogger(ArangoDBBatchVertex.class);

	private ArangoDBBatchVertex(ArangoDBBatchGraph graph, ArangoDBSimpleVertex vertex) {
		this.graph = graph;
		this.document = vertex;
	}

	/**
	 * Creates a vertex
	 * 
	 * @param graph
	 *            a ArangoDBBatchGraph
	 * @param id
	 *            the id (key) of the vertex (can be null)
	 */

	static ArangoDBBatchVertex create(ArangoDBBatchGraph graph, Object id) {
		String key = (id != null) ? id.toString() : null;

		if (key == null) {
			key = graph.getNewId().toString();
		}

		Map<String, Object> properties = new HashMap<String, Object>();
		properties.put(ArangoDBBaseDocument._REV, "");
		properties.put(ArangoDBBaseDocument._ID, "");
		properties.put(ArangoDBBaseDocument._KEY, key);

		try {
			ArangoDBSimpleVertex v = new ArangoDBSimpleVertex(properties);
			return build(graph, v);
		} catch (ArangoDBException e) {
			logger.warn("could create batch vertex", e);
			throw new IllegalArgumentException(e.getMessage());
		}
	}

	/**
	 * Creates a vertex by loading it
	 * 
	 * @param graph
	 *            a ArangoDBGraph
	 * @param id
	 *            the id (key) of the vertex (can be null)
	 */

	static ArangoDBBatchVertex load(ArangoDBBatchGraph graph, Object id) {
		if (id == null) {
			throw ExceptionFactory.vertexIdCanNotBeNull();
		}

		String key = id.toString();

		ArangoDBBatchVertex vert = graph.vertexCache.get(key);
		if (vert != null) {
			return vert;
		}

		try {
			ArangoDBSimpleVertex v = graph.client.getVertex(graph.getRawGraph(), key);
			return build(graph, v);
		} catch (ArangoDBException e) {
			logger.warn("could read batch vertex", e);
			return null;
		}
	}

	static ArangoDBBatchVertex build(ArangoDBBatchGraph graph, ArangoDBSimpleVertex simpleVertex)
			throws ArangoDBException {
		String id = simpleVertex.getDocumentKey();

		ArangoDBBatchVertex vert = graph.vertexCache.get(id);
		if (vert != null) {
			vert.setDocument(simpleVertex);
			return vert;
		}

		ArangoDBBatchVertex newVertex = new ArangoDBBatchVertex(graph, simpleVertex);
		graph.vertexCache.put(newVertex.getRaw().getDocumentKey(), newVertex);
		graph.addCreatedVertex(newVertex);
		return newVertex;
	}

	@Override
	public Iterable<Edge> getEdges(Direction direction, String... labels) {
		throw new UnsupportedOperationException();
	}

	@Override
	public Iterable<Vertex> getVertices(Direction direction, String... labels) {
		throw new UnsupportedOperationException();
	}

	@Override
	public VertexQuery query() {
		throw new UnsupportedOperationException();
	}

	/**
	 * Returns the arangodb raw vertex
	 * 
	 * @return a ArangoDBSimpleVertex
	 */

	public ArangoDBSimpleVertex getRawVertex() {
		return (ArangoDBSimpleVertex) document;
	}

	@Override
	public String toString() {
		return StringFactory.vertexString(this);
	}

	@Override
	public void remove() {
		throw new UnsupportedOperationException();
	}

	@Override
	public void save() throws ArangoDBException {
		throw new UnsupportedOperationException();
	}

	@Override
	public Edge addEdge(String label, Vertex inVertex) {
		return ArangoDBBatchEdge.create(this.graph, null, this, inVertex, label);
	}

}