使用netty實現im聊天

簡書地址圖文更清晰： https://www.jianshu.com/p/f455814f3c40

1、新建maven工程
2、引入maven依賴

<dependencies> <dependency> <groupId>io.netty</groupId> <artifactId>netty-all</artifactId> <version>4.1.49.Final</version> </dependency> <dependency> <groupId>com.alibaba</groupId> <artifactId>fastjson</artifactId> <version>1.2.68</version> </dependency> </dependencies> <build> <plugins> <plugin> <groupId>org.apache.maven.plugins</groupId> <artifactId>maven-compiler-plugin</artifactId> <version>3.6.1</version> <configuration> <source>1.8</source> <target>1.8</target> </configuration> </plugin> </plugins> </build> 

3、定義消息體MsgBody

public class MsgBody { //發送人名稱 private String sendUserName; private String msg; public String getSendUserName() { return sendUserName; } public void setSendUserName(String sendUserName) { this.sendUserName = sendUserName; } public String getMsg() { return msg; } public void setMsg(String msg) { this.msg = msg; } } 

4、新建服務器端的NettyServer和ServerHandler

/** * netty的服務器 * @Author: yeyongjian * @Date: 2020-05-03 23:34 */ public class NettyServer { private int port; public NettyServer(int port) { this.port = port; bind(); } private void bind() { EventLoopGroup boss = new NioEventLoopGroup(); EventLoopGroup worker = new NioEventLoopGroup(); try { ServerBootstrap bootstrap = new ServerBootstrap(); bootstrap.group(boss, worker); bootstrap.channel(NioServerSocketChannel.class); bootstrap.option(ChannelOption.SO_BACKLOG, 1024); // 連接數 bootstrap.option(ChannelOption.TCP_NODELAY, true); // 不延遲，消息立即發送 bootstrap.childOption(ChannelOption.SO_KEEPALIVE, true); // 長連接 bootstrap.childHandler(new ChannelInitializer<SocketChannel>() { @Override protected void initChannel(SocketChannel socketChannel){ ChannelPipeline p = socketChannel.pipeline(); ServerHandler serverHandler = new ServerHandler(); p.addLast(serverHandler);// 添加NettyServerHandler，用來處理Server端接收和處理消息的邏輯 } }); ChannelFuture channelFuture = bootstrap.bind(port).sync(); if (channelFuture.isSuccess()) { System.err.println("啟動Netty服務成功，端口號：" + this.port); } // 關閉連接 channelFuture.channel().closeFuture().sync(); } catch (Exception e) { System.err.println("啟動Netty服務異常，異常信息：" + e.getMessage()); e.printStackTrace(); } finally { boss.shutdownGracefully(); worker.shutdownGracefully(); } } public static void main(String[] args) throws InterruptedException { new NettyServer(10086); } } 

import com.alibaba.fastjson.JSONObject; import com.eujian.im.MsgBody; import io.netty.buffer.ByteBuf; import io.netty.buffer.Unpooled; import io.netty.channel.Channel; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelId; import io.netty.channel.SimpleChannelInboundHandler; import java.io.UnsupportedEncodingException; import java.util.HashMap; import java.util.Map; /** * 服務器的處理器 * @Author: yeyongjian * @Date: 2020-05-03 23:35 */ public class ServerHandler extends SimpleChannelInboundHandler { //連接id與容器的關系 private static Map<String, ChannelHandlerContext> map = new HashMap<>(); @Override protected void channelRead0(ChannelHandlerContext ctx, Object msg) { Channel channel = ctx.channel(); ChannelId id = channel.id(); map.put(id.toString(),ctx); ByteBuf buf = (ByteBuf) msg; String recieved = getMessage(buf); MsgBody msgBody = JSONObject.parseObject(recieved, MsgBody.class); String format = String.format("服務器接收到客戶端消息，發送人：%s，發送信息：%s", msgBody.getSendUserName(), msgBody.getMsg()); System.err.println(format); map.forEach((k,v)->{ try { if(id.toString().equals(k)){ return; } MsgBody sendMsgBody = new MsgBody(); sendMsgBody.setSendUserName(msgBody.getSendUserName()); sendMsgBody.setMsg(msgBody.getMsg()); v.writeAndFlush(getSendByteBuf(JSONObject.toJSONString(sendMsgBody))); System.err.println("服務器回復消息："+JSONObject.toJSONString(sendMsgBody)); } catch (UnsupportedEncodingException e) { e.printStackTrace(); } }); } /* * 從ByteBuf中獲取信息 使用UTF-8編碼返回 */ private String getMessage(ByteBuf buf) { byte[] con = new byte[buf.readableBytes()]; buf.readBytes(con); try { return new String(con, "UTF8"); } catch (UnsupportedEncodingException e) { e.printStackTrace(); return null; } } private ByteBuf getSendByteBuf(String message) throws UnsupportedEncodingException { byte[] req = message.getBytes("UTF-8"); ByteBuf pingMessage = Unpooled.buffer(); pingMessage.writeBytes(req); return pingMessage; } } 

5、新建客戶端代碼

import com.alibaba.fastjson.JSONObject; import com.eujian.im.MsgBody; import io.netty.buffer.ByteBuf; import io.netty.buffer.Unpooled; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.SimpleChannelInboundHandler; import java.io.UnsupportedEncodingException; public class NettyClientHandler extends SimpleChannelInboundHandler { private ByteBuf firstMessage; private ChannelHandlerContext ctx; private String userName; public String getUserName() { return userName; } public void setUserName(String userName) { this.userName = userName; } public void sendMsg(String str){ byte[] data = str.getBytes(); firstMessage = Unpooled.buffer(); firstMessage.writeBytes(data); ctx.writeAndFlush(firstMessage); } @Override public void channelActive(ChannelHandlerContext ctx) { this.ctx= ctx; MsgBody msgBody = new MsgBody(); msgBody.setSendUserName(userName); msgBody.setMsg("進入聊天室"); byte[] data = JSONObject.toJSONString(msgBody).getBytes(); firstMessage = Unpooled.buffer(); firstMessage.writeBytes(data); ctx.writeAndFlush(firstMessage); } @Override protected void channelRead0(ChannelHandlerContext channelHandlerContext, Object msg){ ByteBuf buf = (ByteBuf) msg; String rev = getMessage(buf); MsgBody msgBody = JSONObject.parseObject(rev, MsgBody.class); String format = String.format("客戶端收到服務器消息，發送人：%s，發送信息：%s", msgBody.getSendUserName(), msgBody.getMsg()); System.err.println(format); } private String getMessage(ByteBuf buf) { byte[] con = new byte[buf.readableBytes()]; buf.readBytes(con); try { return new String(con, "UTF8"); } catch (UnsupportedEncodingException e) { e.printStackTrace(); return null; } } } 

import com.alibaba.fastjson.JSONObject; import com.eujian.im.MsgBody; import io.netty.bootstrap.Bootstrap; import io.netty.channel.ChannelFuture; import io.netty.channel.ChannelInitializer; import io.netty.channel.ChannelOption; import io.netty.channel.EventLoopGroup; import io.netty.channel.nio.NioEventLoopGroup; import io.netty.channel.socket.SocketChannel; import io.netty.channel.socket.nio.NioSocketChannel; import java.util.Scanner; public class NettyClient { public NettyClientHandler nettyClientHandler; /* * 服務器端口號 */ private int port; private String sendUserName; /* * 服務器IP */ private String host; public NettyClient(int port, String host, String sendUserName) throws InterruptedException { this.port = port; this.host = host; this.sendUserName = sendUserName; start(); } private void start() throws InterruptedException { EventLoopGroup eventLoopGroup = new NioEventLoopGroup(); try { Bootstrap bootstrap = new Bootstrap(); bootstrap.channel(NioSocketChannel.class); bootstrap.option(ChannelOption.SO_KEEPALIVE, true); bootstrap.group(eventLoopGroup); bootstrap.remoteAddress(host, port); bootstrap.handler(new ChannelInitializer<SocketChannel>() { @Override protected void initChannel(SocketChannel socketChannel) throws Exception { nettyClientHandler = new NettyClientHandler(); nettyClientHandler.setUserName(sendUserName); socketChannel.pipeline().addLast(nettyClientHandler); } }); ChannelFuture channelFuture = bootstrap.connect(host, port).sync(); if (channelFuture.isSuccess()) { new Thread(new Runnable() { @Override public void run() { Scanner sc = new Scanner(System.in); while (sc.hasNext()){ MsgBody msgBody = new MsgBody(); msgBody.setSendUserName(sendUserName); msgBody.setMsg(sc.next()); nettyClientHandler.sendMsg(JSONObject.toJSONString(msgBody)); } } }).start(); System.err.println(sendUserName+"連接服務器成功"); } channelFuture.channel().closeFuture().sync(); } finally { eventLoopGroup.shutdownGracefully(); } } } 

6、新建2個main函數，模擬兩個客戶端

    public static void main(String[] args) throws InterruptedException {
        new NettyClient(10086, "localhost","tom");
    }
}

    public static void main(String[] args) throws InterruptedException {
        new NettyClient(10086, "localhost","jack");
    }
}

7、啟動nettyServer，main1和main2
在main1輸入 jack,你好，
在main2輸入 tom,hello，我很好
server顯示

 

 

 

main1顯示

 

 

 

main2顯示

 

github代碼： https://github.com/hd-eujian/netty-im.git

碼雲： https://gitee.com/guoeryyj/netty-im.git

歡迎關注我的微信公眾號：進階者euj