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<string, arraylist<chatclient>> groups = new hashmap<string, arraylist<chatclient>>();

 // 构造器
 public chatserver() {
 try {
  // 消息接受套接字对象的构造初始化
  server = new datagramsocket(port);
 } catch (socketexception e) {
  e.printstacktrace();
 }
 }

 // 注册聊天室新登录用户
 public static void logingroup(string groupid, chatclient client) {
 // 通过聊天室id，获取该聊天室的所有在线用户
 arraylist<chatclient> clients = groups.get(groupid);
 if (clients == null) {
  clients = new arraylist<chatclient>();
 }
 // 将此次进入聊天室的用户登记
 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<chatclient> 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", "小黑");
 }
}

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