org.jnosql.artemis.document.spi.RepositoryUnitDocumentBean Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artemis-document Show documentation
Eclipse JNoSQL Mapping, Artemis API, to document NoSQL databases
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.document.spi;

import org.jnosql.artemis.Repository;
import org.jnosql.artemis.document.DocumentRepositoryProducer;
import org.jnosql.artemis.spi.AbstractBean;
import org.jnosql.artemis.util.RepositoryUnit;
import org.jnosql.diana.api.document.DocumentCollectionManager;
import org.jnosql.diana.api.document.DocumentCollectionManagerFactory;

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 RepositoryUnitDocumentBean extends AbstractBean> {

    private final Set types;

    private final Set qualifiers;

    private final RepositoryUnit repositoryUnit;

    RepositoryUnitDocumentBean(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() {
        DocumentRepositoryProducer producer = getInstance(DocumentRepositoryProducer.class);
        DocumentConfigurationProducer configurationProducer = getInstance(DocumentConfigurationProducer.class);
        Class repository  = (Class) repositoryUnit.getRepository();
        DocumentCollectionManagerFactory managerFactory = configurationProducer.getFactory(repositoryUnit.getUnit());
        DocumentCollectionManager manager = managerFactory.get(repositoryUnit.getDatabase());
        return producer.get(repository, manager);
    }

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

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

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