/*
* Licensed to Elastic Search and Shay Banon under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. Elastic Search licenses this
* file to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.elasticsearch.index.similarity;
import org.apache.lucene.search.Similarity;
import org.elasticsearch.common.Nullable;
import org.elasticsearch.common.collect.ImmutableMap;
import org.elasticsearch.common.inject.Inject;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.index.AbstractIndexComponent;
import org.elasticsearch.index.Index;
import org.elasticsearch.index.settings.IndexSettings;
import java.util.Map;
import static org.elasticsearch.common.collect.Maps.*;
/**
* @author kimchy (Shay Banon)
*/
public class SimilarityService extends AbstractIndexComponent {
private final ImmutableMap<String, SimilarityProvider> similarityProviders;
private final ImmutableMap<String, Similarity> similarities;
public SimilarityService(Index index) {
this(index, ImmutableSettings.Builder.EMPTY_SETTINGS, null);
}
@Inject public SimilarityService(Index index, @IndexSettings Settings indexSettings,
@Nullable Map<String, SimilarityProviderFactory> providerFactories) {
super(index, indexSettings);
Map<String, SimilarityProvider> similarityProviders = newHashMap();
if (providerFactories != null) {
Map<String, Settings> providersSettings = indexSettings.getGroups("index.similarity");
for (Map.Entry<String, SimilarityProviderFactory> entry : providerFactories.entrySet()) {
String similarityName = entry.getKey();
SimilarityProviderFactory similarityProviderFactory = entry.getValue();
Settings similaritySettings = providersSettings.get(similarityName);
if (similaritySettings == null) {
similaritySettings = ImmutableSettings.Builder.EMPTY_SETTINGS;
}
SimilarityProvider similarityProvider = similarityProviderFactory.create(similarityName, similaritySettings);
similarityProviders.put(similarityName, similarityProvider);
}
}
// add defaults
if (!similarityProviders.containsKey("index")) {
similarityProviders.put("index", new DefaultSimilarityProvider(index, indexSettings, "index", ImmutableSettings.Builder.EMPTY_SETTINGS));
}
if (!similarityProviders.containsKey("search")) {
similarityProviders.put("search", new DefaultSimilarityProvider(index, indexSettings, "search", ImmutableSettings.Builder.EMPTY_SETTINGS));
}
this.similarityProviders = ImmutableMap.copyOf(similarityProviders);
Map<String, Similarity> similarities = newHashMap();
for (SimilarityProvider provider : similarityProviders.values()) {
similarities.put(provider.name(), provider.get());
}
this.similarities = ImmutableMap.copyOf(similarities);
}
public Similarity similarity(String name) {
return similarities.get(name);
}
public Similarity defaultIndexSimilarity() {
return similarities.get("index");
}
public Similarity defaultSearchSimilarity() {
return similarities.get("search");
}
}