forklift.replay.ReplayESWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of forklift-replay Show documentation
forklift-replay
There is a newer version: 3.8
package forklift.replay;

import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.transport.InetSocketTransportAddress;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.InetSocketAddress;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;

public class ReplayESWriter extends ReplayStoreThread {
    private static final Logger log = LoggerFactory.getLogger(ReplayES.class);

    private final TransportClient client;

    public ReplayESWriter(boolean ssl, String hostname) {
        this(ssl, hostname, 9200);
    }

    public ReplayESWriter(boolean ssl, String hostname, int port) {
        this.client = TransportClient.builder().build()
            .addTransportAddress(new InetSocketTransportAddress(new InetSocketAddress(hostname, port)));
    }

    @Override
    protected void poll(ReplayESWriterMsg t) {
        final String index = "forklift-replay-" + LocalDate.now().format(DateTimeFormatter.BASIC_ISO_DATE);

        // In order to ensure there is only one replay msg for a given id we have to clean the msg id from
        // any previously created indexes.
        try {
            final SearchResponse resp = client.prepareSearch("forklift-replay*").setTypes("log")
                .setQuery(QueryBuilders.termQuery("_id", t.getId()))
                .setFrom(0).setSize(50).setExplain(false)
                .execute()
                .actionGet();

            if (resp != null && resp.getHits() != null && resp.getHits().getHits() != null) {
                for (SearchHit hit : resp.getHits().getHits()) {
                    client.prepareDelete(hit.getIndex(), "log", t.getId()).execute().actionGet();
                }
            }
        } catch (Exception e) {
            log.error("", e);
            log.error("Unable to search for old replay logs {}", t.getId());
        }

        // Index the new information.
        client.prepareIndex(index, "log").setId(t.getId()).setSource(t.getFields()).execute();
    }

    public void shutdown() {
        if (client != null)
            client.close();
    }
}