자바채팅프로그래밍 -다중채팅

//서버

// STEP 3
// 로그온 메시지와 대화말 메시지를 전송함
import java.io.*;
import java.net.*;
import java.util.*;
import java.util.List;
import java.awt.*;
import java.awt.event.*;

public class ChatMessageS extends Frame {
   TextArea display;
   Label info;
   List<ServerThread> list;
	
   public ServerThread SThread;
	
   public ChatMessageS() {
      super("서버");
      info = new Label();
      add(info, BorderLayout.CENTER);
      display = new TextArea("", 0, 0, TextArea.SCROLLBARS_VERTICAL_ONLY);
      display.setEditable(false);
      add(display, BorderLayout.SOUTH);
      addWindowListener(new WinListener());
      setSize(300,250);
      setVisible(true);
   }
	
   public void runServer() {
      ServerSocket server;
      Socket sock;
      //ServerThread SThread;
      try {
         list = new ArrayList<ServerThread>();
         server = new ServerSocket(5000, 100);
         try {
            while(true) {
               sock = server.accept();
 //              SThread = new ServerThread(this, sock, display, info);
               SThread = new ServerThread(this, sock, display);
               SThread.start();
               info.setText(sock.getInetAddress().getHostName() + " 서버는 클라이언트와 연결됨");
            }
         } catch(IOException ioe) {
            server.close();
            ioe.printStackTrace();
         }
      } catch(IOException ioe) {
         ioe.printStackTrace();
      }
   }
		
   public static void main(String args[]) {
      ChatMessageS s = new ChatMessageS();
      s.runServer();
   }
		
   class WinListener extends WindowAdapter {
      public void windowClosing(WindowEvent e) {
    	  
         System.exit(0);
      }
   }
}

class ServerThread extends Thread {
   Socket sock;
   BufferedWriter output;
   BufferedReader input;
   TextArea display;
   Label info;
   TextField text;
   String clientdata;
//   String serverdata = "";
   ChatMessageS cs;
	
   private static final String SEPARATOR = "|";
   private static final int REQ_LOGON = 1001;
   private static final int REQ_SENDWORDS = 1021;
   private static final int LOGOUT = 1004;
	
//   public ServerThread(ChatMessageS c, Socket s, TextArea ta, Label l) {
   public ServerThread(ChatMessageS c, Socket s, TextArea ta) {
      sock = s;
      display = ta;
 //     info = l;
      cs = c;
      try {
         input = new BufferedReader(new InputStreamReader(sock.getInputStream()));
         output = new BufferedWriter(new OutputStreamWriter(sock.getOutputStream()));
      } catch(IOException ioe) {
         ioe.printStackTrace();
      }
   }
   public void run() {
    //  cs.list.add(this);
      try {
         while((clientdata = input.readLine()) != null) {
            StringTokenizer st = new StringTokenizer(clientdata, SEPARATOR);
            int command = Integer.parseInt(st.nextToken());
            int cnt = cs.list.size();
            switch(command) {             //-->로그온 한 애들만 메세지를 보냄
               case REQ_LOGON : { // “1001|아이디”를 수신한 경우
            	   cs.list.add(this);    
                  String ID = st.nextToken();
                  String logms = st.nextToken();
                  display.append("클라이언트가 " + ID);
                  display.append(logms+"\r\n");
                  break;
               }
               case REQ_SENDWORDS : { // “1021|아이디|대화말”를 수신
                  String ID = st.nextToken();
                  String message = st.nextToken();
                  display.append(ID + " : " + message + "\r\n");
                  for(int i=0; i<cnt; i++) { // 모든 클라이언트에 전송
                     ServerThread SThread = (ServerThread)cs.list.get(i);
                     SThread.output.write(ID + " : " + message + "\r\n");
                     SThread.output.flush();
                  }  
                  break;
               }
               case LOGOUT : {  // 1004|아이디 로그아웃
            	   cs.list.remove(this);     //로그아웃시 소켓 해제
            	   String ID = st.nextToken();
                   String logms = st.nextToken();
                   display.append("클라이언트가 " + ID);
                   display.append(logms+"\r\n");
                   break; 	   
               }
            }
         }
      } catch(IOException e) {
         e.printStackTrace();
      }
      cs.list.remove(this);
      try{
         sock.close();
      }catch(IOException ea){
         ea.printStackTrace();
      }
   }
}

 

//클라이언트

// step3
// 아이디와 대화말 메시지를 전송함
import java.io.*;
import java.net.*;
import java.util.StringTokenizer;
import java.awt.*;
import java.awt.event.*;

public class ChatMessageC extends Frame implements ActionListener, KeyListener {
	
   TextArea display;
   TextField wtext, ltext;
   Label mlbl, wlbl, loglbl;
   BufferedWriter output;
   BufferedReader input;
   Socket client;
   StringBuffer clientdata = new StringBuffer(2048);
   String serverdata;
   String ID,logm;
   private Button logout;
	
   private static final String SEPARATOR = "|";
   private static final int REQ_LOGON = 1001;
   private static final int REQ_SENDWORDS = 1021;
   private static final int REQ_LOGOUT = 1004;
	
   public ChatMessageC() {
      super("클라이언트");
      
      Panel northp =  new Panel(new BorderLayout());
      mlbl = new Label("채팅 상태를 보여줍니다.");
      logout = new Button("로그아웃");
      logout.setVisible(false);
      logout.addActionListener(this);
      northp.add(mlbl,BorderLayout.WEST);
      northp.add(logout,BorderLayout.EAST);
      add(northp, BorderLayout.NORTH);

      display = new TextArea("", 0, 0, TextArea.SCROLLBARS_VERTICAL_ONLY);
      display.setEditable(false);
      
      add(display, BorderLayout.CENTER);

      Panel ptotal = new Panel(new BorderLayout());
 
      Panel pword = new Panel(new BorderLayout());
      wlbl = new Label("대화말");
      wtext = new TextField(30); //전송할 데이터를 입력하는 필드
      wtext.addKeyListener(this); //입력된 데이터를 송신하기 위한 이벤트 연결
      pword.add(wlbl, BorderLayout.WEST);
      pword.add(wtext, BorderLayout.EAST);
      ptotal.add(pword, BorderLayout.CENTER);
      

      Panel plabel = new Panel(new BorderLayout());
      loglbl = new Label("로그온");
      ltext = new TextField(30); //전송할 데이터를 입력하는 필드
      ltext.addActionListener(this); //입력된 데이터를 송신하기 위한 이벤트 연결
      plabel.add(loglbl, BorderLayout.WEST);
      plabel.add(ltext, BorderLayout.EAST);
      ptotal.add(plabel, BorderLayout.SOUTH);

      add(ptotal, BorderLayout.SOUTH);

      addWindowListener(new WinListener());
      setSize(300,250);
      setVisible(true);
   }
	
   public void runClient() {
      try {
         client = new Socket(InetAddress.getLocalHost(), 5000);
         
         mlbl.setText("연결된 서버이름 : " + client.getInetAddress().getHostName());
         input = new BufferedReader(new InputStreamReader(client.getInputStream()));
         output = new BufferedWriter(new OutputStreamWriter(client.getOutputStream()));
         //clientdata = new StringBuffer(2048);
         mlbl.setText("접속 완료 사용할 아이디를 입력하세요.");
         while(true) {
            serverdata = input.readLine();
            display.append(serverdata+"\r\n");
         }
      } catch(IOException e) {
         e.printStackTrace();
      }
   }
		
   public void actionPerformed(ActionEvent ae){
	  ID = ltext.getText();
	  
	  if(ae.getSource() == ltext){
		  if(ID.equals("") != true) {
			  logout.setVisible(true);
			  mlbl.setText(ID + "(으)로 로그인 하였습니다.");
			  try {
				  clientdata.setLength(0);
				  clientdata.append(REQ_LOGON);
				  clientdata.append(SEPARATOR);
				  clientdata.append(ID);
				  clientdata.append(SEPARATOR);
				  clientdata.append("(으)로 로그인 하였습니다.");
				  output.write(clientdata.toString()+"\r\n");
				  output.flush();
				  ltext.setVisible(false);
			  } catch(Exception e) {
				  e.printStackTrace();
			  }
		  } else {
    	  mlbl.setText("다시 로그인 하세요!!!");
		  }
	  }

	  else if(ae.getSource()==logout) {
    	  mlbl.setText("채팅 상태를 보여줍니다.");
    	  logout.setVisible(false);
    	  try {
    	  ltext.setText("");
    	  clientdata.setLength(0);
          clientdata.append(REQ_LOGOUT);
          clientdata.append(SEPARATOR);
          clientdata.append(ID);
          clientdata.append(SEPARATOR);
          clientdata.append("이(가) 로그아웃하였습니다.");
          output.write(clientdata.toString() +"\r\n");
          output.flush();
    	  ltext.setVisible(true);
    	  }catch(Exception e) {
    		  e.printStackTrace();
    	  }
	  }
   }
	
   public static void main(String args[]) {
      ChatMessageC c = new ChatMessageC();
      c.runClient();
   }
		
   class WinListener extends WindowAdapter {        //창은 닫히는데 로그아웃을 한 상태를 만들어줘야함.
      public void windowClosing(WindowEvent e){
    	  try {
    		  ltext.setText("");
        	  clientdata.setLength(0);
              clientdata.append(REQ_LOGOUT);
              clientdata.append(SEPARATOR);
              clientdata.append(ID);
              clientdata.append(SEPARATOR);
              clientdata.append("이(가) 로그아웃하였습니다.");
              output.write(clientdata.toString() +"\r\n");
              output.flush();
        	  ltext.setVisible(true);
        	  }catch(Exception e2) {
        		  e2.printStackTrace();
        	  }
         System.exit(0);
      }
   }

   public void keyPressed(KeyEvent ke) {
      if(ke.getKeyChar() == KeyEvent.VK_ENTER) {
         String message = new String();
         message = wtext.getText();
         if (ID == null) {
            mlbl.setText("다시 로그인 하세요!!!");
            wtext.setText("");
         } else {
            try {
               clientdata.setLength(0);
               clientdata.append(REQ_SENDWORDS);
               clientdata.append(SEPARATOR);
               clientdata.append(ID);
               clientdata.append(SEPARATOR);
               clientdata.append(message);
               output.write(clientdata.toString()+"\r\n");
               output.flush();
               wtext.setText("");
            } catch (IOException e) {
               e.printStackTrace();
            }
         }
      }
   }

   public void keyReleased(KeyEvent ke) {
   }

   public void keyTyped(KeyEvent ke) {
   }
}

 

//만약 창을 강제종료 한다면