Java tutorial
/* The contents of this file are subject to the license and copyright terms * detailed in the license directory at the root of the source tree (also * available online at http://fedora-commons.org/license/). */ package fedora.client; import java.awt.Dimension; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.util.HashMap; import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager; import org.apache.commons.httpclient.UsernamePasswordCredentials; import org.apache.log4j.Logger; import fedora.common.Constants; import fedora.server.utilities.StreamUtility; /** * A client to a Fedora server's upload facility, accessed via a * basic-authenticated multipart POST to the server. See * server.management.UploadServlet for protocol details. * * @author Chris Wilper */ public class Uploader { private final MultiThreadedHttpConnectionManager m_cManager = new MultiThreadedHttpConnectionManager(); private final String m_uploadURL; private final UsernamePasswordCredentials m_creds; private final FedoraClient fc; private static final Logger logger = Logger.getLogger(FedoraClient.class.getName()); /** * Construct an uploader to a certain repository as a certain user. */ public Uploader(String host, int port, String context, String user, String pass) throws IOException { m_uploadURL = Administrator.getProtocol() + "://" + host + ":" + port + "/" + context + "/management/upload"; m_creds = new UsernamePasswordCredentials(user, pass); String baseURL = Administrator.getProtocol() + "://" + host + ":" + port + "/" + context; fc = new FedoraClient(baseURL, user, pass); } /** * Construct an uploader to a certain repository as a certain user. */ public Uploader(String protocol, String host, int port, String context, String user, String pass) throws IOException { m_uploadURL = protocol + "://" + host + ":" + port + context + "/" + "/management/upload"; m_creds = new UsernamePasswordCredentials(user, pass); String baseURL = protocol + "://" + host + ":" + port + "/" + context; fc = new FedoraClient(baseURL, user, pass); } /** * Send the data from the stream to the server. This is less efficient than * <i>upload(File)</i>, but if you already have a stream, it's convenient. * This method takes care of temporarily making a File out of the stream, * making the request, and removing the temporary file. Having a File source * for the upload is necessary because the content-length must be sent along * with the request as per the HTTP Multipart POST protocol spec. */ public String upload(InputStream in) throws IOException { File tempFile = File.createTempFile("fedora-upload-", null); FileOutputStream out = new FileOutputStream(tempFile); try { StreamUtility.pipeStream(in, out, 8192); return upload(tempFile); } finally { in.close(); out.close(); if (!tempFile.delete()) { System.err.println("WARNING: Could not remove temporary file: " + tempFile.getName()); tempFile.deleteOnExit(); } } } /** * Send a file to the server, getting back the identifier. */ public String upload(File file) throws IOException { if (Administrator.INSTANCE == null) { return fc.uploadFile(file); } else { // paint initial status to the progress bar String msg = "Uploading " + file.length() + " bytes to " + fc.getUploadURL(); Dimension d = Administrator.PROGRESS.getSize(); Administrator.PROGRESS.setString(msg); Administrator.PROGRESS.setValue(100); Administrator.PROGRESS.paintImmediately(0, 0, (int) d.getWidth() - 1, (int) d.getHeight() - 1); // then start the thread, passing parms in HashMap<String, Object> PARMS = new HashMap<String, Object>(); PARMS.put("fc", fc); PARMS.put("file", file); SwingWorker worker = new SwingWorker(PARMS) { @Override public Object construct() { try { FedoraClient fc = (FedoraClient) parms.get("fc"); File file = (File) parms.get("file"); return fc.uploadFile(file); } catch (IOException e) { thrownException = e; return ""; } } }; worker.start(); // keep updating status till the worker's finished int ms = 200; while (!worker.done) { try { Administrator.PROGRESS.setValue(ms); Administrator.PROGRESS.paintImmediately(0, 0, (int) d.getWidth() - 1, (int) d.getHeight() - 1); Thread.sleep(100); ms = ms + 100; if (ms >= 2000) { ms = 200; } } catch (InterruptedException ie) { } } // reset the status bar to normal Administrator.PROGRESS.setValue(2000); Administrator.PROGRESS.paintImmediately(0, 0, (int) d.getWidth() - 1, (int) d.getHeight() - 1); try { Thread.sleep(100); } catch (InterruptedException ie) { } // report if there was an error; otherwise return the response if (worker.thrownException != null) { throw (IOException) worker.thrownException; } else { return (String) worker.getValue(); } } } /** * Test this class by uploading the given file three times. First, with the * provided credentials, as an InputStream. Second, with the provided * credentials, as a File. Third, with bogus credentials, as a File. */ public static void main(String[] args) { try { if (args.length == 5 || args.length == 6) { String protocol = args[0]; int port = Integer.parseInt(args[1]); String user = args[2]; String password = args[3]; String fileName = args[4]; String context = Constants.FEDORA_DEFAULT_APP_CONTEXT; if (args.length == 6 && !args[5].equals("")) { context = args[5]; } Uploader uploader = new Uploader(protocol, port, context, user, password); File f = new File(fileName); System.out.println(uploader.upload(new FileInputStream(f))); System.out.println(uploader.upload(f)); uploader = new Uploader(protocol, port, context, user + "test", password); System.out.println(uploader.upload(f)); } else { System.err.println("Usage: Uploader host port user password file [context]"); } } catch (Exception e) { System.err.println("ERROR: " + e.getMessage()); } } }