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

Go to download

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 javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.AfterBeanDiscovery;
import javax.enterprise.inject.spi.BeanManager;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.ProcessAnnotatedType;
import java.lang.reflect.Modifier;
import java.util.Collection;
import java.util.HashSet;
import java.util.logging.Logger;
import java.util.stream.Stream;

public class CassandraExtension implements Extension {

    private static final Logger LOGGER = Logger.getLogger(CassandraExtension.class.getName());

    private final Collection> crudTypes = new HashSet<>();

    private final Collection> crudAsyncTypes = new HashSet<>();


     void onProcessAnnotatedType(@Observes final ProcessAnnotatedType repo) {
        Class javaClass = repo.getAnnotatedType().getJavaClass();

        if(CassandraRepository.class.equals(javaClass)) {
            return;
        }

        if (Stream.of(javaClass.getInterfaces()).anyMatch(CassandraRepository.class::equals)
                && Modifier.isInterface(javaClass.getModifiers())) {
            crudTypes.add(javaClass);
        }
    }

     void onProcessAnnotatedTypeAsync(@Observes final ProcessAnnotatedType repo) {
        Class javaClass = repo.getAnnotatedType().getJavaClass();

        if(CassandraRepositoryAsync.class.equals(javaClass)) {
            return;
        }

        if (Stream.of(javaClass.getInterfaces()).anyMatch(CassandraRepositoryAsync.class::equals)
                && Modifier.isInterface(javaClass.getModifiers())) {
            crudAsyncTypes.add(javaClass);
        }
    }


    void onAfterBeanDiscovery(@Observes final AfterBeanDiscovery afterBeanDiscovery, final BeanManager beanManager) {
        LOGGER.info("Starting the onAfterBeanDiscovery with elements number: " + crudTypes.size());

        crudTypes.forEach(type -> afterBeanDiscovery.addBean(new CassandraRepositoryBean(type, beanManager)));

        crudAsyncTypes.forEach(type -> afterBeanDiscovery.addBean(new CassandraRepositoryAsyncBean(type, beanManager)));

        LOGGER.info("Finished the onAfterBeanDiscovery");
    }
}