UpdateTreeSetAction.java

/**
 * Copyright 2008 The University of North Carolina at Chapel Hill
 *
 * 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 edu.unc.lib.dl.data.ingest.solr.action;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import edu.unc.lib.dl.data.ingest.solr.ChildSetRequest;
import edu.unc.lib.dl.data.ingest.solr.SolrUpdateRequest;
import edu.unc.lib.dl.data.ingest.solr.exception.IndexingException;
import edu.unc.lib.dl.fedora.PID;

/**
 * @author bbpennel
 * @date Jun 29, 2015
 */
public class UpdateTreeSetAction extends UpdateTreeAction {
	
	private static final Logger log = LoggerFactory.getLogger(UpdateTreeSetAction.class);

	@Override
	public void performAction(SolrUpdateRequest updateRequest) throws IndexingException {
		if (!(updateRequest instanceof ChildSetRequest))
			throw new IndexingException("ChildSetRequest required to perform TreeSet update, received "
					+ updateRequest.getClass().getName());
		ChildSetRequest childSetRequest = (ChildSetRequest) updateRequest;

		// Calculate total number of objects to be indexed
		int indexTargetTotal = 0;
		for (PID pid : childSetRequest.getChildren()) {
			indexTargetTotal += this.countDescendants(pid) + 1;
		}
		updateRequest.setChildrenPending(indexTargetTotal);

		// Index the tree for each pid in the set
		RecursiveTreeIndexer treeIndexer = new RecursiveTreeIndexer(updateRequest, this, this.addDocumentMode);
		for (PID pid : childSetRequest.getChildren()) {
			treeIndexer.index(pid, null);
		}

		if (log.isDebugEnabled())
			log.debug("Finished updating tree of {}.  {} objects updated in {} ms.", new Object[] {
					updateRequest.getPid().getPid(), updateRequest.getChildrenPending(),
					(System.currentTimeMillis() - updateRequest.getTimeStarted()) });
	}
}