com.scaleset.search.mongo.MongoSearchDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scaleset-search Show documentation
Java object model for representing query requests in REST protocols.
There is a newer version: 0.24.0
package com.scaleset.search.mongo;

import com.mongodb.*;
import com.scaleset.search.AbstractSearchDao;
import com.scaleset.search.Query;
import com.scaleset.search.Results;

import java.util.ArrayList;
import java.util.List;

public class MongoSearchDao extends AbstractSearchDao {

    private DB db;
    private Class typeClass;
    private SearchMapping searchMapping;

    public MongoSearchDao(DB db, SearchMapping searchMapping) {
        this.db = db;
        this.searchMapping = searchMapping;
    }

    @Override
    public void close() {
    }

    @Override
    public Results search(Query query) throws Exception {
        SchemaMapper schemaMapper = searchMapping.schemaMapperForQuery(query);
        MongoQueryConverter converter = new MongoQueryConverter(schemaMapper);
        DBCollection collection = db.getCollection(searchMapping.collectionForQuery(query));

        DBObject mongoQ = null;

        List filters = new ArrayList<>();
        converter.addQ(query, filters);
        converter.addFilters(query, filters);
        if (filters.size() > 1) {

            mongoQ = new BasicDBObject("$and", filters);
        } else if (filters.size() == 1) {
            mongoQ = filters.get(0);
        } else {
            mongoQ = new BasicDBObject();
        }
        DBCursor cursor = collection.find(mongoQ);

        converter.addPaging(query, cursor);
        converter.addSorts(query, cursor);
        Results results = new ResultsConverter(query, cursor, searchMapping).convert();
        return results;
    }

    @Override
    public T findById(K key) throws Exception {
        String id = searchMapping.idForKey(key);
        DBCollection collection = db.getCollection(searchMapping.collectionForKey(key));
        DBObject object = collection.findOne(new BasicDBObject("_id", id));
        T result = searchMapping.fromDocument(id, object);
        return result;
    }

    @Override
    public List saveBatch(List entities) throws Exception {
        List results = new ArrayList<>();
        for (T entity : entities) {
            save(entity);
            results.add(entity);
        }
        return results;
    }

    @Override
    public void delete(T entity) throws Exception {
        DBCollection collection = db.getCollection(searchMapping.collection(entity));

        throw new UnsupportedOperationException();
    }

    @Override
    public void deleteByKey(K id) throws Exception {
        DBCollection collection = db.getCollection(searchMapping.collectionForKey(id));
        collection.remove(new BasicDBObject("_id", id));
    }

    @Override
    public void deleteByQuery(Query query) throws Exception {
        //collection.remove(query.getQ());
    }

    @Override
    public T save(T entity) throws Exception {
        DBCollection collection = db.getCollection(searchMapping.collection(entity));
        DBObject object = searchMapping.toDocument(entity);
        collection.save(object);
        return entity;
    }

}