org.jnosql.artemis.graph.spi.RepositoryUnitGraphBean 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) 2019 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.spi;

import org.apache.tinkerpop.gremlin.structure.Graph;
import org.jnosql.artemis.Repository;
import org.jnosql.artemis.graph.GraphRepositoryProducer;
import org.jnosql.artemis.spi.AbstractBean;
import org.jnosql.artemis.util.RepositoryUnit;

import javax.enterprise.context.spi.CreationalContext;
import javax.enterprise.inject.spi.BeanManager;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.Collections;
import java.util.Set;

class RepositoryUnitGraphBean  extends AbstractBean> {

    private final Set types;

    private final Set qualifiers;

    private final RepositoryUnit repositoryUnit;

    RepositoryUnitGraphBean(BeanManager beanManager, RepositoryUnit repositoryUnit) {
        super(beanManager);
        this.types = Collections.singleton(repositoryUnit.getRepository());
        this.qualifiers = Collections.singleton(repositoryUnit.getUnit());
        this.repositoryUnit = repositoryUnit;
    }


    @Override
    public Class getBeanClass() {
        return repositoryUnit.getRepository();
    }

    @Override
    public Repository create(CreationalContext> context) {
        return get();
    }

    private > R get() {
        GraphRepositoryProducer producer = getInstance(GraphRepositoryProducer.class);
        GraphConfigurationProducer configurationProducer = getInstance(GraphConfigurationProducer.class);
        Class repository  = (Class) repositoryUnit.getRepository();
        Graph graph = configurationProducer.getGraph(repositoryUnit.getUnit());
        return producer.get(repository, graph);
    }

    @Override
    public Set getTypes() {
        return types;
    }

    @Override
    public Set getQualifiers() {
        return qualifiers;
    }

    @Override
    public String getId() {
        return "graph: " + repositoryUnit.toString();
    }
}