org.jnosql.diana.mongodb.document.DocumentQueryConversor.java Source code

Introduction

Here is the source code for org.jnosql.diana.mongodb.document.DocumentQueryConversor.java

Source

/*
 *  Copyright (c) 2017 Otvio Santana and others
 *   All rights reserved. This program and the accompanying materials
 *   are made available under the terms of the Eclipse Public License v1.0
 *   and Apache License v2.0 which accompanies this distribution.
 *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
 *
 *   You may elect to redistribute this code under either of these licenses.
 *
 *   Contributors:
 *
 *   Otavio Santana
 */

package org.jnosql.diana.mongodb.document;

import com.mongodb.BasicDBObject;
import com.mongodb.client.model.Filters;
import org.bson.conversions.Bson;
import org.jnosql.diana.api.TypeReference;
import org.jnosql.diana.api.document.Document;
import org.jnosql.diana.api.document.DocumentCondition;

import java.util.List;

final class DocumentQueryConversor {

    private DocumentQueryConversor() {
    }

    public static Bson convert(DocumentCondition condition) {
        Document document = condition.getDocument();
        Object value = document.getValue().get();
        switch (condition.getCondition()) {
        case EQUALS:
            return Filters.eq(document.getName(), value);
        case GREATER_THAN:
            return Filters.gt(document.getName(), value);
        case GREATER_EQUALS_THAN:
            return Filters.gte(document.getName(), value);
        case LESSER_THAN:
            return Filters.lt(document.getName(), value);
        case LESSER_EQUALS_THAN:
            return Filters.lte(document.getName(), value);
        case IN:
            return Filters.in(document.getName(), value);
        case LIKE:
            return Filters.regex(document.getName(), value.toString());
        case AND:
            List<Document> andList = condition.getDocument().getValue().get(new TypeReference<List<Document>>() {
            });
            return Filters.and(andList.stream().map(d -> new BasicDBObject(d.getName(), d.getValue().get()))
                    .toArray(BasicDBObject[]::new));
        case OR:
            List<Document> orList = condition.getDocument().getValue().get(new TypeReference<List<Document>>() {
            });
            return Filters.or(orList.stream().map(d -> new BasicDBObject(d.getName(), d.getValue().get()))
                    .toArray(BasicDBObject[]::new));
        default:
            throw new UnsupportedOperationException(
                    "The condition " + condition.getCondition() + " is not supported from mongoDB diana driver");
        }
    }
}