JAVA 기반 TCP/IP 다중 채팅 프로그램

IT/통신 프로그래밍

JAVA 기반 TCP/IP 다중 채팅 프로그램

리드론 2017. 10. 28. 00:19

자바(JAVA)로 작성한 콘솔 기반 TCP/IP 다중 채팅 프로그램

프로젝트 구조

소스코드

ChatClient.java

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

public class  ChatClient {
	String ipAddress;
	static final int port=6000;
	Socket client=null;
	BufferedReader read;
	PrintWriter oos;
	BufferedReader ois; 
	String sendData;
	String receiveData;

String user_id;
	ReceiveDataThread rt;
	boolean endflag=false;

public ChatClient(String id, String ip) {
		user_id=id;
		ipAddress=ip;
		try{
			System.out.println("**** 클라이언트*****");
			client = new Socket(ipAddress, port);

read= new BufferedReader(new InputStreamReader(System.in));
			ois = new BufferedReader( new InputStreamReader( client.getInputStream() ) );
			oos = new PrintWriter( client.getOutputStream() );

oos.println( user_id );
			oos.flush();

rt= new ReceiveDataThread(client, ois);
			Thread t = new Thread(rt);
			t.start();

while(true){
				sendData = read.readLine();

oos.println( sendData );
				oos.flush();

if(sendData.equals( "/quit") ) { 
					endflag = true;
					break;
				}
			}
			System.out.print("클라이트의 접속을 종료합니다. ");
			System.exit( 0 );
		} catch(Exception e){  
			if(!endflag) e.printStackTrace();
		}
		finally{
			try{
				ois.close();
				oos.close();
				client.close();   
				System.exit(0);
			}catch(IOException e2){
				e2.printStackTrace();
			}
		}
	}

public static void main(String[] args) {
		if(args.length < 2) {
			System.out.print("USAGE : java  ChatClient  사용자_id  서버_ip");
			System.exit(0);
		}
		new ChatClient(args[0], args[1]);
	}
	
}

ReciveDataThread.java

ChatServer.java

ChatSverThread.java

import java.net.*;

import java.io.*;

import java.util.*;

public class ChatSverThread implements Runnable

{
	Socket child;
	BufferedReader ois;
	PrintWriter oos;

String user_id;
	HashMap<String, PrintWriter> hm;
	InetAddress ip;
	String msg;

public ChatSverThread(Socket s, HashMap<String, PrintWriter> h ) {
		child = s;
		hm = h;

try	{
			ois = new BufferedReader( new InputStreamReader( child.getInputStream() ) );
			oos = new PrintWriter( child.getOutputStream() );

user_id = ois.readLine();
			ip = child.getInetAddress();
			System.out.println( ip + "로부터 " + user_id + "님이 접속하였습니다." );
			broadcast(user_id + "님이 접속하셨습니다.");

synchronized( hm ) {	
				hm.put( user_id, oos );
			}
		}

catch (Exception e ) {
			e.printStackTrace();
		}

}

public void run() {
		String receiveData;

try
		{
			while( (receiveData = ois.readLine()) != null ) {

if( receiveData.equals( "/quit" ) ) {
					synchronized( hm ) {
						hm.remove( user_id );
					}
		
					break;
				}

else if( receiveData.indexOf( "/to" ) >= 0 ) {
					sendMsg( receiveData );
				}

else {
					System.out.println(user_id + " >> " + receiveData );
					broadcast( user_id + " >> " + receiveData );
				}

}

catch (Exception e ) {
			e.printStackTrace();
		}

finally {
			synchronized( hm ) {
				hm.remove( user_id );
			}

broadcast( user_id + "님이 퇴장했습니다." );
			System.out.println( user_id + "님이 퇴장했습니다." );

try
			{
				if( child != null ) {
					ois.close();
					oos.close();
					child.close();
				}
			}

catch ( Exception e) {}
		}
	}

public void broadcast(String message){
		synchronized( hm ) {
			try{
				for( PrintWriter oos : hm.values( )){
				  oos.println( message );
				  oos.flush();
				}
			}catch(Exception e){ }
		}
	}

public void sendMsg(String message){
		int begin = message.indexOf(" ") + 1;
		int end   = message.indexOf(" ", begin);

if(end != -1){
			String id = message.substring(begin, end);
			String msg = message.substring(end+1);
			PrintWriter oos = hm.get(id);

try{
				if(oos != null){
					oos.println( user_id + "님이 다음과 같은 귓속말을 보내셨습니다. : " + msg );
					oos.flush();
				}

}catch(Exception e)
			{ 
				
			}
		}
	}

}

시연영상

저작자표시 변경금지