org.netarchivesuite.heritrix3wrapper.CommandLauncherTest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of heritrix3-wrapper Show documentation

Wrapper library for unpacking and communicating with Heritrix 3.

There is a newer version: 1.0.3

package org.netarchivesuite.heritrix3wrapper;

import java.io.File;
import java.io.IOException;
import java.net.URL;

import org.apache.http.client.ClientProtocolException;

public class CommandLauncherTest {

    protected static ClassLoader clsLdr = Heritrix3WrapperTest.class.getClassLoader();

    public static final File getTestResourceFile(String fname) {
        URL url = clsLdr.getResource(fname);
        String path = url.getFile();
        path = path.replaceAll("%5b", "[");
        path = path.replaceAll("%5d", "]");
        File file = new File(path);
        return file;
    }

    public static void main(String[] args) {
        String basedirStr = "/home/nicl/heritrix-3.2.0-unzip/heritrix-3.2.0/";
        String[] cmd = {
                "./bin/heritrix",
                "-b 192.168.1.101",
                "-p 6443",
                "-a h3server:h3server",
                "-s h3server.jks,h3server,h3server"
        };
        CommandLauncher h3launcher;
        Heritrix3Wrapper h3wrapper;
        EngineResult engineResult;
        JobResult jobResult;
        try {
            h3launcher = CommandLauncher.getInstance();

            File basedir = new File(basedirStr);
            h3launcher.init(basedir, cmd);
            h3launcher.env.put("FOREGROUND", "true");
            h3launcher.start(new ConsoleLaunchResultHandler());
            try {
                Thread.sleep(5 * 1000);
            } catch (InterruptedException e) {
            }

            File ksFile = getTestResourceFile("h3client.jks");
            h3wrapper = Heritrix3Wrapper.getInstance("192.168.1.101", 6443, ksFile, "h3client", "h3server", "h3server");

            engineResult = h3wrapper.rescanJobDirectory();
            System.out.println(new String(engineResult.response, "UTF-8"));

            try {
                Thread.sleep(30 * 1000);
            } catch (InterruptedException e) {
            }

            engineResult = h3wrapper.exitJavaProcess(null);

            h3launcher.process.destroy();
        } catch (ClientProtocolException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}