java tcp 多人聊天_Java TCP简易多人聊天室内

一.说明：

​本例是一个简易的Java TCP多人聊天室，先启动服务器端，再启动客户端，客户端敲入用户名，然后可以开始聊天，敲入信息，每一个在线的用户都会收到相应信息。

演示如下图：

二.服务器端代码

ServerMain.java

import java.io.*;

public class ServerMain {

public static void main(String[] args) {

ChatServer server = new ChatServer();

try {

server.ServerStart();

}

catch (IOException e) {

e.printStackTrace();

}

}

}

ChatServer.java

import java.io.*;

import java.net.*;

import java.util.HashMap;

public class ChatServer {

// 设定端口号

private final int port = 12345;

public static HashMap clientMap = new HashMap<>();

public static int onlineNum = 0;

public static void ServerSendToAll(ClientThread thread,String msg) throws IOException {

for (ClientThread t : clientMap.keySet()) {

if (t.clientName.equals(thread.clientName)) {

Message.printMessage(msg);

continue;

}

t.objOutput.writeObject(new DataPacket(Message.CMD.SENDALL, msg));

}

}

public static void ServerSendToAll(ClientThread thread) throws IOException {

onlineNum++;

// 服务端记录

Message.printMessage(thread.clientName + "上线了，当前在线人数" + onlineNum + "人");

// 群发

for (ClientThread t : clientMap.keySet()) {

if (t.login == true) {

t.objOutput.writeObject(new DataPacket(Message.CMD.LOGIN, "当前在线人数" + onlineNum + "人"));

}

}

}

public void ServerStart() throws IOException {

ServerSocket serverSocket = new ServerSocket(port);

Message.printMessage("Server Start.");

// 死循环等待客户端连接

while (true) {

// 每来一个客户端连接，保存Socket对象

Socket clientSocket = serverSocket.accept();

ClientThread client = new ClientThread(clientSocket);

// 每连接一个客户端，创建一个对应的线程类

Thread clientThread = new Thread(client);

// 添加对象-Socket映射

clientMap.put(client, clientSocket);

// 启动线程

clientThread.start();

}

}

}

ClientThread.java

import java.io.*;

import java.net.*;

/*

线程类：处理每一个登录成功的客户端

*/

public class ClientThread implements Runnable {

// 保存每个客户端Socket

private Socket client = null;

public ObjectInputStream objInput = null;

public ObjectOutputStream objOutput = null;

public String clientName = null;

public boolean login = false;

public ClientThread(Socket client) {

this.client = client;

try {

this.objInput = new ObjectInputStream(this.client.getInputStream());

this.objOutput = new ObjectOutputStream(this.client.getOutputStream());

} catch (IOException e) {

Message.printLog(Message.logType.ERROR, "ClientThread:IOException!", true);

e.printStackTrace();

}

}

public int HandleReceiveMsg(Object data) throws IOException {

DataPacket dataPacket = (DataPacket) data;

switch (dataPacket.cmd) {

case LOGIN:

clientName = dataPacket.msg;

login = true;

ChatServer.ServerSendToAll(this);

break;

case SEND:

// 收到消息,调用群发其他用户

String msg = this.clientName + "说:" + dataPacket.msg;

ChatServer.ServerSendToAll(this, msg);

break;

case SENDALL:

// 收到其他用户群发的消息

SendMessage(dataPacket);

break;

}

return 0;

}

public void SendMessage(DataPacket data) throws IOException {

this.objOutput.writeObject(data);

}

public Object ReceiveMessage() throws ClassNotFoundException, IOException {

try {

Object obj = this.objInput.readObject();

return obj;

} catch (EOFException e) {

}

return null;

}

@Override

public void run() {

try {

Object data;

while ((data = ReceiveMessage()) != null) {

HandleReceiveMsg(data);

}

} catch (SocketException e) {

login = false;

} catch (IOException e) {

Message.printLog(Message.logType.ERROR, "ClientThread-run:IOException!", true);

e.printStackTrace();

} catch (ClassNotFoundException e) {

Message.printLog(Message.logType.ERROR, "ClientThread-run:ClassNotFoundException!", true);

e.printStackTrace();

} finally {

// 用户断开连接

if (login == false) {

// 从HashMap 移走映射

ChatServer.clientMap.remove(this);

// 释放资源

if (client != null) {

try {

client.close();

} catch (IOException e) {

e.printStackTrace();

}

}

ChatServer.onlineNum--;

// 打印信息

String msg = this.clientName + "下线了，当前在线" + ChatServer.onlineNum + "人";

Message.printMessage(msg);

try {

ChatServer.ServerSendToAll(this, msg);

} catch (IOException e) {

e.printStackTrace();

}

}

}

}

}

三.客户端代码

ClientMain.java

public class ClientMain {

public static void main(String[] args) {

ChatClient client = new ChatClient();

client.ClientStart();

}

}

ChatClient.java

import java.io.*;

import java.net.*;

import java.util.Scanner;

public class ChatClient {

private static final int port = 12345;

private static String serverAddress = "localhost";

private Socket client = null;

private ObjectOutputStream objOutput = null;

private ObjectInputStream objInput = null;

public void SendMessage(DataPacket data) throws IOException {

// 发送数据给服务器

objOutput.writeObject(data);

}

public void ClientStart() {

Scanner in = new Scanner(System.in);

String userInput = null;

// 用户输入用户名登录

Message.printLog(Message.logType.INFO, "请输入你的用户名：", false);

userInput = in.nextLine();

// 连接服务器

try {

this.client = new Socket(serverAddress, port);

// 服务器已连接

if (this.client.isConnected()) {

// 获取输出流

this.objOutput = new ObjectOutputStream(this.client.getOutputStream());

// 获取输入流

this.objInput = new ObjectInputStream(this.client.getInputStream());

SendMessage(new DataPacket(Message.CMD.LOGIN, userInput));

// 创建一个线程用于接收信息

ClientReceiveMessageThread receiveMessage = new ClientReceiveMessageThread(this.objInput, userInput);

Thread receiveThread = new Thread(receiveMessage);

receiveThread.start();

// 这个线程用于发送信息

while (true) {

userInput = in.nextLine();

if (userInput != null) {

SendMessage(new DataPacket(Message.CMD.SEND, userInput));

}

}

}

} catch (ConnectException e) {

Message.printLog(Message.logType.ERROR, "ChatClient-ClientStart:ConnectException!", true);

e.printStackTrace();

} catch (SocketException e) {

Message.printLog(Message.logType.ERROR, "服务器异常！", true);

System.exit(0);

} catch (IOException e) {

Message.printLog(Message.logType.ERROR, "ChatClient-ClientStart:IOException!", true);

e.printStackTrace();

}

}

}

class ClientReceiveMessageThread implements Runnable {

private ObjectInputStream in = null;

private String clientName = null;

public ClientReceiveMessageThread(ObjectInputStream in, String name) {

this.in = in;

this.clientName = name;

}

public int HandleReceiveMsg(Object data) throws IOException {

DataPacket dataPacket = (DataPacket) data;

switch (dataPacket.cmd) {

case LOGIN:

case SEND:

case SENDALL:

Message.printMessage(dataPacket.msg);

break;

}

return 0;

}

@Override

public void run() {

Object data;

try {

while ((data = in.readObject()) != null) {

HandleReceiveMsg(data);

}

} catch (EOFException e) {

} catch (SocketException e) {

Message.printLog(Message.logType.ERROR, "服务器异常！", true);

System.exit(0);

} catch (IOException e) {

e.printStackTrace();

} catch (ClassNotFoundException e) {

e.printStackTrace();

}

}

}

四.共用的类

Message.java

import java.text.SimpleDateFormat;

import java.util.Date;

public class Message {

// 是否打开debug输出

private static boolean debug = false;

// 设定日期时间格式

public static SimpleDateFormat dateFormat = new SimpleDateFormat("yyy-MM-dd hh:mm:ss");

public static enum logType {

ERROR,

INFO,

WARN,

DEBUG,

};

public static enum CMD {

LOGIN,

SEND,

SENDALL,

}

public static void printLog(logType type, String msg, boolean newLine) {

if( debug == true || type != logType.DEBUG ) {

StringBuilder sb = new StringBuilder(128);

sb.append("[").append(type).append("]").append(msg);

if (newLine)

System.out.println(sb);

else

System.out.print(sb);

}

}

public static void printMessage(String msg) {

StringBuilder sb = new StringBuilder(1024);

sb.append("[").append(dateFormat.format(new Date())).append("]").append(msg);

System.out.println(sb);

}

}

DataPacket.java

import java.io.Serializable;

public class DataPacket implements Serializable {

/**

* 自动生成的UID

*/

private static final long serialVersionUID = -5366828169862565011L;

public Message.CMD cmd;

public String msg;

public DataPacket(Message.CMD cmd, String msg) {

this.cmd = cmd;

this.msg = msg;

}

}