HSEARCH-1492 Logger generation exceptions should be thrown

hibernate · Mar 4, 2014 · 2a87269 · 2a87269
1 parent f337ac7
commit 2a87269
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/engine/src/main/java/org/hibernate/search/query/dsl/impl/MoreLikeThisBuilder.java b/engine/src/main/java/org/hibernate/search/query/dsl/impl/MoreLikeThisBuilder.java
@@ -50,6 +50,7 @@
 import org.apache.lucene.search.ConstantScoreQuery;
 import org.apache.lucene.search.Query;
 import org.apache.lucene.search.TermQuery;
+import org.apache.lucene.search.similarities.Similarity;
 import org.apache.lucene.search.similarities.TFIDFSimilarity;
 import org.apache.lucene.util.BytesRef;
 import org.apache.lucene.util.CharsRef;
@@ -108,9 +109,14 @@ public class MoreLikeThisBuilder<T> {
 	private TermQuery findById;
 
 	public MoreLikeThisBuilder( DocumentBuilderIndexedEntity<T> documentBuilder, SearchFactoryImplementor searchFactory ) {
-		log.requireTFIDFSimilarity( documentBuilder.getBeanClass() );
 		this.documentBuilder = documentBuilder;
-		this.similarity = (TFIDFSimilarity) searchFactory.getIndexBindings().get( documentBuilder.getBeanClass() ).getSimilarity();
+		Similarity configuredSimilarity = searchFactory.getIndexBindings().get( documentBuilder.getBeanClass() ).getSimilarity();
+		if ( configuredSimilarity instanceof TFIDFSimilarity ) {
+			this.similarity = (TFIDFSimilarity) configuredSimilarity;
+		}
+		else {
+			throw log.requireTFIDFSimilarity( documentBuilder.getBeanClass() );
+		}
 	}
 
 	public MoreLikeThisBuilder indexReader(IndexReader indexReader) {