<?xml version='1.0' encoding='UTF-8'?><project xmlns='http://maven.apache.org/POM/4.0.0' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xsi:schemaLocation='http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd'> <modelVersion>4.0.0</modelVersion> <groupId>com.chat.info</groupId> <artifactId>chat-server</artifactId> <version>1.0-SNAPSHOT</version> <parent><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-parent</artifactId><version>2.1.4.RELEASE</version><relativePath/> <!-- lookup parent from repository --> </parent> <properties><project.build.sourceEncoding>UTF-8</project.build.sourceEncoding><java.version>1.8</java.version> </properties> <dependencies><!-- web --><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId></dependency> <dependency> <groupId>io.netty</groupId> <artifactId>netty-all</artifactId> <version>4.1.33.Final</version></dependency> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId></dependency><!-- fastjson --><dependency> <groupId>com.alibaba</groupId> <artifactId>fastjson</artifactId> <version>1.2.56</version></dependency><dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-thymeleaf</artifactId></dependency> </dependencies> <build><plugins> <plugin><groupId>org.springframework.boot</groupId><artifactId>spring-boot-maven-plugin</artifactId> </plugin></plugins> </build></project>二、創建netty 服務端

package com.chat.server; import io.netty.bootstrap.ServerBootstrap;import io.netty.channel.ChannelFuture;import io.netty.channel.EventLoopGroup;import io.netty.channel.nio.NioEventLoopGroup;import io.netty.channel.socket.nio.NioServerSocketChannel;import lombok.extern.slf4j.Slf4j;import org.springframework.stereotype.Component; import javax.annotation.PostConstruct;import javax.annotation.PreDestroy; @Component@Slf4jpublic class ChatServer { private EventLoopGroup bossGroup; private EventLoopGroup workGroup; private void run() throws Exception {log.info('開始啟動聊天服務器');bossGroup = new NioEventLoopGroup(1);workGroup = new NioEventLoopGroup();try { ServerBootstrap serverBootstrap = new ServerBootstrap(); serverBootstrap.group(bossGroup, workGroup) .channel(NioServerSocketChannel.class) .childHandler(new ChatServerInitializer()); //啟動服務器 ChannelFuture channelFuture = serverBootstrap.bind(7000).sync(); log.info('開始啟動聊天服務器結束'); channelFuture.channel().closeFuture().sync(); } finally { bossGroup.shutdownGracefully(); workGroup.shutdownGracefully();} } /** * 初始化服務器 */ @PostConstruct() public void init() {new Thread(() -> { try {run(); } catch (Exception e) {e.printStackTrace(); }}).start(); } @PreDestroy public void destroy() throws InterruptedException {if (bossGroup != null) { bossGroup.shutdownGracefully().sync();}if (workGroup != null) { workGroup.shutdownGracefully().sync();} }}

package com.chat.server; import io.netty.channel.ChannelInitializer;import io.netty.channel.ChannelPipeline;import io.netty.channel.socket.SocketChannel;import io.netty.handler.codec.http.HttpObjectAggregator;import io.netty.handler.codec.http.HttpServerCodec;import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;import io.netty.handler.stream.ChunkedWriteHandler; public class ChatServerInitializer extends ChannelInitializer<SocketChannel> { @Override protected void initChannel(SocketChannel socketChannel) throws Exception {ChannelPipeline pipeline = socketChannel.pipeline();//使用http的編碼器和解碼器pipeline.addLast(new HttpServerCodec());//添加塊處理器pipeline.addLast(new ChunkedWriteHandler()); pipeline.addLast(new HttpObjectAggregator(8192)); pipeline.addLast(new WebSocketServerProtocolHandler('/chat'));//自定義handler,處理業務邏輯pipeline.addLast(new ChatServerHandler()); }}

package com.chat.server; import com.alibaba.fastjson.JSON;import com.alibaba.fastjson.JSONObject;import com.chat.config.ChatConfig;import io.netty.channel.Channel;import io.netty.channel.ChannelHandlerContext;import io.netty.channel.SimpleChannelInboundHandler;import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;import io.netty.util.AttributeKey;import lombok.extern.slf4j.Slf4j; import java.time.LocalDateTime; @Slf4jpublic class ChatServerHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> { @Override protected void channelRead0(ChannelHandlerContext channelHandlerContext, TextWebSocketFrame textWebSocketFrame) throws Exception {//傳過來的是json字符串String text = textWebSocketFrame.text();JSONObject jsonObject = JSON.parseObject(text);//獲取到發送人的用戶idObject msg = jsonObject.get('msg');String userId = (String) jsonObject.get('userId');Channel channel = channelHandlerContext.channel();if (msg == null) { //說明是第一次登錄上來連接，還沒有開始進行聊天，將uid加到map里面 register(userId, channel);} else { //有消息了,開始聊天了 sendMsg(msg, userId);} } /** * 第一次登錄進來 * * @param userId * @param channel */ private void register(String userId, Channel channel) {if (!ChatConfig.concurrentHashMap.containsKey(userId)) { //沒有指定的userId ChatConfig.concurrentHashMap.put(userId, channel); // 將用戶ID作為自定義屬性加入到channel中，方便隨時channel中獲取用戶ID AttributeKey<String> key = AttributeKey.valueOf('userId'); channel.attr(key).setIfAbsent(userId);} } /** * 開發發送消息，進行聊天 * * @param msg * @param userId */ private void sendMsg(Object msg, String userId) {Channel channel1 = ChatConfig.concurrentHashMap.get(userId);if (channel1 != null) { channel1.writeAndFlush(new TextWebSocketFrame('服務器時間' + LocalDateTime.now() + ' ' + msg));} } /** * 一旦客戶端連接上來，該方法被執行 * * @param ctx * @throws Exception */ @Override public void handlerAdded(ChannelHandlerContext ctx) throws Exception {log.info('handlerAdded 被調用' + ctx.channel().id().asLongText()); } /** * 斷開連接，需要移除用戶 * * @param ctx * @throws Exception */ @Override public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {removeUserId(ctx); } /** * 移除用戶 * * @param ctx */ private void removeUserId(ChannelHandlerContext ctx) {Channel channel = ctx.channel();AttributeKey<String> key = AttributeKey.valueOf('userId');String userId = channel.attr(key).get();ChatConfig.concurrentHashMap.remove(userId);log.info('用戶下線,userId：{}', userId); } /** * 處理移除，關閉通道 * * @param ctx * @param cause * @throws Exception */ @Override public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {ctx.close(); }}三、存儲用戶channel 的map

package com.chat.config; import io.netty.channel.Channel; import java.util.concurrent.ConcurrentHashMap; public class ChatConfig { public static ConcurrentHashMap<String, Channel> concurrentHashMap = new ConcurrentHashMap(); }四、客戶端html

<!DOCTYPE html><html lang='en' xmlns:th='http://www.thymeleaf.org'><head> <meta charset='UTF-8'> <title>Title</title> <script>var socket;//判斷當前瀏覽器是否支持websocketif (window.WebSocket) { //go on socket = new WebSocket('ws://localhost:7000/chat'); //相當于channelReado, ev 收到服務器端回送的消息 socket.onmessage = function (ev) {var rt = document.getElementById('responseText');rt.value = rt.value + 'n' + ev.data; } //相當于連接開啟(感知到連接開啟) socket.onopen = function (ev) {var rt = document.getElementById('responseText');rt.value = '連接開啟了..'var userId = document.getElementById('userId').value;var myObj = {userId: userId};var myJSON = JSON.stringify(myObj);socket.send(myJSON) } //相當于連接關閉(感知到連接關閉) socket.onclose = function (ev) {var rt = document.getElementById('responseText');rt.value = rt.value + 'n' + '連接關閉了..' }} else { alert('當前瀏覽器不支持websocket')} //發送消息到服務器function send(message) { if (!window.socket) { //先判斷socket是否創建好return; } if (socket.readyState == WebSocket.OPEN) {//通過socket 發送消息var sendId = document.getElementById('sendId').value;var myObj = {userId: sendId, msg: message};var messageJson = JSON.stringify(myObj);socket.send(messageJson) } else {alert('連接沒有開啟'); }} </script></head><body><h1 th:text='${userId}'></h1><input type='hidden' th:value='${userId}' id='userId'><input type='hidden' th:value='${sendId}' id='sendId'><form onsubmit='return false'> <textarea name='message' style='height: 300px; width: 300px'></textarea> <input type='button' value='發送' onclick='send(this.form.message.value)'> <textarea style='height: 300px; width: 300px'></textarea> <input type='button' value='清空內容' onclick='document.getElementById(’responseText’).value=’’'></form></body></html>五、controller 模擬用戶登錄以及要發送信息給誰

package com.chat.controller; import com.chat.config.ChatConfig;import io.netty.channel.Channel;import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;import org.springframework.stereotype.Controller;import org.springframework.ui.Model;import org.springframework.web.bind.annotation.GetMapping;import org.springframework.web.bind.annotation.RequestParam;@Controllerpublic class ChatController { @GetMapping('login') public String login(Model model, @RequestParam('userId') String userId, @RequestParam('sendId') String sendId) {model.addAttribute('userId', userId);model.addAttribute('sendId', sendId);return 'chat'; } @GetMapping('sendMsg') public String login(@RequestParam('sendId') String sendId) throws InterruptedException {while (true) { Channel channel = ChatConfig.concurrentHashMap.get(sendId); if (channel != null) {channel.writeAndFlush(new TextWebSocketFrame('test'));Thread.sleep(1000); }} } }六、測試

登錄成功要發消息給bbb

登錄成功要發消息給aaa

到此這篇關于Java實戰之用springboot+netty實現簡單的一對一聊天的文章就介紹到這了,更多相關springboot+netty實現一對一聊天內容請搜索好吧啦網以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持好吧啦網！