package server;

import java.io.*; 
import java.net.*; 
import java.util.HashMap; 
import java.util.Map; 
import java.util.concurrent.ExecutorService; 
import java.util.concurrent.Executors; 
import java.util.concurrent.ThreadPoolExecutor; 
import javax.swing.*;
import java.awt.*;

public class TCPServer extends JFrame{  

 private JTextArea m_display=new JTextArea();

 private ServerSocket serverSocket; 

 /** 
 * 創(chuàng)建線程池來管理客戶端的連接線程 
 * 避免系統(tǒng)資源過度浪費(fèi) 
 */ 
 private ExecutorService exec; 

 // 存放客戶端之間私聊的信息 
 private Map<String,PrintWriter> storeInfo; 

 public TCPServer() { 
  super("聊天程序服務(wù)器端");
  Container c=getContentPane();
  c.add(new JScrollPane(m_display),BorderLayout.CENTER);
  try { 

   serverSocket = new ServerSocket(6666); 
   storeInfo = new HashMap<String, PrintWriter>(); 
   exec = Executors.newCachedThreadPool(); 

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

 // 將客戶端的信息以Map形式存入集合中 
 private void putIn(String key,PrintWriter value) { 
  synchronized(this) { 
   storeInfo.put(key, value); 
  } 
 } 

 // 將給定的輸出流從共享集合中刪除 
 private synchronized void remove(String key) { 
  storeInfo.remove(key); 
  m_display.append("當(dāng)前在線人數(shù)為："+ storeInfo.size());
  //for(String name: storeInfo.key)
 } 

 // 將給定的消息轉(zhuǎn)發(fā)給所有客戶端 
 private synchronized void sendToAll(String message) { 
  for(PrintWriter out: storeInfo.values()) { 
   out.println(message); 


   // m_display.append("已經(jīng)發(fā)送了");
  } 
 } 

 // 將給定的消息轉(zhuǎn)發(fā)給私聊的客戶端 
 private synchronized void sendToSomeone(String name,String message) { 
  PrintWriter pw = storeInfo.get(name); //將對應(yīng)客戶端的聊天信息取出作為私聊內(nèi)容發(fā)送出去 
  if(pw != null) pw.println("私聊:  "+message); 
 } 

 public void start() { 
  try { 
   m_display.setVisible(true);
   //m_display.append("mayanshuo");
   while(true) { 

   m_display.append("等待客戶端連接... ... \n"); 

   Socket socket = serverSocket.accept(); 

   // 獲取客戶端的ip地址 
   InetAddress address = socket.getInetAddress(); 
   m_display.append("客戶端：“" + address.getHostAddress() + "”連接成功！ "); 
   /* 
   * 啟動一個線程，由線程來處理客戶端的請求，這樣可以再次監(jiān)聽 
   * 下一個客戶端的連接 
   */ 
   exec.execute(new ListenrClient(socket)); //通過線程池來分配線程 
   } 
  } catch(Exception e) { 
   e.printStackTrace(); 
  } 
 } 

 /** 
 * 該線程體用來處理給定的某一個客戶端的消息，循環(huán)接收客戶端發(fā)送 
 * 的每一個字符串，并輸出到控制臺 
 */ 
 class ListenrClient implements Runnable { 

  private Socket socket; 
  private String name; 

  public ListenrClient(Socket socket) { 
   this.socket = socket; 
  } 

  // 創(chuàng)建內(nèi)部類來獲取昵稱 
  private String getName() throws Exception { 
   try { 
    //服務(wù)端的輸入流讀取客戶端發(fā)送來的昵稱輸出流 
    BufferedReader bReader = new BufferedReader( 
     new InputStreamReader(socket.getInputStream(), "UTF-8")); 
    //服務(wù)端將昵稱驗(yàn)證結(jié)果通過自身的輸出流發(fā)送給客戶端 
    PrintWriter ipw = new PrintWriter( 
     new OutputStreamWriter(socket.getOutputStream(), "UTF-8"),true); 

    //讀取客戶端發(fā)來的昵稱 
    while(true) { 
     String nameString = bReader.readLine(); 
     if ((nameString.trim().length() == 0) || storeInfo.containsKey(nameString)) { 
      ipw.println("FAIL"); 
     } else { 
      ipw.println("OK"); 
      return nameString; 
     } 
    } 
   } catch(Exception e) { 
    throw e; 
   } 
  } 

  @Override   
  public void run() { 
   try { 
    /* 
    * 通過服務(wù)器端的socket分配給每一個 
    * 用來將消息發(fā)送給客戶端 
    */ 
    PrintWriter pw = new PrintWriter( 
     new OutputStreamWriter(socket.getOutputStream(), "UTF-8"), true); 

    /* 
    * 將客戶昵稱和其所說的內(nèi)容存入共享集合HashMap中 
    */ 
    name = getName(); 
    putIn(name, pw); 
    Thread.sleep(100); 

    // 服務(wù)端通知所有客戶端，某用戶上線 
    sendToAll("*系統(tǒng)消息* “" + name + "”已上線"); 

    /* 
    * 通過客戶端的Socket獲取輸入流 
    * 讀取客戶端發(fā)送來的信息 
    */ 
    BufferedReader bReader = new BufferedReader( 
     new InputStreamReader(socket.getInputStream(), "UTF-8")); 
    String msgString = null; 


    while((msgString = bReader.readLine()) != null) { 
     // 檢驗(yàn)是否為私聊（格式：@昵稱：內(nèi)容） 
     if(msgString.startsWith("@")) { 
      int index = msgString.indexOf("："); 
      if(index >= 0) { 
       //獲取昵稱 
       String theName = msgString.substring(1, index); 
       String info = msgString.substring(index+1, msgString.length()); 
       info = name + "："+ info; 
       //將私聊信息發(fā)送出去 
       sendToSomeone(theName, info);

       sendToSomeone(name,info);

       continue; 
      } 
     } 
     // 遍歷所有輸出流，將該客戶端發(fā)送的信息轉(zhuǎn)發(fā)給所有客戶端 
     m_display.append(name+"："+ msgString+"\n"); 
     sendToAll(name+"："+ msgString); 
    }  
   } catch (Exception e) { 
    // e.printStackTrace(); 
   } finally { 
    remove(name); 
    // 通知所有客戶端，某某客戶已經(jīng)下線 
    sendToAll("*系統(tǒng)消息* "+name + "已經(jīng)下線了。\n"); 

    if(socket!=null) { 
     try { 
      socket.close(); 
     } catch(IOException e) { 
      e.printStackTrace(); 
     } 
    }  
   } 
  } 
 } 

 public static void main(String[] args) { 
  TCPServer server = new TCPServer();
  server.setSize(400,400);
  server.setVisible(true);
  server.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  server.start(); 
 } 
} 

package server;

import java.io.*; 
import java.net.*; 
import java.util.Scanner; 
import java.util.concurrent.ExecutorService; 
import java.util.concurrent.Executors; 
import java.util.concurrent.ThreadPoolExecutor; 
import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionListener;
import java.awt.event.ActionEvent;

public class TCPClient extends JFrame { 

 private JTextField m_enter=new JTextField();
 private JTextArea m_display=new JTextArea();
 private int m_count=0;
 private static Socket clientSocket; 
 //private ExecutorService exec = Executors.newCachedThreadPool(); 
 private BufferedReader br;
 private PrintWriter pw;

 public TCPClient() 
 {
  super("聊天程序客戶端");


  Container c=getContentPane();
  //m_enter.setSize(100,99);
  //m_display.setSize(200,100);
  m_enter.setVisible(true);
  m_display.setVisible(true);
  m_enter.requestFocusInWindow();  //轉(zhuǎn)移輸入焦點(diǎn)到輸入?yún)^(qū)域

  //將光標(biāo)放置在文本區(qū)域的尾部
  m_display.setCaretPosition(m_display.getText().length());


  c.add(m_enter,BorderLayout.SOUTH);
  c.add(new JScrollPane(m_display),BorderLayout.CENTER); 
  // this.add(panel);
  // this.setContentPane(jp);

 } 


 public static void main(String[] args) throws Exception { 
  TCPClient client = new TCPClient();
  client.setVisible(true);
  client.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  client.setSize(470,708);
  client.start(); 
 } 


 public void start() { 
  try { 
   m_display.append("請創(chuàng)建用戶名：");
   clientSocket=new Socket("localhost",6666);
   BufferedReader br = new BufferedReader( 
     new InputStreamReader(clientSocket.getInputStream(), "UTF-8")); 
   PrintWriter pw = new PrintWriter( 
     new OutputStreamWriter(clientSocket.getOutputStream(), "UTF-8"), true); 
   //ListenrServser l=new ListenrServser();
   m_enter.addActionListener(new ActionListener(){
    public void actionPerformed(ActionEvent event)
    {
     try{
     String s=event.getActionCommand();
     m_enter.setText("");
      if(m_count==0)
     {
       pw.println(s);
       m_display.append("\n'"+s+"'"+"昵稱設(shè)置成功。\n");

     }
     else
     {

       pw.println(s);

     }
     m_count++;

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


   String msgString;
   while((msgString = br.readLine())!= null) { 
    m_display.append(msgString+"\n"); 
   } 


  } catch(Exception e) { 
   e.printStackTrace(); 
  } finally { 
   if (clientSocket !=null) { 
    try { 
     clientSocket.close(); 
    } catch(IOException e) { 
     e.printStackTrace(); 
    } 
   } 
  } 
 } 



}