org.dspace.statistics.export.RetryFailedOpenUrlTrackerScriptConfiguration 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.statistics.export;

import java.sql.SQLException;

import org.apache.commons.cli.Options;
import org.dspace.authorize.service.AuthorizeService;
import org.dspace.core.Context;
import org.dspace.scripts.configuration.ScriptConfiguration;
import org.springframework.beans.factory.annotation.Autowired;

/**
 * The {@link ScriptConfiguration} for the {@link RetryFailedOpenUrlTracker} script
 */
public class RetryFailedOpenUrlTrackerScriptConfiguration
        extends ScriptConfiguration {

    @Autowired
    private AuthorizeService authorizeService;

    private Class dspaceRunnableClass;

    @Override
    public Class getDspaceRunnableClass() {
        return dspaceRunnableClass;
    }

    /**
     * Generic setter for the dspaceRunnableClass
     *
     * @param dspaceRunnableClass The dspaceRunnableClass to be set on this RetryFailedOpenUrlTrackerScriptConfiguration
     */
    @Override
    public void setDspaceRunnableClass(Class dspaceRunnableClass) {
        this.dspaceRunnableClass = dspaceRunnableClass;
    }

    @Override
    public boolean isAllowedToExecute(Context context) {
        try {
            return authorizeService.isAdmin(context);
        } catch (SQLException e) {
            throw new RuntimeException("SQLException occurred when checking if the current user is an admin", e);
        }
    }

    @Override
    public Options getOptions() {
        if (options == null) {
            Options options = new Options();

            options.addOption("a", true, "Add a new \"failed\" row to the table with a url (test purposes only)");
            options.getOption("a").setType(String.class);

            options.addOption("r", false,
                              "Retry sending requests to all urls stored in the table with failed requests. " +
                                      "This includes the url that can be added through the -a option.");
            options.getOption("r").setType(boolean.class);

            options.addOption("h", "help", false, "print this help message");
            options.getOption("h").setType(boolean.class);

            super.options = options;
        }
        return options;
    }

}