com.arangodb.internal.cursor.ArangoCursorImpl Maven / Gradle / Ivy
/*
* DISCLAIMER
*
* Copyright 2016 ArangoDB GmbH, Cologne, Germany
*
* 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.
*
* Copyright holder is ArangoDB GmbH, Cologne, Germany
*/
package com.arangodb.internal.cursor;
import com.arangodb.ArangoCursor;
import com.arangodb.ArangoIterator;
import com.arangodb.entity.CursorStats;
import com.arangodb.entity.CursorWarning;
import com.arangodb.internal.ArangoCursorExecute;
import com.arangodb.internal.InternalArangoDatabase;
import com.arangodb.internal.cursor.entity.InternalCursorEntity;
import com.arangodb.shaded.fasterxml.jackson.databind.JsonNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.NoSuchElementException;
import java.util.List;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
/**
* @author Mark Vollmary
*/
public class ArangoCursorImpl implements ArangoCursor {
private final static Logger LOG = LoggerFactory.getLogger(ArangoCursorImpl.class);
protected final ArangoCursorIterator iterator;
private final Class type;
private final String id;
private final ArangoCursorExecute execute;
private final boolean pontentialDirtyRead;
private final boolean allowRetry;
public ArangoCursorImpl(final InternalArangoDatabase db, final ArangoCursorExecute execute,
final Class type, final InternalCursorEntity result) {
super();
this.execute = execute;
this.type = type;
id = result.getId();
pontentialDirtyRead = result.isPontentialDirtyRead();
iterator = new ArangoCursorIterator<>(id, type, execute, db, result);
this.allowRetry = result.getNextBatchId() != null;
}
@Override
public void close() {
if (getId() != null && (allowRetry || iterator.result.getHasMore())) {
getExecute().close(getId());
}
}
@Override
public T next() {
return iterator.next();
}
@Override
public String getId() {
return id;
}
@Override
public Class getType() {
return type;
}
@Override
public Integer getCount() {
return iterator.result.getCount();
}
@Override
public CursorStats getStats() {
final InternalCursorEntity.Extras extra = iterator.result.getExtra();
return extra != null ? extra.getStats() : null;
}
@Override
public Collection getWarnings() {
final InternalCursorEntity.Extras extra = iterator.result.getExtra();
return extra != null ? extra.getWarnings() : null;
}
@Override
public boolean isCached() {
final Boolean cached = iterator.result.getCached();
return Boolean.TRUE.equals(cached);
}
@Override
public boolean hasNext() {
return iterator.hasNext();
}
@Override
public List asListRemaining() {
final List remaining = new ArrayList<>();
while (hasNext()) {
remaining.add(next());
}
try {
close();
} catch (final Exception e) {
LOG.warn("Could not close cursor: ", e);
}
return remaining;
}
@Override
public boolean isPotentialDirtyRead() {
return pontentialDirtyRead;
}
@Override
public ArangoIterator iterator() {
return iterator;
}
@Override
public String getNextBatchId() {
return iterator.result.getNextBatchId();
}
protected ArangoCursorExecute getExecute() {
return execute;
}
public Stream stream() {
return StreamSupport.stream(spliterator(), false);
}
protected static class ArangoCursorIterator implements ArangoIterator {
private final String cursorId;
private final Class type;
private final InternalArangoDatabase db;
private final ArangoCursorExecute execute;
private InternalCursorEntity result;
private Iterator arrayIterator;
protected ArangoCursorIterator(final String cursorId, final Class type, final ArangoCursorExecute execute,
final InternalArangoDatabase db, final InternalCursorEntity result) {
this.cursorId = cursorId;
this.type = type;
this.execute = execute;
this.db = db;
this.result = result;
arrayIterator = result.getResult().iterator();
}
@Override
public boolean hasNext() {
return arrayIterator.hasNext() || result.getHasMore();
}
@Override
public T next() {
if (!arrayIterator.hasNext() && Boolean.TRUE.equals(result.getHasMore())) {
result = execute.next(cursorId, result.getNextBatchId());
arrayIterator = result.getResult().iterator();
}
if (!hasNext()) {
throw new NoSuchElementException();
}
return deserialize(db.getSerde().serialize(arrayIterator.next()), type);
}
private R deserialize(final byte[] result, final Class type) {
return db.getSerde().deserializeUserData(result, type);
}
}
}