fr.ortolang.diffusion.client.cmd.ReindexCommand.java Source code

Introduction

Here is the source code for fr.ortolang.diffusion.client.cmd.ReindexCommand.java

Source

package fr.ortolang.diffusion.client.cmd;

/*
 * #%L
 * ORTOLANG
 * A online network structure for hosting language resources and tools.
 * 
 * Jean-Marie Pierrel / ATILF UMR 7118 - CNRS / Universit de Lorraine
 * Etienne Petitjean / ATILF UMR 7118 - CNRS
 * Jrme Blanchard / ATILF UMR 7118 - CNRS
 * Bertrand Gaiffe / ATILF UMR 7118 - CNRS
 * Cyril Pestel / ATILF UMR 7118 - CNRS
 * Marie Tonnelier / ATILF UMR 7118 - CNRS
 * Ulrike Fleury / ATILF UMR 7118 - CNRS
 * Frdric Pierre / ATILF UMR 7118 - CNRS
 * Cline Moro / ATILF UMR 7118 - CNRS
 *  
 * This work is based on work done in the equipex ORTOLANG (http://www.ortolang.fr/), by several Ortolang contributors (mainly CNRTL and SLDR)
 * ORTOLANG is funded by the French State program "Investissements d'Avenir" ANR-11-EQPX-0032
 * %%
 * Copyright (C) 2013 - 2016 Ortolang Team
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/lgpl-3.0.html>.
 * #L%
 */

import fr.ortolang.diffusion.client.OrtolangClient;
import fr.ortolang.diffusion.client.OrtolangClientException;
import fr.ortolang.diffusion.client.account.OrtolangClientAccountException;
import org.apache.commons.cli.*;

import javax.json.JsonArray;
import javax.json.JsonObject;
import javax.json.JsonString;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public class ReindexCommand extends Command {

    private final Options options = new Options();

    public ReindexCommand() {
        options.addOption("h", "help", false, "show help.");
        options.addOption("U", "username", true, "username for login");
        options.addOption("P", "password", true, "password for login");
        options.addOption("s", "service", true, "service of the object to index");
        options.addOption("t", "type", true, "type of the object to index");
        options.addOption("a", "status", true, "status of the object to index");
        options.addOption("F", "fake", false, "fake mode");
    }

    @Override
    public void execute(String[] args) {
        CommandLineParser parser = new DefaultParser();
        CommandLine cmd;
        try {
            cmd = parser.parse(options, args);
            if (cmd.hasOption("h")) {
                help();
            }

            String[] credentials = getCredentials(cmd);
            String username = credentials[0];
            String password = credentials[1];

            String type = cmd.getOptionValue("t");
            String service = cmd.getOptionValue("s");
            String status = cmd.getOptionValue("a", "PUBLISHED");
            boolean fakeMode = cmd.hasOption("F");

            if (service != null) {

                OrtolangClient client = OrtolangClient.getInstance();
                if (username.length() > 0) {
                    client.getAccountManager().setCredentials(username, password);
                    client.login(username);
                }
                System.out.println("Connected as user: " + client.connectedProfile());
                System.out.println("Retrieving for published objects from service " + service + ", with type "
                        + type + " and with status " + status + " ...");

                List<String> objectKeys = new ArrayList<>();

                int offset = 0;
                int limit = 100;
                JsonObject listOfObjects = client.listObjects(service, type, status, offset, limit);
                JsonArray keys = listOfObjects.getJsonArray("entries");

                while (!keys.isEmpty()) {
                    objectKeys.addAll(keys.getValuesAs(JsonString.class).stream().map(JsonString::getString)
                            .collect(Collectors.toList()));
                    offset += limit;
                    listOfObjects = client.listObjects(service, type, status, offset, limit);
                    keys = listOfObjects.getJsonArray("entries");
                }
                System.out.println("Reindex keys (" + objectKeys.size() + ") : " + objectKeys);
                if (!fakeMode) {
                    for (String key : objectKeys) {
                        client.reindex(key);
                    }
                    System.out.println("All keys reindexed.");
                }
                client.logout();
                client.close();
            } else {
                help();
            }
        } catch (ParseException e) {
            System.out.println("Failed to parse command line properties " + e.getMessage());
            help();
        } catch (OrtolangClientException | OrtolangClientAccountException e) {
            System.out.println("Unexpected error !!");
            e.printStackTrace();
        }
    }

    private void help() {
        HelpFormatter formatter = new HelpFormatter();
        formatter.printHelp("Reindex", options);
        System.exit(0);
    }

}