/**
* Copyright (c) 2000-present Liferay, Inc. All rights reserved.
*
* This library is free software; you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This library is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*/
package com.liferay.portal.search.elasticsearch.internal;
import com.liferay.portal.kernel.search.AbstractSearchEngineConfigurator;
import com.liferay.portal.kernel.search.IndexSearcher;
import com.liferay.portal.kernel.search.IndexWriter;
import com.liferay.portal.kernel.search.SearchEngine;
import com.liferay.portal.kernel.search.SearchEngineConfigurator;
import com.liferay.portal.kernel.search.SearchEngineHelper;
import com.liferay.portal.kernel.util.MapUtil;
import com.liferay.portal.kernel.util.Validator;
import com.liferay.portal.search.elasticsearch.connection.ElasticsearchConnection;
import com.liferay.portal.search.elasticsearch.connection.ElasticsearchConnectionManager;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import org.osgi.service.component.annotations.Activate;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Reference;
/**
* @author Michael C. Han
*/
@Component(
immediate = true, property = {"search.engine.impl=Elasticsearch"},
service = SearchEngineConfigurator.class
)
public class ElasticsearchEngineConfigurator
extends AbstractSearchEngineConfigurator {
@Override
public void destroy() {
ElasticsearchConnection elasticsearchConnection =
_elasticsearchConnectionManager.getElasticsearchConnection();
elasticsearchConnection.close();
super.destroy();
}
@Activate
protected void activate() {
setSearchEngines(_searchEngines);
}
@Override
protected String getDefaultSearchEngineId() {
return SearchEngineHelper.SYSTEM_ENGINE_ID;
}
@Override
protected IndexSearcher getIndexSearcher() {
return _indexSearcher;
}
@Override
protected IndexWriter getIndexWriter() {
return _indexWriter;
}
@Override
protected ClassLoader getOperatingClassloader() {
Class<?> clazz = getClass();
return clazz.getClassLoader();
}
@Override
protected SearchEngineHelper getSearchEngineHelper() {
return searchEngineHelper;
}
@Override
protected void initialize() {
_elasticsearchConnectionManager.connect();
super.initialize();
}
@Reference(
target = "(&(search.engine.id=SYSTEM_ENGINE)(search.engine.impl=Elasticsearch))"
)
protected void setSearchEngine(
SearchEngine searchEngine, Map<String, Object> properties) {
String searchEngineId = MapUtil.getString(
properties, "search.engine.id");
_searchEngines.put(searchEngineId, searchEngine);
}
protected void unsetSearchEngine(
SearchEngine searchEngine, Map<String, Object> properties) {
String searchEngineId = MapUtil.getString(
properties, "search.engine.id");
if (Validator.isNull(searchEngineId)) {
return;
}
_searchEngines.remove(searchEngineId);
}
@Reference
protected SearchEngineHelper searchEngineHelper;
@Reference
private ElasticsearchConnectionManager _elasticsearchConnectionManager;
@Reference(target = "(!(search.engine.impl=*))")
private IndexSearcher _indexSearcher;
@Reference(target = "(!(search.engine.impl=*))")
private IndexWriter _indexWriter;
private final Map<String, SearchEngine> _searchEngines =
new ConcurrentHashMap<>();
}