org.jnosql.artemis.graph.query.SelectQueryConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artemis-graph Show documentation
Eclipse JNoSQL mapping layer, Artemis, to Graph API
The newest version!
/*
 *  Copyright (c) 2017 Otávio 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.artemis.graph.query;

import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.jnosql.aphrodite.antlr.method.SelectMethodFactory;
import org.jnosql.artemis.Pagination;
import org.jnosql.artemis.reflection.ClassMapping;
import org.jnosql.artemis.reflection.DynamicReturn;
import org.jnosql.query.SelectQuery;
import org.jnosql.query.Sort;

import java.util.List;
import java.util.function.BiFunction;
import java.util.function.Consumer;

import static org.apache.tinkerpop.gremlin.process.traversal.Order.asc;
import static org.apache.tinkerpop.gremlin.process.traversal.Order.desc;

final class SelectQueryConverter extends AbstractQueryConvert implements BiFunction> {


    @Override
    public List apply(GraphQueryMethod graphQuery, Object[] params) {

        SelectMethodFactory selectMethodFactory = SelectMethodFactory.get();
        SelectQuery query = selectMethodFactory.apply(graphQuery.getMethod(), graphQuery.getEntityName());
        ClassMapping mapping = graphQuery.getMapping();

        GraphTraversal traversal = getGraphTraversal(graphQuery, query::getWhere, mapping);

        query.getOrderBy().forEach(getSort(traversal, mapping));
        setSort(params, traversal);
        setPagination(params, traversal, query);
        traversal.hasLabel(mapping.getName());
        return traversal.toList();
    }


    private Consumer getSort(GraphTraversal traversal, ClassMapping mapping) {
        return o -> {
            if (Sort.SortType.ASC.equals(o.getType())) {
                traversal.order().by(mapping.getColumnField(o.getName()), asc);
            } else {
                traversal.order().by(mapping.getColumnField(o.getName()), desc);
            }
        };
    }


    static void setPagination(Object[] args, GraphTraversal traversal) {
        setPagination(args, traversal, null);
    }


    private static void setPagination(Object[] args, GraphTraversal traversal, SelectQuery query) {
        Pagination pagination = DynamicReturn.findPagination(args);
        if (pagination != null) {
            traversal.skip(pagination.getSkip())
                    .limit(pagination.getLimit());
            return;
        }

        if (query != null) {
            if (query.getSkip() > 0) {
                traversal.skip(query.getSkip());
            }

            if (query.getLimit() > 0) {
                traversal.next((int) query.getLimit());
            }
        }

    }

    static void setSort(Object[] args, GraphTraversal traversal) {
        List sorts = DynamicReturn.findSorts(args);
        if (!sorts.isEmpty()) {
            for (org.jnosql.diana.api.Sort sort : sorts) {
                traversal.order().by(sort.getName(), sort.getType() == org.jnosql.diana.api.Sort.SortType.ASC ? asc : desc);
            }
        }
    }

}