Java NIO实现聊天室功能
更新时间：2021年11月24日 09:13:22 作者：岁月如歌似梦
这篇文章主要为大家详细介绍了Java NIO实现聊天室功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
本文实例为大家分享了Java NIO实现聊天室功能的具体代码，供大家参考，具体内容如下
代码里面已经包含了必要的注释，这里不详述了。实现了基本的聊天室功能。
常量类：
public class Constant {
public static final int serverPort = 44444;
}
服务端：
package server;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Iterator;
import java.util.Set;
import constant.Constant;
public class SocketServer {
private Charset charset = Charset.forName("UTF-8");
private ServerSocketChannel serverSocketChannel;
private Selector serverSocketSelector;
private SelectionKey serverRegisterKey;
private ByteBuffer buffer = ByteBuffer.allocate(1024);
public static void main(String[] args) throws IOException {
new SocketServer().openServer(new InetSocketAddress(Constant.serverPort));
}
public void openServer(SocketAddress address) throws IOException {
init(address);
handle();
}
private void init(SocketAddress address) throws IOException {
serverSocketSelector = Selector.open();
serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);
serverRegisterKey = serverSocketChannel.register(serverSocketSelector, SelectionKey.OP_ACCEPT);
serverSocketChannel.socket().bind(address);
}
private void handle() throws IOException {
System.out.println("服务端open");
while (serverSocketSelector.select() > 0) {
Iterator<SelectionKey> iterator = serverSocketSelector.selectedKeys().iterator();
// 为什么这里要用迭代器，而不用增强for循环之类的呢？是因为这里获得一个key之后，要对其进行移除，避免二次处理，造成影响
while (iterator.hasNext()) {
dispatch(iterator.next());
iterator.remove();
}
}
}
private void dispatch(SelectionKey key) throws IOException {
if (key.isAcceptable()) {
accept(key);
} else if (key.isReadable()) {
readMessage(key);
} else if (key.isValid() && key.isWritable()) {
writeMessage(key);
}
}
private void accept(SelectionKey key) throws IOException, ClosedChannelException {
// 主要的是，接收事件是发生在服务器这边的，所以这边的通道要强转为ServerSocketChannel
ServerSocketChannel server = (ServerSocketChannel) key.channel();
SocketChannel client = server.accept();
client.configureBlocking(false);
// 同时再给该通道注册选择器，监听的内容的读取
client.register(serverSocketSelector, SelectionKey.OP_READ);
}
private void readMessage(SelectionKey key) throws IOException {
SocketChannel client = (SocketChannel) key.channel();
client.read(buffer);
// 调整为读取模式
buffer.flip();
String content = charset.decode(buffer).toString();
// 压缩空间，即抛弃已经读取的内容（实际上还在里面，只是处于等待被覆盖状态）
buffer.compact();
// 这里可以根据业务逻辑，设置不设置都可以，但是这里想接受到消息后立马回复一条消息，所以设置下一次感兴趣的（监听）事件为写
key.interestOps(SelectionKey.OP_WRITE);
// 设置系统回复信息
key.attach("系统已经收到你的消息\n");
// 开始广播这个客户端的内容到其他客户端
broadcast(key, content);
}
private void broadcast(SelectionKey self, String content) throws IOException {
Set<SelectionKey> selectedKeys = self.selector().keys();
for (SelectionKey key : selectedKeys) {
// 不能发送给自己,也不要服务器自己本身对这个有反应
if (key != self && key != serverRegisterKey) {
String oldMessage = (String) key.attach(null);
// 如果有旧消息的话，在下一次发送时，连同旧消息一起发送
key.attach(oldMessage != null ? oldMessage + content : content);
key.interestOps(key.interestOps() | SelectionKey.OP_WRITE);
}
}
}
private void writeMessage(SelectionKey key) throws IOException {
SocketChannel client = (SocketChannel) key.channel();
// 获取发给这个客户端的消息，并清空消息
client.write(charset.encode((String) key.attach(null)));
key.interestOps(SelectionKey.OP_READ);
}
}
客户端（包含了Socket版本和SocketChannel版本）：
package client;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Scanner;
import constant.Constant;
public class SocketClient {
public static void main(String[] args) throws IOException {
nioVersion();
// ioVersion();
}
private static void ioVersion() throws UnknownHostException, IOException {
System.out.println("客户端");
final Socket socket = new Socket();
socket.connect(new InetSocketAddress(Constant.serverPort));
new Thread() {
@Override
public void run() {
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String line = scanner.nextLine();
try {
socket.getOutputStream().write((line + "\n").getBytes("UTF-8"));
} catch (IOException e) {
e.printStackTrace();
}
}
scanner.close();
try {
socket.close();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
};
}.start();
new Thread() {
@Override
public void run() {
try {
Scanner scanner = new Scanner(socket.getInputStream(), "utf-8");
while (scanner.hasNext()) {
String line = scanner.nextLine();
System.out.println("收到消息：" + line);
}
scanner.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}.start();
}
private static void nioVersion() throws IOException {
Charset charset = Charset.forName("UTF-8");
System.out.println("客户端");
SocketChannel socketChannel = SocketChannel.open();
// 设置为非阻塞模式
socketChannel.configureBlocking(false);
socketChannel.connect(new InetSocketAddress(Constant.serverPort));
while (true) {
if (socketChannel.finishConnect()) {
new Thread() {
@Override
public void run() {
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String input = scanner.nextLine();
try {
socketChannel.write(charset.encode(input));
} catch (IOException e) {
e.printStackTrace();
}
}
scanner.close();
}
}.start();
new Thread() {
ByteBuffer dst = ByteBuffer.allocate(1024);
@Override
public void run() {
while (true) {
try {
int len = socketChannel.read(dst);
if (len > 0) {
dst.flip();
System.out.println("收到消息：" + charset.decode(dst));
dst.compact();
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}.start();
return;
}
}
}
}
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Java垃圾回收机制算法详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-08-08
本文将介绍通过Java程序来读取PDF文档中的文本和图片的方法。分别调用方法extractText()和extractImages()来读取,需要的朋友可以参考下2019-07-07
Spring Boot插件spring tool suite安装及使用详解
这篇文章主要介绍了Spring Boot插件spring tool suite安装及使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
这篇文章主要介绍了java中的实体类时间格式化方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-06-06
这篇文章主要介绍了Java线程编程中Thread类的基础学习教程,Thread类包含诸多操作线程的方法,非常重要,需要的朋友可以参考下2015-12-12
本文主要介绍了JAVA删除字符串固定下标字串的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
这篇文章主要为大家详细介绍了java连接mysql数据库的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-05-05
详解SpringMVC使用MultipartFile实现文件的上传
本篇文章主要介绍了SpringMVC使用MultipartFile实现文件的上传，本地的文件上传到资源服务器上，比较好的办法就是通过ftp上传。这里是结合SpringMVC+ftp的形式上传的，有兴趣的可以了解一下。2016-12-12
Spring中的InitializingBean接口源码解析
这篇文章主要介绍了Spring中的InitializingBean接口源码解析,InitializingBean接口为Bean初始化提供了一种方式,实现InitializingBean接口的Bean,在BeanFactory设置其所有属性后会调用其afterPropertiesSet()方法,需要的朋友可以参考下2024-02-02
Spring的请求映射handlerMapping以及原理详解
这篇文章主要介绍了Spring的请求映射handlerMapping以及原理详解,我们每次发请求,它到底是怎么找到我们哪个方法来去处理这个请求,因为我们知道所有的请求过来都会来到DispatcherServlet,springboot底层还是使用的是springMVC,需要的朋友可以参考下2023-08-08
最新评论