org.hibernate.search.query.dsl.impl.ConnectedQueryContextBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hibernate-search-v5migrationhelper-engine Show documentation

Helper to migrate from Hibernate Search 5 to 6, providing partial support for Hibernate Search 5 Engine APIs on top of Hibernate Search 6

The newest version!

/*
 * SPDX-License-Identifier: Apache-2.0
 * Copyright Red Hat Inc. and Hibernate Authors
 */

package org.hibernate.search.query.dsl.impl;

import java.util.HashMap;
import java.util.Map;

import org.hibernate.search.query.dsl.EntityContext;
import org.hibernate.search.query.dsl.QueryBuilder;
import org.hibernate.search.query.dsl.QueryContextBuilder;
import org.hibernate.search.scope.spi.V5MigrationSearchScope;
import org.hibernate.search.spi.SearchIntegrator;

/**
 * Assuming connection with the search factory
 *
 * @author Emmanuel Bernard
 */
public class ConnectedQueryContextBuilder implements QueryContextBuilder {

	private final SearchIntegrator integrator;

	public ConnectedQueryContextBuilder(SearchIntegrator integrator) {
		this.integrator = integrator;
	}

	@Override
	public EntityContext forEntity(Class entityType) {
		return new HSearchEntityContext( entityType );
	}

	public final class HSearchEntityContext implements EntityContext {
		private final V5MigrationSearchScope scope;
		private final Map analyzerOverrides = new HashMap<>();

		public HSearchEntityContext(Class entityType) {
			this.scope = integrator.scope( entityType );
		}

		@Override
		public EntityContext overridesForField(String field, String analyzerName) {
			// Not checking that the analyzer exists here, for the sake of simplicity.
			// If the analyzer does not exist, an exception will be thrown later
			// when the field is actually used in an analysis-sensitive query.
			analyzerOverrides.put( field, analyzerName );
			return this;
		}

		@Override
		public QueryBuilder get() {
			return new ConnectedQueryBuilder( new QueryBuildingContext( integrator, scope, analyzerOverrides ) );
		}
	}
}