tcp聊天室转发消息

1.服务器连接

package com.wxh.chatroom;

import java.io.*; 
import java.net.Socket;
import java.net.SocketException;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;


public class ServerConnection extends Thread{
  private Socket socket;
  private PrintWriter pw;
  private BufferedReader br;

  public ServerConnection(Socket socket) {
    super();
    this.socket = socket;
    // 直接开启线程
    this.start();
  }

  // 发送信息的方法
  public void send(String msg) {
    // 使用socket将msg写出去
    try {
      if (pw == null) {
        pw = new PrintWriter(socket.getOutputStream());
      }
      // 输出信息
      pw.println(msg);
      pw.flush();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  // 关闭socket
  public void disConnect() {
    if (socket != null) {
      try {
        socket.close();
      } catch (Exception e) {
      }
    }
  }

  // 接收功能(线程)
  public void run() {
    // 不断读取服务器数据
    try {
      if (br == null) {
        br = new BufferedReader(new InputStreamReader(
            socket.getInputStream()));
      }
      String msg = null;
      while ((msg = br.readLine()) != null) {
        System.out.println(msg);
        
        // 转发
        Set<Map.Entry<String, ServerConnection>> entrySet = DB.connMap
            .entrySet();

        Iterator<Map.Entry<String, ServerConnection>> ite = entrySet
            .iterator();
        while (ite.hasNext()) {
          Map.Entry<String, ServerConnection> entry = ite.next();
          ServerConnection conn = entry.getValue();
          conn.send(socket.getInetAddress() + ":" + msg);
        }       
      }
    } catch (SocketException e) {
      System.out.println("下线了");
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
  public String getIp() {
    return this.socket.getInetAddress().getHostAddress();
  }
}

2.保存connection

package com.wxh.chatroom;

import java.util.HashMap;
import java.util.Map;

/**
 * 保存connection
 * @author Administrator
 *
 */
public class DB {
    public static Map<String,ServerConnection> connMap=new HashMap<String,ServerConnection>();
    
    public static void add(ServerConnection conn){
      connMap.put(conn.getIp(), conn);
    }
}

3.聊天室服务器端

package com.wxh.chatroom;

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

public class ChatRoomServer {

  public static void main(String[] args) {
    ServerConnection connection = null;
    try {
      // 创建ServerSocket
      ServerSocket serverSocket = new ServerSocket(9090);
      // 等待客户端连接
      boolean flag = true;
      while (flag) {
        Socket socket = serverSocket.accept();
        System.out.println(socket.getInetAddress() + "上线了");

        connection = new ServerConnection(socket);
        DB.add(connection);
      }

    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (connection != null) {
        connection.disConnect();
      }
    }
  }

}