org.icij.extract.solr.SolrIntersectionConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of extract-cli Show documentation

There is a newer version: 7.4.0

package org.icij.extract.solr;

import java.util.Map;

import java.io.IOException;

import org.apache.solr.common.SolrDocument;
import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.SolrServerException;

/**
 * A consumer that tags the intersection of documents in
 * two cores.
 *
 *
 */
public class SolrIntersectionConsumer extends SolrTaggingConsumer {

	private final SolrClient other;

	public SolrIntersectionConsumer(final SolrClient other,
		final SolrClient destination, final Map literals) {
		super(destination, literals);
		this.other = other;
	}

	@Override
	protected void consume(final SolrDocument input) throws SolrServerException, IOException {
		final String id = (String) input.getFieldValue(idField);

		if (null != other.getById(id)) {
			super.consume(input);
		}
	}
}