java udp 聊天室_Java基于UDP协议实现简单的聊天室程序

最近比较闲，一直在抽空回顾一些Java方面的技术应用。

今天没什么事做，基于UDP协议，写了一个非常简单的聊天室程序。

现在的工作，很少用到socket，也算是对Java网络编程方面的一个简单回忆。

先看一下效果：

实现的效果可以说是非常非常简单，但还是可以简单的看到一个实现原理。

“聊天室001”的用户，小红和小绿相互聊了两句，“聊天室002”的小黑无人理会，在一旁寂寞着。

看一下代码实现：

1、首先是消息服务器的实现，功能很简单：•将客户端的信息(进入了哪一个聊天室等)进行登记；

•构造UDP协议套接字对象，接受各个客户端发送的消息；

•解析消息内容，将聊天信息推送回对应聊天室的各个客户端；

package com.tsr.simplechat.receive_server;

import java.io.IOException;

import java.net.DatagramPacket;

import java.net.DatagramSocket;

import java.net.SocketException;

import java.util.ArrayList;

import java.util.HashMap;

import com.google.gson.Gson;

import com.tsr.simplechat.bean.MessageEntity;

import com.tsr.simplechat.client.ChatClient;

//聊天服务器

public class ChatServer extends Thread {

// 程序占用端口号

private static final int PORT = 10000;

// 消息接受套接字对象

private static DatagramSocket server = null;

// 字典对象(Key：聊天室ID，Value：该聊天室下的客户端用户集合);

private static HashMap> groups = new HashMap>();

// 构造器

public ChatServer() {

try {

// 消息接受套接字对象的构造初始化

server = new DatagramSocket(PORT);

} catch (SocketException e) {

e.printStackTrace();

}

}

// 注册聊天室新登录用户

public static void logInGroup(String groupID, ChatClient client) {

// 通过聊天室ID，获取该聊天室的所有在线用户

ArrayList clients = groups.get(groupID);

if (clients == null) {

clients = new ArrayList();

}

// 将此次进入聊天室的用户登记

clients.add(client);

// 更新聊天室信息

groups.put(groupID, clients);

}

// 循环接收消息

@Override

public void run() {

while (true) {

receiveMessage();

}

}

private void receiveMessage() {

// UDP数据包

byte[] buf = new byte[1024];

DatagramPacket packet = new DatagramPacket(buf, buf.length);

while (true) {

try {

// 接受数据包

server.receive(packet);

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

// 解析数据包，获取聊天信息

String content = new String(packet.getData(), 0, packet.getLength());

// 通过第三方包解析json数据

Gson gson = new Gson();

MessageEntity me = gson.fromJson(content, MessageEntity.class);

// 解析消息内容，通过聊天室ID，获取该聊天室的所有在线用户

ArrayList clients = groups.get(me.getGroupId());

// 将接收到的消息推送回该聊天室的各个用户

for (ChatClient client : clients) {

client.pushBackMessage(me);

}

}

}

}

2、客户端程序，依然很简单：

•简单的定义客户端聊天室界面。

•构造消息发送套接字对象。

•获取聊天信息框的内容，发送到服务器。

package com.tsr.simplechat.client;

import java.awt.Button;

import java.awt.Event;

import java.awt.Frame;

import java.awt.TextArea;

import java.awt.TextField;

import java.awt.event.WindowAdapter;

import java.awt.event.WindowEvent;

import java.io.IOException;

import java.net.DatagramPacket;

import java.net.DatagramSocket;

import java.net.InetAddress;

import java.net.SocketException;

import java.net.UnknownHostException;

import com.tsr.simplechat.bean.MessageEntity;

import com.tsr.simplechat.receive_server.ChatServer;

//客户端程序

public class ChatClient extends Frame {

private static final long serialVersionUID = 1L;

// 聊天室ID

private String groupID;

// 客户端用户名

private String clientName;

// 客户端消息发送服务套接字

private DatagramSocket msg_send;

// 服务端口

private final int PORT = 10000;

// 服务器IP地址

private InetAddress ip;

// 客户端控件

TextField tf = new TextField(20);

TextArea ta = new TextArea();

Button send = new Button("send");

// 客户端构造器

public ChatClient(String groupID, String clientName) {

super("聊天室:" + groupID + "/" + clientName);

this.clientName = clientName;

this.groupID = groupID;

// 设置客户端界面样式

add("North", tf);

add("Center", ta);

add("South", send);

setSize(250, 250);

show();

// 聊天相关服务器初始化

init();

// 监视器

addWindowListener(new WindowAdapter() {

public void windowClosing(WindowEvent e) {

// 关闭消息发送服务

msg_send.close();

// 关闭客户端程序

dispose();

System.exit(0);

}

});

}

// 聊天相关服务器初始化

private void init() {

// 注册当前用户及所在聊天室信息注册到服务器

ChatServer.logInGroup(groupID, this);

try {

// 初始化消息发送套接字对象

msg_send = new DatagramSocket();

// 指定消息服务器

try {

ip = InetAddress.getByName("127.0.0.1");

} catch (UnknownHostException e) {

System.out.println("未知的主机异常..");

}

} catch (SocketException e) {

System.out.println("套接字连接异常..");

}

}

// 消息发送按钮时间监听

public boolean action(Event evt, Object arg) {

if (evt.target.equals(send)) {

try {

// 获取输入内容

String content = tf.getText();

// 发送消息

send_message(content);

// 清空聊天框

tf.setText(null);

} catch (Exception ioe) {

System.out.print(ioe.getMessage());

}

}

return true;

}

// 消息发送

private void send_message(String content) {

// 消息格式化(json格式)

String message = messageFormat(content);

// 将消息封装成UDP数据包

byte[] buf = message.getBytes();

DatagramPacket packet = new DatagramPacket(buf, buf.length, ip, PORT);

try {

// 通过UDP协议发送消息

msg_send.send(packet);

} catch (IOException e) {

System.out.println("IO异常..");

}

}

// 消息格式化

private String messageFormat(String content) {

StringBuffer buffer = new StringBuffer();

buffer.append("{\"groupId\":").append("\"").append(groupID).append(

"\",");

buffer.append("\"userName\":\"").append(clientName).append("\",");

buffer.append("\"text\":\"").append(content).append("\"}");

return buffer.toString();

}

// 从服务器获取当前聊天室最新消息(回调..)

public void pushBackMessage(MessageEntity me) {

ta.append(me.getUserName() + ":" + me.getText());

ta.append("\n");

}

}

3、消息实体类

主要就是用于将消息封装成对象，包含：聊天室ID、消息发送者昵称，消息内容。使用json格式解析。

package com.tsr.simplechat.bean;

//消息实体

public class MessageEntity {

private String groupId;

private String userName;

private String text;

public String getGroupId() {

return groupId;

}

public void setGroupId(String groupId) {

this.groupId = groupId;

}

public String getUserName() {

return userName;

}

public void setUserName(String userName) {

this.userName = userName;

}

public String getText() {

return text;

}

public void setText(String text) {

this.text = text;

}

}

4、OK，到了这里就基本搞定了，建立一个测试类。

•开启消息服务器。

•开启三个客户端，其中两个进入“聊天室001”，另一个进入“聊天室002”。

import com.tsr.simplechat.client.ChatClient;

import com.tsr.simplechat.receive_server.ChatServer;

public class Test {

public static void main(String[] args) {

ChatServer r = new ChatServer();

r.start();

ChatClient c1 = new ChatClient("001", "小红");

ChatClient c2 = new ChatClient("001", "小绿");

ChatClient c3 = new ChatClient("002", "小黑");

}

}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。