org.eclipse.jnosql.databases.elasticsearch.communication.ElasticsearchDocumentConfiguration Maven / Gradle / Ivy

Go to download
/*
 *  Copyright (c) 2022 Contributors to the Eclipse Foundation
 *   All rights reserved. This program and the accompanying materials
 *   are made available under the terms of the Eclipse Public License v1.0
 *   and Apache License v2.0 which accompanies this distribution.
 *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
 *
 *   You may elect to redistribute this code under either of these licenses.
 *
 *   Contributors:
 *
 *   Otavio Santana
 *   Alessandro Moscatelli
 */
package org.eclipse.jnosql.databases.elasticsearch.communication;


import co.elastic.clients.elasticsearch.ElasticsearchClient;
import co.elastic.clients.json.jsonb.JsonbJsonpMapper;
import co.elastic.clients.transport.rest_client.RestClientTransport;
import org.apache.http.Header;
import org.apache.http.HttpHost;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.eclipse.jnosql.communication.Configurations;
import org.eclipse.jnosql.communication.Settings;
import org.eclipse.jnosql.communication.semistructured.DatabaseConfiguration;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestClientBuilder;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

import static java.util.Arrays.asList;
import static java.util.Objects.requireNonNull;

/**
 * The implementation of {@link DatabaseConfiguration}
 * that returns {@link ElasticsearchDocumentManagerFactory}.
 *
 * @see ElasticsearchConfigurations
 */
public class ElasticsearchDocumentConfiguration implements DatabaseConfiguration {

    private static final int DEFAULT_PORT = 9200;

    private List httpHosts = new ArrayList<>();

    private List headers = new ArrayList<>();


    public ElasticsearchDocumentConfiguration() {

    }

    /**
     * Adds a host in the configuration
     *
     * @param host the host
     * @throws NullPointerException when host is null
     */
    public void add(HttpHost host) {
        this.httpHosts.add(Objects.requireNonNull(host, "host is required"));
    }

    /**
     * Adds a header in the configuration
     *
     * @param header the header
     * @throws NullPointerException when header is null
     */
    public void add(Header header) {
        this.headers.add(Objects.requireNonNull(header, "header is required"));
    }

    @Override
    public ElasticsearchDocumentManagerFactory apply(Settings settings) {
        ElasticsearchClient elasticsearchClient = buildElasticsearchClient(settings);

        return new ElasticsearchDocumentManagerFactory(elasticsearchClient);
    }

    public ElasticsearchClient buildElasticsearchClient(Settings settings) {
        requireNonNull(settings, "settings is required");

        settings.prefixSupplier(asList(ElasticsearchConfigurations.HOST, Configurations.HOST))
                .stream()
                .map(Object::toString)
                .map(h -> ElasticsearchAddress.of(h, DEFAULT_PORT))
                .map(ElasticsearchAddress::toHttpHost)
                .forEach(httpHosts::add);

        RestClientBuilder builder = RestClient.builder(httpHosts.toArray(new HttpHost[0]));
        builder.setDefaultHeaders(headers.stream().toArray(Header[]::new));

        final Optional username = settings
                .getSupplier(asList(Configurations.USER,
                        ElasticsearchConfigurations.USER))
                .map(Object::toString);
        final Optional password = settings
                .getSupplier(asList(Configurations.PASSWORD,
                        ElasticsearchConfigurations.PASSWORD))
                .map(Object::toString);

        if (username.isPresent()) {
            final CredentialsProvider credentialsProvider =
                    new BasicCredentialsProvider();
            credentialsProvider.setCredentials(AuthScope.ANY,
                    new UsernamePasswordCredentials(username.orElse(null), password.orElse(null)));
            builder.setHttpClientConfigCallback(httpClientBuilder -> httpClientBuilder
                    .setDefaultCredentialsProvider(credentialsProvider));
        }

        RestClient httpClient = builder.build();

        var transport = new RestClientTransport(httpClient, new JsonbJsonpMapper());

        var elasticsearchClient = new ElasticsearchClient(transport);
        return elasticsearchClient;
    }

}