All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.eclipse.jnosql.artemis.column.query.AbstractMapperQuery Maven / Gradle / Ivy

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.eclipse.jnosql.artemis.column.query;

import jakarta.nosql.column.Column;
import jakarta.nosql.column.ColumnCondition;
import jakarta.nosql.mapping.Converters;
import jakarta.nosql.mapping.reflection.ClassMapping;
import org.eclipse.jnosql.artemis.util.ConverterUtil;

import java.util.List;
import java.util.stream.StreamSupport;

import static java.util.Arrays.asList;
import static java.util.Objects.nonNull;
import static java.util.Objects.requireNonNull;
import static java.util.stream.Collectors.toList;

abstract class AbstractMapperQuery {

    protected final String columnFamily;

    protected boolean negate;

    protected ColumnCondition condition;

    protected boolean and;

    protected String name;

    protected final ClassMapping mapping;

    protected final Converters converters;

    protected long start;

    protected long limit;


    AbstractMapperQuery(ClassMapping mapping, Converters converters) {
        this.mapping = mapping;
        this.converters = converters;
        this.columnFamily = mapping.getName();
    }

    protected void appendCondition(ColumnCondition newCondition) {
        ColumnCondition columnCondition = getColumnCondition(newCondition);
        if (nonNull(condition)) {
            if (and) {
                this.condition = condition.and(columnCondition);
            } else {
                this.condition = condition.or(columnCondition);
            }
        } else {
            this.condition = columnCondition;
        }
        this.negate = false;
        this.name = null;
    }

    protected  void betweenImpl(T valueA, T valueB) {
        requireNonNull(valueA, "valueA is required");
        requireNonNull(valueB, "valueB is required");
        ColumnCondition newCondition = ColumnCondition
                .between(Column.of(mapping.getColumnField(name), asList(getValue(valueA), getValue(valueB))));
        appendCondition(newCondition);
    }


    protected  void inImpl(Iterable values) {

        requireNonNull(values, "values is required");
        List convertedValues = StreamSupport.stream(values.spliterator(), false)
                .map(this::getValue).collect(toList());
        ColumnCondition newCondition = ColumnCondition
                .in(Column.of(mapping.getColumnField(name), convertedValues));
        appendCondition(newCondition);
    }

    protected  void eqImpl(T value) {
        requireNonNull(value, "value is required");

        ColumnCondition newCondition = ColumnCondition
                .eq(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected void likeImpl(String value) {
        requireNonNull(value, "value is required");
        ColumnCondition newCondition = ColumnCondition
                .like(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected  void gteImpl(T value) {
        requireNonNull(value, "value is required");
        ColumnCondition newCondition = ColumnCondition
                .gte(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected  void gtImpl(T value) {
        requireNonNull(value, "value is required");
        ColumnCondition newCondition = ColumnCondition
                .gt(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected  void ltImpl(T value) {
        requireNonNull(value, "value is required");
        ColumnCondition newCondition = ColumnCondition
                .lt(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected  void lteImpl(T value) {
        requireNonNull(value, "value is required");
        ColumnCondition newCondition = ColumnCondition
                .lte(Column.of(mapping.getColumnField(name), getValue(value)));
        appendCondition(newCondition);
    }

    protected Object getValue(Object value) {
        return ConverterUtil.getValue(value, mapping, name, converters);
    }

    private ColumnCondition getColumnCondition(ColumnCondition newCondition) {
        if (negate) {
            return newCondition.negate();
        } else {
            return newCondition;
        }
    }
}