org.dspace.authority.indexer.AuthorityIndexClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dspace-api Show documentation
DSpace core data model and service APIs.
There is a newer version: 8.0
/**
 * The contents of this file are subject to the license and copyright
 * detailed in the LICENSE and NOTICE files at the root of the source
 * tree and available online at
 *
 * http://www.dspace.org/license/
 */
package org.dspace.authority.indexer;

import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.logging.log4j.Logger;
import org.dspace.authority.AuthorityValue;
import org.dspace.authority.factory.AuthorityServiceFactory;
import org.dspace.authority.service.AuthorityService;
import org.dspace.content.Item;
import org.dspace.content.factory.ContentServiceFactory;
import org.dspace.content.service.ItemService;
import org.dspace.core.Context;

/**
 * @author Antoine Snyers (antoine at atmire.com)
 * @author Kevin Van de Velde (kevin at atmire dot com)
 * @author Ben Bosman (ben at atmire dot com)
 * @author Mark Diggory (markd at atmire dot com)
 */
public class AuthorityIndexClient {

    private static Logger log = org.apache.logging.log4j.LogManager.getLogger(AuthorityIndexClient.class);

    protected static final AuthorityService authorityService =
        AuthorityServiceFactory.getInstance().getAuthorityService();
    protected static final AuthorityIndexingService indexingService =
        AuthorityServiceFactory.getInstance().getAuthorityIndexingService();
    protected static final List indexers =
        AuthorityServiceFactory.getInstance().getAuthorityIndexers();
    protected static final ItemService itemService =
        ContentServiceFactory.getInstance().getItemService();

    /**
     * Default constructor
     */
    private AuthorityIndexClient() { }

    public static void main(String[] args) throws Exception {

        //Populate our solr
        Context context = new Context();
        //Ensure that we can update items if we are altering our authority control
        context.turnOffAuthorisationSystem();


        if (!authorityService.isConfigurationValid()) {
            //Cannot index, configuration not valid
            System.out.println(
                "Cannot index authority values since the configuration isn't valid. Check dspace logs for more " +
                    "information.");

            return;
        }

        System.out.println("Retrieving all data");
        log.info("Retrieving all data");

        //Get all our values from the input forms
        Map toIndexValues = new HashMap<>();
        for (AuthorityIndexerInterface indexerInterface : indexers) {
            log.info("Initialize " + indexerInterface.getClass().getName());
            System.out.println("Initialize " + indexerInterface.getClass().getName());
            Iterator allItems = itemService.findAll(context);
            Map authorityCache = new HashMap<>();
            while (allItems.hasNext()) {
                Item item = allItems.next();
                List authorityValues = indexerInterface.getAuthorityValues(
                    context, item, authorityCache);
                for (AuthorityValue authorityValue : authorityValues) {
                    toIndexValues.put(authorityValue.getId(), authorityValue);
                }
                context.uncacheEntity(item);
            }
        }


        log.info("Cleaning the old index");
        System.out.println("Cleaning the old index");
        indexingService.cleanIndex();
        log.info("Writing new data");
        System.out.println("Writing new data");
        for (String id : toIndexValues.keySet()) {
            indexingService.indexContent(toIndexValues.get(id));
            indexingService.commit();
        }

        //In the end commit our server
        indexingService.commit();
        context.complete();
        System.out.println("All done !");
        log.info("All done !");
    }
}