客户端互相通信

Author: coderbruis

Spring-Netty/src/main/java/com/bruis/learnnetty/im/client/NettyClient.java

@@ -1,13 +1,13 @@
 packagecom.bruis.learnnetty.im.client;
 
-importcom.bruis.learnnetty.im.client.handler.FirstClientHandler;
 importcom.bruis.learnnetty.im.client.handler.LoginResponseHandler;
 importcom.bruis.learnnetty.im.client.handler.MessageResponseHandler;
 importcom.bruis.learnnetty.im.codec.PacketDecoder;
 importcom.bruis.learnnetty.im.codec.PacketEncoder;
 importcom.bruis.learnnetty.im.codec.Spliter;
+importcom.bruis.learnnetty.im.model.LoginRequestPacket;
 importcom.bruis.learnnetty.im.model.MessageRequestPacket;
-importcom.bruis.learnnetty.im.util.LoginUtil;
+importcom.bruis.learnnetty.im.util.SessionUtil;
 importio.netty.bootstrap.Bootstrap;
 importio.netty.channel.Channel;
 importio.netty.channel.ChannelFuture;
@@ -45,13 +45,15 @@ public static void main(String[] args){
  .handler(newChannelInitializer<SocketChannel>(){
 @Override
 publicvoidinitChannel(SocketChannelch){
+// 拆包粘包处理
 ch.pipeline().addLast(newSpliter());
-// ch.pipeline().addLast(new FirstClientHandler());
-// 解码
+// 编码
 ch.pipeline().addLast(newPacketDecoder());
+// 登录响应
 ch.pipeline().addLast(newLoginResponseHandler());
+// 消息返回
 ch.pipeline().addLast(newMessageResponseHandler());
-// 编码
+// 解码
 ch.pipeline().addLast(newPacketEncoder());
  }
  });
@@ -80,16 +82,35 @@ private static void connect(Bootstrap bootstrap, String host, int port, int retr
  }
 
 privatestaticvoidstartConsoleThread(Channelchannel){
+Scannersc = newScanner(System.in);
+LoginRequestPacketloginRequestPacket = newLoginRequestPacket();
+
 newThread(() ->{
 while (!Thread.interrupted()){
-if (LoginUtil.hasLogin(channel)){
-System.out.println("输入消息发送至服务端: ");
-Scannersc = newScanner(System.in);
-Stringline = sc.nextLine();
+if (!SessionUtil.hasLogin(channel)){
+System.out.print("输入用户名登录: ");
+Stringusername = sc.nextLine();
+loginRequestPacket.setUserName(username);
+
+// 密码使用默认的
+loginRequestPacket.setPassword("pwd");
 
-channel.writeAndFlush(newMessageRequestPacket(line));
+// 发送登录数据包
+channel.writeAndFlush(loginRequestPacket);
+waitForLoginResponse();
+ } else{
+StringtoUserId = sc.next();
+Stringmessage = sc.next();
+channel.writeAndFlush(newMessageRequestPacket(toUserId, message));
  }
  }
  }).start();
  }
+
+privatestaticvoidwaitForLoginResponse(){
+try{
+Thread.sleep(1000);
+ } catch (InterruptedExceptionignored){
+ }
+ }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/client/handler/FirstClientHandler.java

@@ -1,13 +1,12 @@
 packagecom.bruis.learnnetty.im.client.handler;
 
-importcom.bruis.learnnetty.im.model.LoginRequestPacket;
 importcom.bruis.learnnetty.im.model.LoginResponsePacket;
-importcom.bruis.learnnetty.im.util.LoginUtil;
+importcom.bruis.learnnetty.im.session.Session;
+importcom.bruis.learnnetty.im.util.SessionUtil;
 importio.netty.channel.ChannelHandlerContext;
 importio.netty.channel.SimpleChannelInboundHandler;
 
 importjava.util.Date;
-importjava.util.UUID;
 
 /**
  * @Description 登录响应的reponse
@@ -17,24 +16,20 @@
 publicclassLoginResponseHandlerextendsSimpleChannelInboundHandler<LoginResponsePacket>{
 
 @Override
-publicvoidchannelActive(ChannelHandlerContextctx) throwsException{
-// 创建登录对象
-LoginRequestPacketloginRequestPacket = newLoginRequestPacket();
-loginRequestPacket.setUserId(UUID.randomUUID().toString());
-loginRequestPacket.setUserName("flash");
-loginRequestPacket.setPassword("pwd");
+protectedvoidchannelRead0(ChannelHandlerContextctx, LoginResponsePacketloginResponsePacket) throwsException{
+StringuserId = loginResponsePacket.getUserId();
+StringuserName = loginResponsePacket.getUserName();
 
-// 写数据-发起登录
-ctx.channel().writeAndFlush(loginRequestPacket);
- }
-
-@Override
-protectedvoidchannelRead0(ChannelHandlerContextchannelHandlerContext, LoginResponsePacketloginResponsePacket) throwsException{
 if (loginResponsePacket.isSuccess()){
 System.out.println(newDate() + ": 客户端登录成功");
-LoginUtil.markAsLogin(channelHandlerContext.channel());
+SessionUtil.bindSession(newSession(userId, userName), ctx.channel());
  } else{
 System.out.println(newDate() + ": 客户端登录失败，原因：" + loginResponsePacket.getReason());
  }
  }
+
+@Override
+publicvoidchannelInactive(ChannelHandlerContextctx) throwsException{
+System.out.println("客户端连接被关闭");
+ }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/client/handler/LoginResponseHandler.java

@@ -4,8 +4,6 @@
 importio.netty.channel.ChannelHandlerContext;
 importio.netty.channel.SimpleChannelInboundHandler;
 
-importjava.util.Date;
-
 /**
  * @Description
  * @Author luohaiyang
@@ -14,6 +12,8 @@
 publicclassMessageResponseHandlerextendsSimpleChannelInboundHandler<MessageResponsePacket>{
 @Override
 protectedvoidchannelRead0(ChannelHandlerContextchannelHandlerContext, MessageResponsePacketmessageResponsePacket) throwsException{
-System.out.println(newDate() + ": 收到服务端的消息: " + messageResponsePacket.getMessage());
+StringfromUserId = messageResponsePacket.getFromUserId();
+StringfromUserName = messageResponsePacket.getFromUserName();
+System.out.println(fromUserId + ":" + fromUserName + " -> " + messageResponsePacket.getMessage());
  }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/client/handler/MessageResponseHandler.java

@@ -6,7 +6,7 @@
 importio.netty.handler.codec.LengthFieldBasedFrameDecoder;
 
 /**
- * @Description
+ * @Description 拆包、粘包处理
  * @Author luohaiyang
  * @Date 2022/3/23
  */
@@ -20,11 +20,11 @@ public Spliter(){
 
 @Override
 protectedObjectdecode(ChannelHandlerContextctx, ByteBufin) throwsException{
+// 校验协议
 if (in.getInt(in.readerIndex()) != PacketCodeC.MAGIC_NUMBER){
 ctx.channel().close();
 returnnull;
  }
-
 returnsuper.decode(ctx, in);
  }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/codec/Spliter.java

@@ -1,5 +1,6 @@
 packagecom.bruis.learnnetty.im.model;
 
+importcom.bruis.learnnetty.im.session.Session;
 importio.netty.util.AttributeKey;
 
 /**
@@ -8,5 +9,5 @@
  * @Date 2022/3/22
  */
 publicinterfaceAttributes{
-AttributeKey<Boolean> LOGIN = AttributeKey.newInstance("login");
+AttributeKey<Session> SESSION = AttributeKey.newInstance("session");
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/model/Attributes.java

@@ -8,6 +8,11 @@
  * @Date 2022/3/22
  */
 publicclassLoginResponsePacketextendsPacket{
+
+privateStringuserId;
+
+privateStringuserName;
+
 privatebooleansuccess;
 
 privateStringreason;
@@ -33,4 +38,20 @@ public String getReason(){
 publicvoidsetReason(Stringreason){
 this.reason = reason;
  }
+
+publicStringgetUserId(){
+returnuserId;
+ }
+
+publicvoidsetUserId(StringuserId){
+this.userId = userId;
+ }
+
+publicStringgetUserName(){
+returnuserName;
+ }
+
+publicvoidsetUserName(StringuserName){
+this.userName = userName;
+ }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/model/LoginResponsePacket.java

@@ -9,14 +9,17 @@
  */
 publicclassMessageRequestPacketextendsPacket{
 
+privateStringtoUserId;
+
+privateStringmessage;
+
 publicMessageRequestPacket(){}
 
-publicMessageRequestPacket(Stringmessage){
+publicMessageRequestPacket(StringtoUserId, Stringmessage){
+this.toUserId = toUserId;
 this.message = message;
  }
 
-privateStringmessage;
-
 @Override
 publicBytegetCommand(){
 returnMESSAGE_REQUEST;
@@ -29,4 +32,12 @@ public String getMessage(){
 publicvoidsetMessage(Stringmessage){
 this.message = message;
  }
+
+publicStringgetToUserId(){
+returntoUserId;
+ }
+
+publicvoidsetToUserId(StringtoUserId){
+this.toUserId = toUserId;
+ }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/model/MessageRequestPacket.java

@@ -9,6 +9,10 @@
  */
 publicclassMessageResponsePacketextendsPacket{
 
+privateStringfromUserId;
+
+privateStringfromUserName;
+
 privateStringmessage;
 
 @Override
@@ -24,4 +28,20 @@ public String getMessage(){
 publicvoidsetMessage(Stringmessage){
 this.message = message;
  }
+
+publicStringgetFromUserId(){
+returnfromUserId;
+ }
+
+publicvoidsetFromUserId(StringfromUserId){
+this.fromUserId = fromUserId;
+ }
+
+publicStringgetFromUserName(){
+returnfromUserName;
+ }
+
+publicvoidsetFromUserName(StringfromUserName){
+this.fromUserName = fromUserName;
+ }
 }

Spring-Netty/src/main/java/com/bruis/learnnetty/im/model/MessageResponsePacket.java

@@ -3,7 +3,7 @@
 importcom.bruis.learnnetty.im.codec.PacketDecoder;
 importcom.bruis.learnnetty.im.codec.PacketEncoder;
 importcom.bruis.learnnetty.im.codec.Spliter;
-importcom.bruis.learnnetty.im.server.handler.FirstServerHandler;
+importcom.bruis.learnnetty.im.server.handler.AuthHandler;
 importcom.bruis.learnnetty.im.server.handler.LoginRequestHandler;
 importcom.bruis.learnnetty.im.server.handler.MessageRequestHandler;
 importio.netty.bootstrap.ServerBootstrap;
@@ -41,6 +41,7 @@ protected void initChannel(NioSocketChannel ch) throws Exception{
 // ch.pipeline().addLast(new FirstServerHandler());
 ch.pipeline().addLast(newPacketDecoder());
 ch.pipeline().addLast(newLoginRequestHandler());
+ch.pipeline().addLast(newAuthHandler());
 ch.pipeline().addLast(newMessageRequestHandler());
 ch.pipeline().addLast(newPacketEncoder());
  }