org.jnosql.artemis.cassandra.column.DefaultCassandraTemplateAsync Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of cassandra-extension Show documentation
Eclipse JNoSQL mapping layer, Artemis, to Cassandra
There is a newer version: 0.0.9
/*
 *  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.cassandra.column;

import com.datastax.driver.core.ConsistencyLevel;
import com.datastax.driver.core.Statement;
import org.jnosql.artemis.Converters;
import org.jnosql.artemis.column.AbstractColumnTemplateAsync;
import org.jnosql.artemis.column.ColumnEntityConverter;
import org.jnosql.artemis.reflection.ClassRepresentations;
import org.jnosql.diana.api.column.ColumnDeleteQuery;
import org.jnosql.diana.api.column.ColumnEntity;
import org.jnosql.diana.api.column.ColumnFamilyManagerAsync;
import org.jnosql.diana.api.column.ColumnQuery;
import org.jnosql.diana.cassandra.column.CassandraColumnFamilyManagerAsync;

import javax.enterprise.inject.Instance;
import javax.enterprise.inject.Typed;
import javax.inject.Inject;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
import java.util.stream.StreamSupport;

import static java.util.Objects.requireNonNull;
import static java.util.stream.Collectors.toList;

/**
 * The CassandraTemplateAsync default implementation
 */
@Typed(CassandraTemplateAsync.class)
class DefaultCassandraTemplateAsync extends AbstractColumnTemplateAsync
        implements CassandraTemplateAsync {


    private CassandraColumnEntityConverter converter;

    private Instance managerAsync;

    private ClassRepresentations classRepresentations;

    private Converters converters;

    DefaultCassandraTemplateAsync() {
    }

    @Inject
    DefaultCassandraTemplateAsync(CassandraColumnEntityConverter converter,
                                  Instance managerAsync,
                                  ClassRepresentations classRepresentations, Converters converters) {
        this.converter = converter;
        this.managerAsync = managerAsync;
        this.classRepresentations = classRepresentations;
        this.converters = converters;
    }

    @Override
    protected ColumnEntityConverter getConverter() {
        return converter;
    }

    @Override
    protected ColumnFamilyManagerAsync getManager() {
        return managerAsync.get();
    }

    @Override
    protected ClassRepresentations getClassRepresentations() {
        return classRepresentations;
    }

    @Override
    protected Converters getConverters() {
        return converters;
    }

    @Override
    public  void save(T entity, ConsistencyLevel level) {
        requireNonNull(entity, "entity is required");
        requireNonNull(level, "level is required");
        managerAsync.get().save(converter.toColumn(entity), level);
    }

    @Override
    public  void save(T entity, ConsistencyLevel level, Consumer callBack) {
        requireNonNull(entity, "entity is required");
        requireNonNull(level, "level is required");
        requireNonNull(callBack, "callBack is required");
        Consumer dianaCallBack = c -> callBack.accept((T) getConverter().toEntity(entity.getClass(), c));
        managerAsync.get().save(converter.toColumn(entity), level, dianaCallBack);
    }

    @Override
    public  void save(T entity, Duration ttl, ConsistencyLevel level) {
        requireNonNull(entity, "entity is required");
        requireNonNull(level, "level is required");
        requireNonNull(ttl, "ttl is required");
        managerAsync.get().save(converter.toColumn(entity), ttl, level);
    }

    @Override
    public  void save(Iterable entities, ConsistencyLevel level) {
        requireNonNull(entities, "entities is required");
        requireNonNull(level, "level is required");
        StreamSupport.stream(entities.spliterator(), false)
                .map(converter::toColumn)
                .forEach(c -> managerAsync.get().save(c, level));
    }

    @Override
    public  void save(Iterable entities, Duration ttl, ConsistencyLevel level) {
        requireNonNull(entities, "entities is required");
        requireNonNull(level, "level is required");
        requireNonNull(ttl, "ttl is required");
        StreamSupport.stream(entities.spliterator(), false)
                .map(converter::toColumn)
                .forEach(c -> managerAsync.get().save(c, ttl, level));
    }

    @Override
    public  void save(T entity, Duration ttl, ConsistencyLevel level, Consumer callBack) {
        requireNonNull(entity, "entity is required");
        requireNonNull(level, "level is required");
        requireNonNull(ttl, "ttl is required");
        requireNonNull(callBack, "callBack is required");
        Consumer dianaCallBack = c -> callBack.accept((T) getConverter().toEntity(entity.getClass(), c));
        managerAsync.get().save(converter.toColumn(entity), ttl, level, dianaCallBack);
    }

    @Override
    public  void delete(ColumnDeleteQuery query, ConsistencyLevel level) {
        managerAsync.get().delete(query, level);
    }

    @Override
    public void delete(ColumnDeleteQuery query, ConsistencyLevel level, Consumer consumer) {
        managerAsync.get().delete(query, level, consumer);
    }

    @Override
    public  void select(ColumnQuery query, ConsistencyLevel level, Consumer> callBack) {
        requireNonNull(callBack, "callBack is required");
        Consumer> dianaCallBack = d -> callBack.accept(
                d.stream()
                        .map(getConverter()::toEntity)
                        .map(o -> (T) o)
                        .collect(toList()));

        managerAsync.get().select(query, level, dianaCallBack);
    }

    @Override
    public  void cql(String query, Consumer> callBack) {
        requireNonNull(callBack, "callBack is required");
        Consumer> dianaCallBack = d -> callBack.accept(
                d.stream()
                        .map(getConverter()::toEntity)
                        .map(o -> (T) o)
                        .collect(toList()));
        managerAsync.get().cql(query, dianaCallBack);
    }

    @Override
    public  void cql(String query, Map values, Consumer> callBack) {
        requireNonNull(callBack, "callBack is required");
        Consumer> dianaCallBack = d -> callBack.accept(
                d.stream()
                        .map(getConverter()::toEntity)
                        .map(o -> (T) o)
                        .collect(toList()));
        managerAsync.get().cql(query, values, dianaCallBack);
    }

    @Override
    public  void cql(String query, Consumer> callBack, Object... params) {
        requireNonNull(query, "callBack is required");
        requireNonNull(callBack, "callBack is required");
        requireNonNull(params, "params is required");
        Consumer> dianaCallBack = d -> callBack.accept(
                d.stream()
                        .map(getConverter()::toEntity)
                        .map(o -> (T) o)
                        .collect(toList()));
        managerAsync.get().nativeQueryPrepare(query).bind(params).executeQueryAsync(dianaCallBack);
    }

    @Override
    public  void execute(Statement statement, Consumer> callBack) {
        requireNonNull(callBack, "callBack is required");
        Consumer> dianaCallBack = d -> callBack.accept(
                d.stream()
                        .map(getConverter()::toEntity)
                        .map(o -> (T) o)
                        .collect(toList()));
        managerAsync.get().execute(statement, dianaCallBack);
    }


}