[Java] Thread를 이용하여 MultiChatting 프로그램 작성하기

Server 부분

package Chat;

import java.io.*;
import java.net.*;
import java.util.*;

public class TcpMultiChatServer {

	HashMap clients;
	ServerSocket ss = null; // socket 1
	Socket s = null; // socket 2

	public TcpMultiChatServer() {
		clients = new HashMap(); // 객체생성
		Collections.synchronizedMap(clients); // 동기화
	}

	public void start() {
		try {
			ss = new ServerSocket(8000);
			System.out.println("서버 시작되었습니다. 챗 준비 완료");

			while (true) {
				s = ss.accept(); // 응답대기, 클라이언트 소켓
				System.out.println("[" + s.getInetAddress() + ":" + s.getPort() + "] 에서 접속하셨습니다.");
				ServerReceiver thread = new ServerReceiver(s); // user class
				thread.start(); // run() method 유도된다
			}
//		} catch (Exception e) {  e.printStackTrace(); }
		} catch (Exception e) {
			System.out.println("서버 접속 실패!!");
		}
	} // start() end

	public void sendToAll(String msg) {
		Iterator it = clients.keySet().iterator(); // key 값 출력

		while (it.hasNext()) {
			try {
				DataOutputStream dos = (DataOutputStream) clients.get(it.next());
				dos.writeUTF(msg);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	} // sendToAll( ) end

	class ServerReceiver extends Thread { // inner class
		Socket s; // 클라이언트 소켓
		DataInputStream dis;
		DataOutputStream dos;
		String name = "";

		public ServerReceiver(Socket s) {
			// 생성자함수
			this.s = s;
			try {
				dis = new DataInputStream(s.getInputStream()); // 네트워크를 통해서 읽기
				dos = new DataOutputStream(s.getOutputStream()); // 네트워크를 통해서 쓰기
			} catch (Exception e) {
				System.out.println("서버 리시브 실패");
			}
		} // constructor end

		@Override
		public void run() { //thread 실행부
			try {
				name = dis.readUTF();
				sendToAll("#" + name + " 님이 입장하셨습니다."); // user method
				clients.put(name, dos);
				System.out.println("현재 서버 접속자 수는 : " + clients.size() + "입니다.");
				
				while (dis != null) {
					sendToAll(dis.readUTF());
				}
				
				
			} catch (Exception e) {
				e.printStackTrace();
			} fianlly {
				sendToAll("#" + name + "님이 나가셨습니다.");
				clients.remove(name); // 접속자수 제거
				System.out.println("["+s.getInetAddress()+":"+s.getPort()+"] 에서 접속 종료 하셨습니다.");
				System.out.println("현재 서버 접속자 수는 : " + clients.size() + "입니다.");
			}
		}
	} // end ServerReceiver class

	public static void main(String[] args) {
		new TcpMultiChatServer().start();

	}

}

 

Client 부분

package Chat;

import java.io.*;
import java.net.*;
import java.util.*;

public class TcpMultiChatClient {
	public static void main(String[] args) {
		if (args.length != 1) {
			System.out.println("USAGE : java TcpMultiChatClient nicName ");
			System.exit(0);
		}
		try {
			String ip = "211.200.90.77"; // 127.0.0.1
			Socket s = new Socket(ip, 8000);
			System.out.println("서버에 연결 되었습니다. ");

			Thread send = new Thread(new ClientSender(s, args[0])); // inner class
			Thread reciever = new Thread(new ClientReciever(s)); //

			send.start();
			reciever.start();
		} catch (Exception e) {
			e.printStackTrace();
		}
	} // end main

	static class ClientSender extends Thread { // inner class
		Socket s;
		DataOutputStream dos;
		String name;

		public ClientSender(Socket s, String name) { // 매개변수 2개 생성자함수
			this.s = s;
			try {
				dos = new DataOutputStream(s.getOutputStream());
				this.name = name;
			} catch (Exception e) {
				e.printStackTrace();
			}
		}

		@Override
		public void run() {
			Scanner sc = new Scanner(System.in); // 키보드로부터 입력 받기
			try {
				if (dos != null)
					dos.writeUTF(name);
				while (dos != null)
					dos.writeUTF("[" + name + "] " + sc.nextLine());
			} catch (Exception e) {
				e.printStackTrace();
			}
		} // run() end
	} // end ClientSender

	static class ClientReciever extends Thread { // inner class
		Socket s;
		DataInputStream dis;

		public ClientReciever(Socket s) {
			this.s = s;
			try {
				dis = new DataInputStream(s.getInputStream());
			} catch (Exception e) {
				e.printStackTrace();
			}
		}

		@Override
		public void run() {
			while (dis != null) {
				try {
					System.out.println(dis.readUTF());
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		} // run() end
	} // end ClientReciever

}