br.com.anteros.nosql.persistence.mongodb.query.MongoIterator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Anteros-NoSql-Persistence-MongoDB Show documentation
Anteros NoSQL Persistence MongoDB.
There is a newer version: 1.0.6
Show newest version
package br.com.anteros.nosql.persistence.mongodb.query;


import java.util.Iterator;
import java.util.NoSuchElementException;

import org.bson.Document;

import com.mongodb.client.MongoCursor;

import br.com.anteros.nosql.persistence.session.NoSQLSession;
import br.com.anteros.nosql.persistence.session.cache.NoSQLEntityCache;
import br.com.anteros.nosql.persistence.session.mapping.AbstractNoSQLObjectMapper;


public class MongoIterator implements Iterable, Iterator {
    private final Iterator wrapped;
    private final AbstractNoSQLObjectMapper mapper;
    private final Class clazz;
    private final String collection;
    private final NoSQLEntityCache cache;
    private long driverTime;
    private long mapperTime;
    private NoSQLSession session;


    public MongoIterator(final NoSQLSession session, final Iterator it, final AbstractNoSQLObjectMapper mapper, final Class clazz,
                           final String collection, final NoSQLEntityCache cache) {
        wrapped = it;
        this.mapper = mapper;
        this.clazz = clazz;
        this.collection = collection;
        this.cache = cache;
        this.session = session;
    }

  
    public void close() {
        if (wrapped != null && wrapped instanceof MongoCursor) {
            ((MongoCursor) wrapped).close();
        }
    }


    public Class getClazz() {
        return clazz;
    }

    public String getCollection() {
        return collection;
    }

    public MongoCursor getCursor() {
        return (MongoCursor) wrapped;
    }

 
    public long getDriverTime() {
        return driverTime;
    }

  
    public AbstractNoSQLObjectMapper getMapper() {
        return mapper;
    }

    
    public long getMapperTime() {
        return mapperTime;
    }

    @Override
    public boolean hasNext() {
        if (wrapped == null) {
            return false;
        }
        final long start = System.currentTimeMillis();
        final boolean ret = wrapped.hasNext();
        driverTime += System.currentTimeMillis() - start;
        return ret;
    }

    @Override
    public V next() {
        if (!hasNext()) {
            throw new NoSuchElementException();
        }
        final Document dbObj = getNext();
        try {
			return processItem(dbObj);
		} catch (Exception e) {
			throw new RuntimeException(e); 
		}
    }

    @Override
    public void remove() {
        final long start = System.currentTimeMillis();
        wrapped.remove();
        driverTime += System.currentTimeMillis() - start;
    }

    @Override
    public Iterator iterator() {
        return this;
    }

    @SuppressWarnings("unchecked")
    protected V convertItem(final Document dbObj) throws Exception {
        return (V) mapper.fromDocument(session, clazz, dbObj, cache);
    }

    protected Document getNext() {
        final long start = System.currentTimeMillis();
        final Document dbObj = wrapped.next();
        driverTime += System.currentTimeMillis() - start;
        return dbObj;
    }

    private V processItem(final Document dbObj) throws Exception {
        final long start = System.currentTimeMillis();
        final V item = convertItem(dbObj);
        mapperTime += System.currentTimeMillis() - start;
        return item;
    }

    public NoSQLSession getSession() {
        return session;
    }
}