mytubermiserver.RMIClient.java Source code

Introduction

Here is the source code for mytubermiserver.RMIClient.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package mytubermiserver;

import com.healthmarketscience.rmiio.RemoteInputStreamClient;
import com.healthmarketscience.rmiio.RemoteOutputStreamClient;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.rmi.NotBoundException;
import java.rmi.RemoteException;
import java.rmi.registry.LocateRegistry;
import java.rmi.registry.Registry;
import mytubermiserver.Server;
import org.apache.commons.io.IOUtils;

/**
 *
 * @author Arthur
 */
public class RMIClient {

    private final Server server;

    public RMIClient(String address, String registryName) throws RemoteException, NotBoundException {
        System.setProperty("com.healthmarketscience.rmiio.exporter.port", "6667");
        Registry registry = LocateRegistry.getRegistry(address);
        server = (Server) registry.lookup(registryName);
    }

    public InputStream download(String fileName) throws IOException {

        InputStream istreamSaida = RemoteInputStreamClient.wrap(server.getFile(fileName));

        int read;
        try (FileOutputStream out = new FileOutputStream("e://WS/" + fileName)) {
            byte[] bytes = new byte[1024];
            while ((read = istreamSaida.read(bytes)) != -1) {
                out.write(bytes, 0, read);
            }
            out.close();
        }
        return new FileInputStream("e://WS/" + fileName);
    }

    public void upload(String fileName, InputStream videoStream)
            throws RemoteException, NotBoundException, IOException {
        try (OutputStream ostream = RemoteOutputStreamClient.wrap(server.uploadOutputStream(fileName))) {
            ostream.write(IOUtils.toByteArray(videoStream));
            ostream.flush();
            ostream.close();
            server.saveInDatabase(fileName);
        }
    }
}