102 lines
4.3 KiB
Java
102 lines
4.3 KiB
Java
|
package com.cf.chat.netty;
|
|||
|
|
|
|||
|
|
import com.alibaba.fastjson.JSON;
|
|||
|
|
import com.cf.chat.domain.CfUserMessage;
|
|||
|
|
import com.cf.chat.domain.Message;
|
|||
|
|
import com.cf.chat.service.CfUserGroupMessageService;
|
|||
|
|
import com.cf.chat.service.CfUserMessageService;
|
|||
|
|
import com.cf.framework.domain.message.response.MessageCode;
|
|||
|
|
import com.cf.framework.domain.ucenter.ext.UserBasicInfo;
|
|||
|
|
import com.cf.framework.domain.ucenter.response.AuthCode;
|
|||
|
|
import com.cf.framework.exception.ExceptionCast;
|
|||
|
|
import com.cf.framework.utils.SpringUtil;
|
|||
|
|
import io.netty.channel.Channel;
|
|||
|
|
import io.netty.channel.ChannelHandlerContext;
|
|||
|
|
import io.netty.channel.SimpleChannelInboundHandler;
|
|||
|
|
import io.netty.channel.group.ChannelGroup;
|
|||
|
|
import io.netty.channel.group.DefaultChannelGroup;
|
|||
|
|
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
|
|||
|
|
import io.netty.util.concurrent.GlobalEventExecutor;
|
|||
|
|
|
|||
|
|
import java.net.InetSocketAddress;
|
|||
|
|
import java.net.SocketAddress;
|
|||
|
|
import java.text.SimpleDateFormat;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理消息的handler
|
|||
|
|
* TextWebSocketFrame: 在netty中，是用于为websocket专门处理文本的对象，frame是消息的载体
|
|||
|
|
*/
|
|||
|
|
public class ChatHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {
|
|||
|
|
|
|||
|
|
// 用来保存所有的客户端连接
|
|||
|
|
private static ChannelGroup clients = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
|
|||
|
|
private SimpleDateFormat sdf = new SimpleDateFormat("yyyy-mm-dd hh:MM");
|
|||
|
|
|
|||
|
|
// 当Channel中有新的事件消息会自动调用
|
|||
|
|
protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
|
|||
|
|
// 当接收到数据后会自动调用
|
|||
|
|
|
|||
|
|
// 获取客户端发送过来的文本消息
|
|||
|
|
Message message = JSON.parseObject(msg.text(), Message.class);
|
|||
|
|
|
|||
|
|
// 通过SpringUtil工具类获取Spring上下文容器
|
|||
|
|
CfUserMessageService cfUserMessageService = SpringUtil.getBean(CfUserMessageService.class);
|
|||
|
|
CfUserGroupMessageService cfUserGroupMessageService = SpringUtil.getBean(CfUserGroupMessageService.class);
|
|||
|
|
UserBasicInfo userBasicInfo = cfUserMessageService.getBaseUserInfoByToken(message.getToken());
|
|||
|
|
switch (message.getType()) {
|
|||
|
|
case "create_link":
|
|||
|
|
UserChannelMap.put(userBasicInfo.getId(), ctx.channel());
|
|||
|
|
UserChannelMap.print();
|
|||
|
|
Thread messageThread = new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
cfUserMessageService.getNotReadMessage(userBasicInfo.getId());
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
messageThread.start();
|
|||
|
|
Thread groupMessageThread = new Thread(new Runnable() {
|
|||
|
|
@Override
|
|||
|
|
public void run() {
|
|||
|
|
cfUserGroupMessageService.getNotReadMessage(userBasicInfo.getId());
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
groupMessageThread.start();
|
|||
|
|
break;
|
|||
|
|
case "send_message":
|
|||
|
|
CfUserMessage _message = message.getCfUserMessage();
|
|||
|
|
_message.setFromUid(userBasicInfo.getId());
|
|||
|
|
SocketAddress socketAddress = ctx.channel().remoteAddress();
|
|||
|
|
String ip = socketAddress.toString();
|
|||
|
|
ip = ip.substring(1,ip.indexOf(":"));
|
|||
|
|
_message.setIp(ip);
|
|||
|
|
cfUserMessageService.sendMessage(_message);
|
|||
|
|
break;
|
|||
|
|
case "change_status":
|
|||
|
|
cfUserMessageService.updateStatus(userBasicInfo.getId(),
|
|||
|
|
message.getCfUserMessage().getId(),1);
|
|||
|
|
break;
|
|||
|
|
case "heartbeat_check":
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 当有新的客户端连接服务器之后，会自动调用这个方法
|
|||
|
|
@Override
|
|||
|
|
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
|
|||
|
|
// 将新的通道加入到clients
|
|||
|
|
clients.add(ctx.channel());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Override
|
|||
|
|
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
|
|||
|
|
UserChannelMap.removeByChannelId(ctx.channel().id().asLongText());
|
|||
|
|
ctx.channel().close();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Override
|
|||
|
|
public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
|
|||
|
|
System.out.println("关闭通道");
|
|||
|
|
UserChannelMap.removeByChannelId(ctx.channel().id().asLongText());
|
|||
|
|
UserChannelMap.print();
|
|||
|
|
}
|
|||
|
|
}
|