com.github.joschi.nosqlunit.elasticsearch.http.DefaultElasticsearchInsertionStrategy Maven / Gradle / Ivy

Go to download

package com.github.joschi.nosqlunit.elasticsearch.http;

import com.github.joschi.nosqlunit.elasticsearch.http.parser.DataReader;

import java.io.InputStream;
import java.util.Map;

public class DefaultElasticsearchInsertionStrategy implements ElasticsearchInsertionStrategy {
    private final boolean createIndices;
    private final Map indexSettings;
    private final Map> templates;

    public DefaultElasticsearchInsertionStrategy(boolean createIndices,
                                                 Map indexSettings,
                                                 Map> templates) {
        this.createIndices = createIndices;
        this.indexSettings = indexSettings;
        this.templates = templates;
    }

    @Override
    public void insert(ElasticsearchConnectionCallback connection, InputStream dataset) throws Throwable {
        DataReader dataReader = new DataReader(connection.client(), createIndices, indexSettings, templates);
        dataReader.read(dataset);
    }
}