connectiontoserver.java

chat room

package Client;
import ShareData.Message;
import java.io.*;
import java.net.*;


public class ConnectionToServer implements Runnable {

	private Socket socket;
	private String ip="127.0.0.1";
	private int port=9999;
	private boolean quit=false;
	private IClient ic;
	
	public ConnectionToServer(IClient i){
		this("127.0.0.1",9999,i);
	}
	
	public ConnectionToServer(String s,int p,IClient i){
		ip=s;
		port=p;
		ic=i;
	}
	
	@Override
	public void run() {
		connect();
		while(!quit){
			Message m=receiveFromServer();
			ic.updateGUI(m);
		}
		try {
			socket.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	public void connect(){
		
		try {
			InetAddress add=InetAddress.getByName(ip);
			socket=new Socket(add,port);
			System.out.println("!!Connect Build!!!");
		} catch (UnknownHostException e) {
			System.out.println("The server is not exist!");
			e.printStackTrace();
		} catch (Exception e) {
			System.out.println("Cannot connect to server");
			e.printStackTrace();
		} 
		
	}
	
	public void quit(String name){
			Message m=new Message("quit",name);
			sendToServer(m);
			quit=true;
		
	}
	
	public void sendToServer(Message message){
		try {
			ObjectOutputStream oos=new ObjectOutputStream(socket.getOutputStream());
			oos.writeObject(message);
			
		} catch (IOException e) {
			System.out.println("Fail to write");
			e.printStackTrace();
		}
	}
	
	public Message receiveFromServer(){
		
		try {
			ObjectInputStream ois=new ObjectInputStream(socket.getInputStream());
			Message m=(Message)ois.readObject();
			return m;
		}catch(IOException e){
			// TODO Auto-generated catch block
			System.out.println("Fail to Receive Messages!");
			e.printStackTrace();
		} catch (ClassNotFoundException e) {
			// TODO Auto-generated catch block
			System.out.println("Fail to Receive and conncect!");
			e.printStackTrace();
		}
		return null;
		
	}
}