org.hibernate.search.cfg.impl.HibernateOrmBeanContainerBeanResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hibernate-search-orm Show documentation
Hibernate Search integration with Hibernate Core
The newest version!
/*
 * Hibernate Search, full-text search for your domain model
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.search.cfg.impl;

import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;

import org.hibernate.resource.beans.container.spi.BeanContainer;
import org.hibernate.resource.beans.container.spi.ContainedBean;
import org.hibernate.resource.beans.container.spi.ContainedBeanImplementor;
import org.hibernate.resource.beans.spi.BeanInstanceProducer;
import org.hibernate.search.engine.service.beanresolver.spi.ReflectionBeanResolver;
import org.hibernate.search.engine.service.beanresolver.spi.BeanResolver;
import org.hibernate.search.util.impl.Closer;

/**
 * An {@link BeanResolver} relying on a Hibernate ORM {@link BeanContainer} to resolve beans.
 */
final class HibernateOrmBeanContainerBeanResolver implements BeanResolver {

	private static final BeanContainer.LifecycleOptions LIFECYCLE_OPTIONS = new BeanContainer.LifecycleOptions() {
		@Override
		public boolean canUseCachedReferences() {
			return false;
		}

		@Override
		public boolean useJpaCompliantCreation() {
			return false;
		}
	};

	private final BeanContainer beanContainer;

	private final ConcurrentHashMap beansToCleanup = new ConcurrentHashMap<>();

	private final BeanInstanceProducer fallbackInstanceProducer = new BeanInstanceProducer() {
		private final BeanResolver delegate = new ReflectionBeanResolver();

		@Override
		public  B produceBeanInstance(Class aClass) {
			return delegate.resolve( aClass, aClass );
		}

		@Override
		public  B produceBeanInstance(String s, Class aClass) {
			// Should not be called, because we currently don't use named bean references
			throw new UnsupportedOperationException( "Getting beans by name is not implemented yet" );
		}
	};

	public HibernateOrmBeanContainerBeanResolver(BeanContainer beanContainer) {
		Objects.requireNonNull( beanContainer );
		this.beanContainer = beanContainer;
	}

	@Override
	public  T resolve(Class reference, Class expectedClass) {
		ContainedBean containedBean = beanContainer.getBean( reference, LIFECYCLE_OPTIONS, fallbackInstanceProducer );
		if ( containedBean != null ) {
			if ( containedBean instanceof ContainedBeanImplementor ) {
				beansToCleanup.put( (ContainedBeanImplementor) containedBean, containedBean );
			}
			return expectedClass.cast( containedBean.getBeanInstance() );
		}
		else {
			return null;
		}
	}

	@Override
	public void stop() {
		try ( Closer closer = new Closer<>() ) {
			closer.pushAll( ContainedBeanImplementor::release, beansToCleanup.keySet() );
		}
	}
}