org.ektorp.StreamingChangesResult Maven / Gradle / Ivy

package org.ektorp;

import java.io.Closeable;
import java.io.IOException;
import java.io.Serializable;
import java.util.Iterator;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.ektorp.changes.DocumentChange;
import org.ektorp.http.HttpResponse;
import org.ektorp.impl.changes.StdDocumentChange;

/**
 *
 * @author Sverre Kristian Valskrå
 */
public class StreamingChangesResult implements Serializable, Iterable, Closeable{

	private static final long serialVersionUID = 4750290767936801714L;
	private boolean iteratorCalled;

    @edu.umd.cs.findbugs.annotations.SuppressWarnings(value="SE_BAD_FIELD")
	private JsonParser jp;

	private long lastSeq = -1l;

    @edu.umd.cs.findbugs.annotations.SuppressWarnings(value="SE_BAD_FIELD")
    private final HttpResponse response;

	public StreamingChangesResult(ObjectMapper objectMapper, HttpResponse response) {
		this.response = response;
        try{
		    jp = objectMapper.getFactory().createParser(response.getContent());
		    jp.nextValue();
		    jp.nextValue();
		    jp.nextToken();
    	}catch(Exception e) {
            throw new DbAccessException(e);
        }
	}


	public Iterator iterator() {
		if (iteratorCalled) {
			throw new IllegalStateException("Iterator can only be called once!");
		}
		iteratorCalled = true;
		return new StreamingResultIterator();
	}

	/**
	 * Aborts the stream immediately
	 */
	public void abort() {
	    try{
	    response.abort();
	    }finally {
	        close();
	    }
	}
	public void close() {
		try {
			jp.close();
		} catch (IOException e) {
		}
	}

	/**
	 * This method can only be called after stream is fully red
	 *
	 * @return
	 */
	public long getLastSeq() {
	    if (lastSeq == -1) {
	        throw new IllegalStateException("Last seq can only be called after stream is fully iterated");
	    }
        return lastSeq;
    }


	private class StreamingResultIterator implements Iterator{
        private DocumentChange row;
        Boolean hasNext = null;



		public boolean hasNext() {
            findNext();
            return hasNext;
		}

		public DocumentChange next() {
            findNext();
            hasNext = null;
            return row;
		}

		protected void findNext() {
            if (hasNext == null) {
                try {
                    JsonNode jsonNode = jp.readValueAs(JsonNode.class);
                    if (jsonNode == null) {
                        jsonNode = jp.readValueAs(JsonNode.class);
                        lastSeq = jsonNode.get("last_seq").longValue();
                        close();
                        hasNext = false;
                    } else {
                        row = new StdDocumentChange(jsonNode);
                        hasNext = true;
                    }
                } catch (Exception e) {
                    throw new DbAccessException(e);
                }
            }
        }
		public void remove() {
			throw new UnsupportedOperationException();
		}
	}

}