com.arangodb.entity.CursorEntity Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation
ArangoDB Java Driver
There is a newer version: 7.15.0
/*
 * Copyright (C) 2012 tamtam180
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.arangodb.entity;

import java.util.Iterator;
import java.util.List;
import java.util.Map;

import com.arangodb.NonUniqueResultException;
import com.arangodb.util.CollectionUtils;

/**
 * Cursor entity that represents the result of a AQL query.
 *
 * @author tamtam180 - kirscheless at gmail.com
 * @author mrbatista
 *
 */
public class CursorEntity extends BaseEntity implements Iterable {

	/**
	 * True if the cursor has more results.
	 */
	boolean hasMore;

	/**
	 * The amount of results in the cursor
	 */
	int count = -1;

	/**
	 * The number of results before the final LIMIT
	 */
	int fullCount = -1;

	/**
	 * The cursor id
	 */
	long cursorId = -1;

	/**
	 * A list of bind variables returned by the query
	 */
	List bindVars;

	/**
	 * A list of extra data returned by the query
	 */
	Map extra;

	/**
	 * A list of objects containing the results
	 */
	List results;

	@SuppressWarnings("unchecked")
	@Override
	public Iterator iterator() {
		return (Iterator) CollectionUtils.safetyIterator(results);
	}

	/**
	 * The size of the cursor results.
	 *
	 * @return int
	 */
	public int size() {
		if (results == null) {
			return 0;
		}
		return results.size();
	}

	/**
	 * Returns the cursor element at position *index*
	 *
	 * @param index
	 * @return Object
	 */
	public T get(int index) {
		rangeCheck(index);
		return results.get(index);
	}

	private void rangeCheck(int index) {
		int size = size();
		if (index < 0 || index >= size) {
			throw new IndexOutOfBoundsException("Index: " + index + ", Size: " + size);
		}
	}

	/**
	 * Returns the DocumentEntity objects of this CursorDocumentEntity
	 * 
	 * @return list of DocumentEntity objects
	 */
	public List getResults() {
		return results;
	}

	/**
	 * Return a single instance that matches the query, or null if the query
	 * returns no results.
	 * 
	 * Throws NonUniqueResultException (RuntimeException) if there is more than
	 * one matching result
	 * 
	 * @return the single result or null
	 */
	public T getUniqueResult() {
		int size = size();
		if (size == 0) {
			return null;
		}
		if (size > 1) {
			throw new NonUniqueResultException(size);
		}
		return get(0);
	}

	public boolean isHasMore() {
		return hasMore;
	}

	public boolean hasMore() {
		return hasMore;
	}

	public int getFullCount() {
		return fullCount;
	}

	public int getCount() {
		return count;
	}

	public long getCursorId() {
		return cursorId;
	}

	public List getBindVars() {
		return bindVars;
	}

	public Map getExtra() {
		return extra;
	}

	public void setResults(List results) {
		this.results = results;
	}

	public void setHasMore(boolean hasMore) {
		this.hasMore = hasMore;
	}

	public void setFullCount(int count) {
		this.fullCount = count;
	}

	public void setCount(int count) {
		this.count = count;
	}

	public void setCursorId(long cursorId) {
		this.cursorId = cursorId;
	}

	public void setBindVars(List bindVars) {
		this.bindVars = bindVars;
	}

	public void setExtra(Map extra) {
		this.extra = extra;
	}

}