import java.rmi.*;
import java.rmi.RemoteException;
import java.rmi.Naming;
import java.rmi.server.UnicastRemoteObject;
import java.rmi.registry.LocateRegistry;
import java.rmi.NotBoundException;
import java.rmi.registry.Registry;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.BufferedOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.Set;
import java.lang.Exception;
public class NamingServer extends UnicastRemoteObject implements Registration , Service {
private Map > Replicaloc;
private Map > Replica;
private Set StorageServers;
public NamingServer () throws RemoteException {
Replicaloc = new HashMap >();
Replica = new HashMap > ();
StorageServers = new HashSet ();
}
public synchronized void start(String PORT) throws RemoteException {
Registry r = LocateRegistry.createRegistry(Integer.parseInt(PORT));
r.rebind("NamingServer", new NamingServer()); // bind remote obj with name
}
public String[] register (String IP_STORAGE_SERVER , int PORT_STORAGE_SERVER , String [] files, Storage command_stub ) throws RemoteException, NotBoundException{
StorageServers.add(command_stub); //check if server is active
System.out.println("Storage server : "+ IP_STORAGE_SERVER + " "+PORT_STORAGE_SERVER+ " connected");
for (String file : files) {
if (Replicaloc.get(file) == null ){
List temp = new ArrayList();
temp.add(command_stub);
Replicaloc.put(file , temp);
}
else
Replicaloc.get(file).add(command_stub);
}
if ( Replica.get(command_stub) == null){
List temp = new ArrayList();
temp.add(new String(IP_STORAGE_SERVER));
temp.add(new String(PORT_STORAGE_SERVER + "") );
Replica.put( command_stub, temp);
}
//System.out.println(Replicaloc.keySet());
return new String[2];
}
public boolean createFile(String file ) throws RemoteException, FileNotFoundException{
return true ;
}
public List getStorage(String file ) throws RemoteException, FileNotFoundException , IOException {
Storage lucky_replicaserver = Replicaloc.get(file).get(new Random().nextInt(Replicaloc.get(file).size()));
System.out.println("Lucky Replicaserver " + Replica.get(lucky_replicaserver));
lucky_replicaserver.read(file); // start read thread at server
return Replica.get(lucky_replicaserver);
}
public boolean put(String IP , String PORT , String path )throws Exception {
System.out.println("Senfing file to : ");
//if (Replicaloc.get(path) == null){
System.out.println("File "+ path + "not exist" + " storing file ");
for (Storage stub : StorageServers){
stub.write(IP , PORT , path);
}
return true;
//}
//else
// return false;
}
public List list() throws Exception {
return new ArrayList( Replicaloc.keySet()) ;
}
public static void main (String args[] ) throws RemoteException , NotBoundException , UnknownHostException, IOException {
if (args.length < 2){
System.err.println( "Bad usage: " + "ITS OWN IP Address Port oto use > 1100");
System.exit(1);
}
System.setProperty("java.rmi.server.hostname",args[0]);//ip
new NamingServer().start(args[1]);
System.out.println("\nListening Incoming Connections on :: " + args[1]);
}
}