package com.dragon.java.tcpchat;

import java.io.IOException;
import java.net.Socket;
import java.net.UnknownHostException;

/**
 * 客戶端
 * 
 * @author Auser
 * 
 */
public class Client {
  public static void main(String args[]) throws UnknownHostException,
      IOException {
    Socket client = new Socket("192.168.1.188", 10000);

    // 穿件發(fā)送信息的線程
    new ClientSend(client).start();
    // 創(chuàng)建接收信息的線程
    new ClientReceive(client).start();
　　　　　// 因?yàn)橐獙?shí)現(xiàn)聊天功能，而不是只發(fā)送一次信息，所以輸出流和客戶端都不能關(guān)閉。
    // client.shutdownOutput();
    // client.close();

  }
}

package com.dragon.java.tcpchat;

import java.io.IOException;
import java.io.PrintStream;
import java.net.Socket;
import java.util.Scanner;

/**
 * 客戶端向服務(wù)端發(fā)送信息的線程
 * 
 * @author Auser
 * 
 */
public class ClientSend extends Thread {
  private Scanner scanner;
  private Socket socket;

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

  @Override
  public void run() {
    scanner = new Scanner(System.in);

    try {
      PrintStream ps = new PrintStream(socket.getOutputStream());
      String line = "";
      // 阻塞式發(fā)送信息
      while ((line = scanner.nextLine()) != null) {
        ps.println(line);
      }
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}

package com.dragon.java.tcpchat;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.Socket;

/**
 * 客戶端接收信息的線程
 * 
 * @author Auser
 * 
 */
public class ClientReceive extends Thread {
  private Socket socket;

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

  @Override
  public void run() {
    try {
      BufferedReader br = new BufferedReader(new InputStreamReader(
          socket.getInputStream()));
      // 按行接收信息
      String line = "";
      while ((line = br.readLine()) != null) {
        System.out.println(line);
      }
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

}

package com.dragon.java.tcpchat;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

/**
 * 服務(wù)器
 * 
 * @author Auser
 * 
 */
public class Server {
  public static void main(String[] args) throws IOException,
      InterruptedException {
    List<Socket> list = new ArrayList<>();

    // 創(chuàng)建服務(wù)器端的套接字
    ServerSocket server = new ServerSocket(10000);

    while (true) {
      // 接收客戶端的阻塞方法
      Socket socket = server.accept();
      // 設(shè)計(jì)到多個(gè)線程可能會(huì)對(duì)集合進(jìn)行增刪的操作，要進(jìn)行同步的處理
      synchronized (list) {
        list.add(socket);
      }
      // 啟動(dòng)一個(gè)新的線程去處理這個(gè)客戶端的交流
      new HandleSocket(socket, list).start();
    }
　　　　 // 因?yàn)椴恢揽蛻舳耸裁磿r(shí)候發(fā)送信息，所以服務(wù)端要一直開啟不能關(guān)閉。
  }
}

package com.dragon.java.tcpchat;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.net.InetAddress;
import java.net.Socket;
import java.util.List;

/**
 * 處理每個(gè)(單個(gè))連接到服務(wù)器的客戶端的線程
 * 
 * @author Auser
 * 
 */
public class HandleSocket extends Thread {

  private Socket socket;
  private List<Socket> list;

  /**
   * 構(gòu)造方法
   * 
   * @param socket
   *      當(dāng)前連接的客戶端
   * @param list
   *      存儲(chǔ)已連接客戶端的集合
   */
  public HandleSocket(Socket socket, List<Socket> list) {
    this.socket = socket;
    this.list = list;
  }

  /**
   * 線程run方法
   */
  @Override
  public void run() {
    InetAddress address = socket.getInetAddress(); // 獲取連接到服務(wù)器的這的客戶端的地址
    String ip = address.getHostAddress();
    System.out.println(ip + "上線了！");
    if (ip.equals("192.168.1.117")) {
      synchronized (list) {
        sendToAll(ip + "由于違規(guī)操作，已被拉入黑名單！");
        list.remove(socket);
      }
      return;
    }

    try {
      BufferedReader br = new BufferedReader(new InputStreamReader(
          socket.getInputStream(), "gbk"));

      String line = "";
      while ((line = br.readLine()) != null) {
        String msg = ip + ":" + line;
        System.out.println(msg); // 輸出到服務(wù)器端的控制臺(tái)
        // 把這個(gè)客戶端說(shuō)的話，發(fā)給其他所有的客戶端
        sendToAll(msg);
      }

    } catch (IOException e) {
      // e.printStackTrace();
      System.out.println(ip + "下線了！");
      synchronized (list) {
        list.remove(socket);
      }
    }
  }

  /**
   * 把信息發(fā)送給所有的客戶端，去除當(dāng)前socket
   * 
   * @param msg
   *      發(fā)送的信息
   */
  private void sendToAll(String msg) {
    synchronized (list) {
      for (Socket s : list) {
        if (s != socket) {
          try {
            PrintStream ps = new PrintStream(s.getOutputStream());
            ps.println();
          } catch (IOException e) {
            e.printStackTrace();
          }
        }
      }
    }
  }

}