102 lines
2.8 KiB
Java
102 lines
2.8 KiB
Java
|
package zutil.net.update;
|
||
|
|
|
||
|
|
import java.io.FileInputStream;
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.io.ObjectInputStream;
|
||
|
|
import java.io.ObjectOutputStream;
|
||
|
|
import java.net.Socket;
|
||
|
|
import java.util.logging.Level;
|
||
|
|
import java.util.logging.Logger;
|
||
|
|
|
||
|
|
import zutil.io.MultiPrintStream;
|
||
|
|
import zutil.log.LogUtil;
|
||
|
|
import zutil.net.threaded.ThreadedTCPNetworkServer;
|
||
|
|
import zutil.net.threaded.ThreadedTCPNetworkServerThread;
|
||
|
|
|
||
|
|
public class UpdateServer extends ThreadedTCPNetworkServer{
|
||
|
|
public static final Logger logger = LogUtil.getLogger();
|
||
|
|
|
||
|
|
private FileListMessage fileList;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates a UpdateServer Thread
|
||
|
|
*
|
||
|
|
* @param path The path to sync the clients with
|
||
|
|
*/
|
||
|
|
public UpdateServer(int port, String path) throws Exception{
|
||
|
|
super(port);
|
||
|
|
fileList = new FileListMessage(path);
|
||
|
|
MultiPrintStream.out.dump(fileList);
|
||
|
|
|
||
|
|
this.start();
|
||
|
|
logger.info("Update Server Ready.");
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected ThreadedTCPNetworkServerThread getThreadInstance(Socket s) {
|
||
|
|
return new UpdateServerThread(s);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Handles all the connecting clients
|
||
|
|
*
|
||
|
|
* @author Ziver
|
||
|
|
*/
|
||
|
|
class UpdateServerThread implements ThreadedTCPNetworkServerThread{
|
||
|
|
private ObjectOutputStream out;
|
||
|
|
private ObjectInputStream in;
|
||
|
|
private Socket socket;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates a UpdateServerThread
|
||
|
|
*
|
||
|
|
* @param client is the socket to the client
|
||
|
|
*/
|
||
|
|
public UpdateServerThread(Socket c){
|
||
|
|
socket = c;
|
||
|
|
try {
|
||
|
|
out = new ObjectOutputStream( socket.getOutputStream());
|
||
|
|
in = new ObjectInputStream ( socket.getInputStream() );
|
||
|
|
} catch (IOException e) {
|
||
|
|
logger.log(Level.SEVERE, null, e);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public void run(){
|
||
|
|
try {
|
||
|
|
logger.info("Client["+socket.getInetAddress()+"] connectiong...");
|
||
|
|
// receive the clients file list
|
||
|
|
FileListMessage clientFileList = (FileListMessage)in.readObject();
|
||
|
|
MultiPrintStream.out.dump(clientFileList);
|
||
|
|
FileListMessage diff = fileList.getDiff( clientFileList );
|
||
|
|
MultiPrintStream.out.dump(diff);
|
||
|
|
out.writeObject( diff );
|
||
|
|
|
||
|
|
logger.info("Updating client["+socket.getInetAddress()+"]...");
|
||
|
|
for(FileInfo info : diff.getFileList()){
|
||
|
|
// send file data
|
||
|
|
FileInputStream input = new FileInputStream( info.getFile() );
|
||
|
|
byte[] nextBytes = new byte[ socket.getSendBufferSize() ];
|
||
|
|
int bytesRead = 0;
|
||
|
|
while((bytesRead = input.read(nextBytes)) > 0){
|
||
|
|
out.write(nextBytes,0,bytesRead);
|
||
|
|
}
|
||
|
|
out.flush();
|
||
|
|
input.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
out.flush();
|
||
|
|
socket.close();
|
||
|
|
logger.info("Client["+socket.getInetAddress()+"] update done.");
|
||
|
|
} catch (Exception e) {
|
||
|
|
logger.log(Level.SEVERE, "Update error Client["+socket.getInetAddress()+"].", e);
|
||
|
|
} finally {
|
||
|
|
logger.info("Client["+socket.getInetAddress()+"] disconnected.");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|