001/* 002 * Copyright 2012-2018 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.autoconfigure.data.elasticsearch; 018 019import java.util.Properties; 020 021import org.elasticsearch.client.Client; 022import org.elasticsearch.client.transport.TransportClient; 023 024import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; 025import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; 026import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; 027import org.springframework.boot.context.properties.EnableConfigurationProperties; 028import org.springframework.context.annotation.Bean; 029import org.springframework.context.annotation.Configuration; 030import org.springframework.data.elasticsearch.client.TransportClientFactoryBean; 031 032/** 033 * {@link org.springframework.boot.autoconfigure.EnableAutoConfiguration 034 * Auto-configuration} for Elasticsearch. 035 * 036 * @author Artur Konczak 037 * @author Mohsin Husen 038 * @author Andy Wilkinson 039 * @since 1.1.0 040 */ 041@Configuration 042@ConditionalOnClass({ Client.class, TransportClientFactoryBean.class }) 043@ConditionalOnProperty(prefix = "spring.data.elasticsearch", name = "cluster-nodes", matchIfMissing = false) 044@EnableConfigurationProperties(ElasticsearchProperties.class) 045public class ElasticsearchAutoConfiguration { 046 047 private final ElasticsearchProperties properties; 048 049 public ElasticsearchAutoConfiguration(ElasticsearchProperties properties) { 050 this.properties = properties; 051 } 052 053 @Bean 054 @ConditionalOnMissingBean 055 public TransportClient elasticsearchClient() throws Exception { 056 TransportClientFactoryBean factory = new TransportClientFactoryBean(); 057 factory.setClusterNodes(this.properties.getClusterNodes()); 058 factory.setProperties(createProperties()); 059 factory.afterPropertiesSet(); 060 return factory.getObject(); 061 } 062 063 private Properties createProperties() { 064 Properties properties = new Properties(); 065 properties.put("cluster.name", this.properties.getClusterName()); 066 properties.putAll(this.properties.getProperties()); 067 return properties; 068 } 069 070}