com.netflix.metacat.main.services.search.ElasticSearchClientProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of metacat-main Show documentation

metacat-main

There is a newer version: 1.3.1

/*
 * Copyright 2016 Netflix, Inc.
 *    Licensed 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 com.netflix.metacat.main.services.search;

import com.google.common.base.Splitter;
import com.google.common.base.Strings;
import com.netflix.metacat.common.server.Config;
import org.elasticsearch.client.Client;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.InetSocketTransportAddress;

import javax.inject.Inject;
import javax.inject.Provider;

public class ElasticSearchClientProvider implements Provider {
    private Client client;

    @Inject
    public ElasticSearchClientProvider(Config config) {
        String clusterName = config.getElasticSearchClusterName();
        if( clusterName != null ) {
            Settings settings = ImmutableSettings.settingsBuilder().put("cluster.name", clusterName)
                    .put("transport.tcp.connect_timeout", "60s").build();
            client = new TransportClient(settings);
            // Add the transport address if exists
            String clusterNodesStr = config.getElasticSearchClusterNodes();
            if(!Strings.isNullOrEmpty( clusterNodesStr)){
                Iterable clusterNodes = Splitter.on(',').split(clusterNodesStr);
                clusterNodes.forEach( clusterNode -> ((TransportClient)client).addTransportAddress( new InetSocketTransportAddress( clusterNode, config.getElasticSearchClusterPort())));
            }
        }
    }

    @Override
    public Client get() {
        return client;
    }
}