com.despegar.integration.mongo.connector.MongoCollection Maven / Gradle / Ivy

Go to download
package com.despegar.integration.mongo.connector;

import java.util.List;

import org.apache.commons.lang.mutable.MutableInt;

import com.despegar.integration.mongo.entities.GenericIdentifiableEntity;
import com.despegar.integration.mongo.query.AggregateQuery;
import com.despegar.integration.mongo.query.MongoAggregationQuery;
import com.despegar.integration.mongo.query.MongoQuery;
import com.despegar.integration.mongo.query.MongoUpdate;
import com.despegar.integration.mongo.query.Query;
import com.despegar.integration.mongo.query.Update;
import com.mongodb.AggregationOptions;
import com.mongodb.ReadPreference;

public class MongoCollection> {

    protected Class clazz;
    private String collectionName;
    protected MongoDao mongoDao;

    MongoCollection(String collectionName, Class collectionClass, MongoDao mongoDao) {
        this.collectionName = collectionName;
        this.clazz = collectionClass;
        this.mongoDao = mongoDao;
    }

    public T findOne() {
        Query query = new Query();
        query.limit(1);

        return this.findOne(query);
    }

    public T findOne(final Query query) {
        final MongoQuery mhq = new MongoQuery(query);

        return this.mongoDao.findOne(mhq.getQuery(), mhq.getSortInfo(), mhq.getQueryPage());
    }

    public  T findOne(final X id) {
        return this.mongoDao.findOne(id);
    }

    public List find() {
        return this.find(null);
    }

    public List find(final Query query) {
        return this.find(query, null);
    }

    public List find(final Query query, final MutableInt count) {

        if (query == null) {
            return this.mongoDao.find();
        }

        final MongoQuery mongoQuery = new MongoQuery(query);

        return this.mongoDao.find(mongoQuery.getQuery(), null, mongoQuery.getSortInfo(), mongoQuery.getQueryPage(), count,
            this.isCrucialDataIntegration(query));
    }

    public Integer count(final Query query) {
        if (query == null) {
            return this.mongoDao.getTotalObjectsInCollection(this.collectionName);
        }

        final MongoQuery mongoQuery = new MongoQuery(query);
        return this.mongoDao.getTotalObjectsInCollection(this.collectionName, mongoQuery.getQuery());
    }

    public  X add(final T t) {
        t.setId(null);
        return this.mongoDao.insert(t);
    }

    public  X insertIfNotPresent(final T t) {
        return this.mongoDao.insert(t);
    }

    public  X save(final T t) {
        return this.mongoDao.updateOrInsert(t);
    }

    public  Integer update(final Query query, final Update updateQuery) {
        final MongoQuery mongoQuery = new MongoQuery(query);
        final MongoUpdate mongoUpdateQuery = new MongoUpdate(updateQuery);
        return this.mongoDao.update(mongoQuery.getQuery(), mongoUpdateQuery.getUpdate(), false);
    }

    public  Integer update(final X id, final Update updateQuery) {
        Query query = new Query();
        query.equals("_id", id);
        final MongoQuery mongoQuery = new MongoQuery(query);
        final MongoUpdate mongoUpdateQuery = new MongoUpdate(updateQuery);
        return this.mongoDao.update(mongoQuery.getQuery(), mongoUpdateQuery.getUpdate(), false);
    }

    public T getAndUpdate(final Query query, boolean remove, final Update updateQuery, boolean returnNew) {
        MongoQuery mhq = new MongoQuery(query);
        MongoUpdate mhqUpdate = new MongoUpdate(updateQuery);

        return this.mongoDao.findAndModify(mhq.getQuery(), null, mhq.getSortInfo(), remove, mhqUpdate.getUpdate(),
            returnNew, false);
    }

    public  boolean remove(final X id) {
        return this.mongoDao.delete(this.collectionName, id);
    }

    public boolean remove(Query query) {
        final MongoQuery mongoQuery = new MongoQuery(query);
        return this.mongoDao.delete(this.collectionName, mongoQuery.getQuery());
    }

    public void removeAll() {
        this.mongoDao.dropCollection(this.collectionName);
    }

    /**
     * BETA! as Tusam said "this can fail", and we know how Tusam finish. We are working to find the best solution to
     * this framework, but you can test this. WARNING! aggregate only works with mongodb 2.6 or higher
     */
    public List aggregate(AggregateQuery query) {
        MongoAggregationQuery mongoHandlerAggregationQuery = new MongoAggregationQuery(query);
        return this.mongoDao.aggregate(mongoHandlerAggregationQuery.getQuery());
    }

    /**
     * BETA! as Tusam said "this can fail", and we know how Tusam finish. We are working to find the best solution to
     * this framework, but you can test this. WARNING! aggregate only works with mongodb 2.6 or higher
     */
    public  List aggregate(AggregateQuery query, Class returnClazz) {
        MongoAggregationQuery mongoHandlerAggregationQuery = new MongoAggregationQuery(query);
        return this.mongoDao.aggregate(mongoHandlerAggregationQuery.getQuery(), returnClazz);
    }

    /**
     * BETA! as Tusam said "this can fail", and we know how Tusam finish. We are working to find the best solution to
     * this framework, but you can test this. WARNING! aggregate only works with mongodb 2.6 or higher
     */
    public  List aggregate(AggregateQuery query, AggregationOptions options, Class returnClazz) {
        MongoAggregationQuery mongoHandlerAggregationQuery = new MongoAggregationQuery(query);
        return this.mongoDao.aggregate(mongoHandlerAggregationQuery.getQuery(), options, returnClazz);
    }

    public List distinct(String property) {
        return this.mongoDao.distinct(property);
    }

    public List distinct(String property, Query query) {
        MongoQuery q = new MongoQuery(query);
        return this.mongoDao.distinct(property, q.getQuery());
    }

    public Boolean exists(Query query) {
        MongoQuery q = new MongoQuery(query);
        return this.mongoDao.exists(q.getQuery());
    }

    private ReadPreference isCrucialDataIntegration(Query query) {
        if (query.isCrucialDataIntegration()) {
            return ReadPreference.primary();
        }

        return null;
    }

}