Java實現(xiàn)NIO聊天室的示例代碼(群聊+私聊)
更新時間：2021年05月11日 14:13:19 作者：馬走日mazouri
這篇文章主要介紹了Java實現(xiàn)NIO聊天室的示例代碼(群聊+私聊)，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
功能介紹
功能:群聊+私發(fā)+上線提醒+下線提醒+查詢在線用戶
文件
Utils
需要用maven導(dǎo)入下面兩個包
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.16.18</version>
</dependency>
<dependency>
<groupId>ch.qos.logback</groupId>
<artifactId>logback-classic</artifactId>
<version>1.2.3</version>
</dependency>
package moremorechat_nio;
import lombok.extern.slf4j.Slf4j;
import java.io.*;
/**
* @author mazouri
* @create 2021-05-09 22:26
*/
@Slf4j
public class Utils {
/**
* 將二進制數(shù)據(jù)轉(zhuǎn)為對象
*
* @param buf
* @return
* @throws IOException
* @throws ClassNotFoundException
*/
public static Message decode(byte[] buf) throws IOException, ClassNotFoundException {
ByteArrayInputStream bais = new ByteArrayInputStream(buf);
ObjectInputStream ois = new ObjectInputStream(bais);
return (Message) ois.readObject();
}
/**
* 將對象轉(zhuǎn)為二進制數(shù)據(jù)
*
* @param message
* @return
*/
public static byte[] encode(Message message) throws IOException {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(baos);
oos.writeObject(message);
oos.flush();
return baos.toByteArray();
}
}
FinalValue
package moremorechat_nio;
/**
* @author mazouri
* @create 2021-05-05 21:00
*/
public final class FinalValue {
/**
* 系統(tǒng)消息
*/
public static final int MSG_SYSTEM = 0;
/**
* 群發(fā)消息
*/
public static final int MSG_GROUP = 1;
/**
* 私發(fā)消息
*/
public static final int MSG_PRIVATE = 2;
/**
* 客戶端請求在線人員
*/
public static final int MSG_ONLINE = 3;
/**
* 客戶端將用戶名稱發(fā)送給服務(wù)端
*/
public static final int MSG_NAME = 4;
}
Message
package moremorechat_nio;
import java.io.Serializable;
/**
* @author mazouri
* @create 2021-05-05 21:00
*/
public class Message implements Serializable {
public int type;
public String message;
public Message() {
}
public Message(String message) {
this.message = message;
}
public Message(int type, String message) {
this.type = type;
this.message = message;
}
@Override
public String toString() {
return "Message{" +
"type=" + type +
", message='" + message + '\'' +
'}';
}
}
NioServer
package moremorechat_nio;
import lombok.extern.slf4j.Slf4j;
import java.io.*;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;
import static moremorechat_nio.FinalValue.*;
/**
* ctrl+f12 方法
* ctrl+alt+左鍵
* @author mazouri
* @create 2021-05-09 19:24
*/
@Slf4j
public class NioServer {
private Selector selector;
private ServerSocketChannel ssc;
public NioServer() {
try {
// 創(chuàng)建 selector, 管理多個 channel
selector = Selector.open();
//打開ServerSocketChannel，用于監(jiān)聽客戶端的連接，它是所有客戶端連接的父通道
ssc = ServerSocketChannel.open();
ssc.bind(new InetSocketAddress(8888));
//設(shè)置連接為非堵塞模式
ssc.configureBlocking(false);
// 2. 建立 selector 和 channel 的聯(lián)系（注冊）
// SelectionKey 就是將來事件發(fā)生后，通過它可以知道事件和哪個channel的事件
//將ServerSocketChannel注冊到Reactor線程的多路復(fù)用器Selector上，監(jiān)聽ACCEPT事件
ssc.register(selector, SelectionKey.OP_ACCEPT);
} catch (IOException e) {
e.printStackTrace();
}
}
public static void main(String[] args) {
NioServer server = new NioServer();
log.debug("server啟動完成，等待用戶連接...");
try {
server.listen();
} catch (Exception e) {
log.debug("發(fā)生了一些問題");
}
}
/**
* 監(jiān)聽用戶的連接
*
* @throws Exception
*/
private void listen() throws Exception {
while (true) {
// select 方法, 沒有事件發(fā)生，線程阻塞，有事件，線程才會恢復(fù)運行, 通過Selector的select（）方法可以選擇已經(jīng)準備就緒的通道 （這些通道包含你感興趣的的事件）
//通過Selector的select（）方法可以選擇已經(jīng)準備就緒的通道 （這些通道包含你感興趣的的事件）
// select 在事件未處理時，它不會阻塞, 事件發(fā)生后要么處理，要么取消，不能置之不理
selector.select();
// 處理事件, selectedKeys 內(nèi)部包含了所有發(fā)生的事件
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while (iterator.hasNext()) {
SelectionKey key = iterator.next();
// 處理key 時，要從 selectedKeys 集合中刪除，否則下次處理就會有問題
iterator.remove();
// 區(qū)分事件類型
if (key.isAcceptable()) {
ServerSocketChannel channel = (ServerSocketChannel) key.channel();
SocketChannel sc = channel.accept();
sc.configureBlocking(false);
sc.register(selector, SelectionKey.OP_READ);
} else if (key.isReadable()) {
dealReadEvent(key);
}
}
}
}
/**
* 處理讀事件
*
* @param key
*/
private void dealReadEvent(SelectionKey key) {
SocketChannel channel = null;
try {
channel = (SocketChannel) key.channel();
ByteBuffer buffer = ByteBuffer.allocate(1024);
int read = channel.read(buffer);
// 如果是正常斷開，read 的方法的返回值是 -1
if (read == -1) {
//cancel 會取消注冊在 selector 上的 channel，并從 keys 集合中刪除 key 后續(xù)不會再監(jiān)聽事件
key.cancel();
} else {
buffer.flip();
Message msg = Utils.decode(buffer.array());
log.debug(msg.toString());
dealMessage(msg, key, channel);
}
} catch (IOException | ClassNotFoundException e) {
System.out.println((key.attachment() == null ? "匿名用戶" : key.attachment()) + " 離線了..");
dealMessage(new Message(MSG_SYSTEM, key.attachment() + " 離線了.."), key, channel);
//取消注冊
key.cancel();
//關(guān)閉通道
try {
channel.close();
} catch (IOException ioException) {
ioException.printStackTrace();
}
}
}
/**
* 處理各種消息，并發(fā)送給客戶端
*
* @param msg
* @param key
* @param channel
*/
private void dealMessage(Message msg, SelectionKey key, SocketChannel channel) {
switch (msg.type) {
case MSG_NAME:
key.attach(msg.message);
log.debug("用戶{}已上線", msg.message);
getConnectedChannel(channel).forEach(selectionKey -> {
SocketChannel sc = (SocketChannel) selectionKey.channel();
sendMsgToClient(new Message("收到一條系統(tǒng)消息: " + msg.message + "已上線"), sc);
});
break;
case MSG_GROUP:
getConnectedChannel(channel).forEach(selectionKey -> {
SocketChannel sc = (SocketChannel) selectionKey.channel();
sendMsgToClient(new Message(key.attachment() + "給大家發(fā)送了一條消息: " + msg.message), sc);
});
break;
case MSG_PRIVATE:
String[] s = msg.message.split("_");
AtomicBoolean flag = new AtomicBoolean(false);
getConnectedChannel(channel).stream().filter(sk -> s[0].equals(sk.attachment())).forEach(selectionKey -> {
SocketChannel sc = (SocketChannel) selectionKey.channel();
sendMsgToClient(new Message(key.attachment() + "給你發(fā)送了一條消息: " + s[1]), sc);
flag.set(true);
});
if (!flag.get()){
sendMsgToClient(new Message(s[1]+"用戶不存在,請重新輸入?。?！"), channel);
}
break;
case MSG_ONLINE:
ArrayList<String> onlineList = new ArrayList<>();
onlineList.add((String) key.attachment());
getConnectedChannel(channel).forEach(selectionKey -> onlineList.add((String) selectionKey.attachment()));
sendMsgToClient(new Message(onlineList.toString()), channel);
break;
case MSG_SYSTEM:
getConnectedChannel(channel).forEach(selectionKey -> {
SocketChannel sc = (SocketChannel) selectionKey.channel();
sendMsgToClient(new Message("收到一條系統(tǒng)消息: " + msg.message), sc);
});
break;
default:
break;
}
}
/**
* 發(fā)送消息給客戶端
*
* @param msg
* @param sc
*/
private void sendMsgToClient(Message msg, SocketChannel sc) {
try {
byte[] bytes = Utils.encode(msg);
sc.write(ByteBuffer.wrap(bytes));
} catch (IOException e) {
log.debug("sendMsgToClient出現(xiàn)了一些問題");
}
}
/**
* 獲取所有channel,除去調(diào)用者
*
* @param channel
* @return
*/
private Set<SelectionKey> getConnectedChannel(SocketChannel channel) {
return selector.keys().stream()
.filter(item -> item.channel() instanceof SocketChannel && item.channel().isOpen() && item.channel() != channel)
.collect(Collectors.toSet());
}
}
NioClient
package moremorechat_nio;
import lombok.extern.slf4j.Slf4j;
import java.io.*;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.util.Iterator;
import java.util.Scanner;
import static moremorechat_nio.FinalValue.*;
/**
* @author mazouri
* @create 2021-04-29 12:02
*/
@Slf4j
public class NioClient {
private Selector selector;
private SocketChannel socketChannel;
private String username;
private static Scanner input;
public NioClient() throws IOException {
selector = Selector.open();
socketChannel = SocketChannel.open(new InetSocketAddress("127.0.0.1", 8888));
socketChannel.configureBlocking(false);
socketChannel.register(selector, SelectionKey.OP_READ);
log.debug("client啟動完成......");
log.debug("請輸入你的名字完成注冊");
input = new Scanner(System.in);
username = input.next();
log.debug("歡迎{}來到聊天系統(tǒng)", username);
}
public static void main(String[] args) throws IOException {
System.out.println("tips: \n1. 直接發(fā)送消息會發(fā)給當前的所有用戶 \n2. @用戶名:消息 會私發(fā)給你要發(fā)送的用戶 \n3. 輸入 查詢在線用戶 會顯示當前的在線用戶");
NioClient client = new NioClient();
//啟動一個子線程接受服務(wù)器發(fā)送過來的消息
new Thread(() -> {
try {
client.acceptMessageFromServer();
} catch (Exception e) {
e.printStackTrace();
}
}, "receiveClientThread").start();
//調(diào)用sendMessageToServer,發(fā)送消息到服務(wù)端
client.sendMessageToServer();
}
/**
* 將消息發(fā)送到服務(wù)端
*
* @throws IOException
*/
private void sendMessageToServer() throws IOException {
//先把用戶名發(fā)給客戶端
Message message = new Message(MSG_NAME, username);
byte[] bytes = Utils.encode(message);
socketChannel.write(ByteBuffer.wrap(bytes));
while (input.hasNextLine()) {
String msgStr = input.next();
Message msg;
boolean isPrivate = msgStr.startsWith("@");
if (isPrivate) {
int idx = msgStr.indexOf(":");
String targetName = msgStr.substring(1, idx);
msgStr = msgStr.substring(idx + 1);
msg = new Message(MSG_PRIVATE, targetName + "_" + msgStr);
} else if ("查詢在線用戶".equals(msgStr)) {
msg = new Message(MSG_ONLINE, "請求在線人數(shù)");
} else {
msg = new Message(MSG_GROUP, msgStr);
}
byte[] bytes1 = Utils.encode(msg);
socketChannel.write(ByteBuffer.wrap(bytes1));
}
}
/**
* 接受從服務(wù)器發(fā)送過來的消息
*/
private void acceptMessageFromServer() throws Exception {
while (selector.select() > 0) {
Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
while (iterator.hasNext()) {
SelectionKey key = iterator.next();
iterator.remove();
if (key.isReadable()) {
SocketChannel sc = (SocketChannel) key.channel();
ByteBuffer buffer = ByteBuffer.allocate(1024);
sc.read(buffer);
Message message = Utils.decode(buffer.array());
log.debug(String.valueOf(message.message));
}
}
}
}
}
到此這篇關(guān)于Java實現(xiàn)NIO聊天室的示例代碼(群聊+私聊)的文章就介紹到這了,更多相關(guān)Java NIO聊天室內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
springcloud?gateway高級功能之集成apollo后動態(tài)刷新路由方式
這篇文章主要介紹了springcloud?gateway高級功能之集成apollo后動態(tài)刷新路由方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
這篇文章主要介紹了Java深度優(yōu)先遍歷解決排列組合問題詳解,深度優(yōu)先搜索是遞歸過程,帶有回退操作,因此需要使用棧存儲訪問的路徑信息,當訪問到的當前頂點沒有可以前進的鄰接頂點時,需要進行出棧操作,將當前位置回退至出棧元素位置,需要的朋友可以參考下2024-01-01
Spring Security基于過濾器實現(xiàn)圖形驗證碼功能
驗證碼就是為了防止惡意用戶采用暴力重試的攻擊手段而設(shè)置的一種防護措施，接下來在Spring Security的環(huán)境中，我們可以用兩種方案實現(xiàn)圖形驗證碼，具體實現(xiàn)方法跟隨小編一起看看吧2021-09-09
SpringBoot yml配置文件調(diào)用過程解析
這篇文章主要介紹了SpringBoot yml配置文件調(diào)用過程解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-11-11
本篇文章主要介紹了JAVA實現(xiàn)雙向鏈表的增刪功能的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-03-03
舉例講解Java編程中this關(guān)鍵字與super關(guān)鍵字的用法
這篇文章主要介紹了Java編程中this關(guān)鍵字與super關(guān)鍵字的用法示例,super是this的父輩,在繼承過程中兩個關(guān)鍵字經(jīng)常被用到,需要的朋友可以參考下2016-03-03
這篇文章主要介紹了SpringBoot整合Swagger3的流程詳解,Swagger最核心的類就是Docket、它可以配置作者信息、掃描類型,在SwaggerConfig配置類,添加@Configuration和@EnableOpenApi注解,需要的朋友可以參考下2024-01-01
這篇文章主要詳細介紹了Java中的6種請求方式,@RequestParam、@PathVariable、@MatrixVariable、@RequestBody、@RequestHeader和@CookieValue的基本知識、詳細分析以及示例,需要的朋友可以參考下2024-07-07
Java?Web中ServletContext對象詳解與應(yīng)用
ServletContext是一個容器,可以用來存放變量,供一個web項目中多個Servlet共享,下面這篇文章主要給大家介紹了關(guān)于Java?Web中ServletContext對象詳解與應(yīng)用的相關(guān)資料,需要的朋友可以參考下2023-04-04
Idea 同一窗口導(dǎo)入多個項目的實現(xiàn)步驟
本文主要介紹了Idea 同一窗口導(dǎo)入多個項目的實現(xiàn)步驟，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07