nettybishuai-chat.zip

package com.mytest.processor; import com.alibaba.fastjson.JSONObject; import com.mytest.protocol.ChatDecoder; import com.mytest.protocol.ChatEncoder; import com.mytest.protocol.ChatMessage; import com.mytest.protocol.ChatType; import io.netty.channel.Channel; import io.netty.channel.group.ChannelGroup; import io.netty.channel.group.DefaultChannelGroup; import io.netty.handler.codec.http.websocketx.TextWebSocketFrame; import io.netty.util.AttributeKey; import io.netty.util.concurrent.GlobalEventExecutor; public class ChatMessageProcessor { //记录在线用户 private static ChannelGroup onlineUsers = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE); //定义一些扩展属性 private final AttributeKey<String> NICK_NAME = AttributeKey.valueOf("nickName"); private final AttributeKey<String> IP_ADDR = AttributeKey.valueOf("ipAddr"); private final AttributeKey<JSONObject> ATTRS = AttributeKey.valueOf("attrs"); //解码器 private ChatDecoder decoder = new ChatDecoder(); //编码器 private ChatEncoder encoder = new ChatEncoder(); /** * 获取用户昵称 * * @param client * @return */ public String getNickName(Channel client) { return client.attr(NICK_NAME).get(); } /** * 获取用户远程IP地址 * * @param client * @return */ public String getAddress(Channel client) { return client.remoteAddress().toString().replaceFirst("/", ""); } /** * 获取扩展属性 * * @param client * @return */ public JSONObject getAttrs(Channel client) { try { return client.attr(ATTRS).get(); } catch (Exception e) { return null; } } /** * 设置扩展属性 * * @param client * @return */ private void setAttrs(Channel client, String key, Object value) { try { JSONObject json = client.attr(ATTRS).get(); json.put(key, value); client.attr(ATTRS).set(json); } catch (Exception e) { JSONObject json = new JSONObject(); json.put(key, value); client.attr(ATTRS).set(json); } } /** * 登录退出通知 * * @param client */ public void logout(Channel client) { if (getNickName(client) == null) { return; } for (Channel channel : onlineUsers) { ChatMessage request = new ChatMessage(ChatType.SYSTEM.getName(), sysTime(), onlineUsers.size(), getNickName(client) + "离开"); String content = encoder.encode(request); channel.writeAndFlush(new TextWebSocketFrame(content)); } onlineUsers.remove(client); } /** * 发送消息 * * @param client * @param msg */ public void sendMsg(Channel client, ChatMessage msg) { sendMsg(client, encoder.encode(msg)); } /** * 发送消息 * * @param client * @param msg */ public void sendMsg(Channel client, String msg) { ChatMessage request = decoder.decode(msg); if (null == request) { return; } String addr = getAddress(client); //登录消息 if (request.getCmd().equals(ChatType.LOGIN.getName())) { client.attr(NICK_NAME).getAndSet(request.getSender()); client.attr(IP_ADDR).getAndSet(addr); onlineUsers.add(client); for (Channel channel : onlineUsers) { if (channel != client) { request = new ChatMessage(ChatType.SYSTEM.getName(), sysTime(), onlineUsers.size(), getNickName(client) + "加入"); } else { request = new ChatMessage(ChatType.SYSTEM.getName(), sysTime(), onlineUsers.size(), "已与服务器建立连接！"); } String content = encoder.encode(request); //把数据通过websocket协议写到浏览器 channel.writeAndFlush(new TextWebSocketFrame(content)); } //聊天消息 } else if (request.getCmd().equals(ChatType.CHAT.getName())) { for (Channel channel : onlineUsers) { if (channel == client) { request.setSender("你"); } else { request.setSender(getNickName(client)); } request.setTime(sysTime()); String content = encoder.encode(request); channel.writeAndFlush(new TextWebSocketFrame(content)); } //送鲜花消息 } else if (request.getCmd().equals(ChatType.FLOWER.getName())) { JSONObject attrs = getAttrs(client); long currTime = sysTime(); if (null != attrs) { long lastTime = attrs.getLongValue("lastFlowerTime"); //10秒之内不允许重复刷鲜花 int secends = 10; long sub = currTime - lastTime; if (sub < 1000 * secends) { request.setSender("你"); request.setCmd(ChatType.SYSTEM.getName()); request.setContent("送鲜花太频繁了, " + (secends - Math.round(sub / 1000)) + "秒后再试."); String content = encoder.encode(request); client.writeAndFlush(new TextWebSocketFrame(content)); return; } } //正常送花 for (Channel channel : onlineUsers) { if (channel == client) { request.setSender("你"); request.setContent("你给大家送了一波鲜花雨"); setAttrs(client, "lastFlowerTime", currTime); } else { request.setSender(getNickName(client)); request.setContent(getNickName(client) + "送来一波鲜花雨"); } request.setTime(sysTime()); String content = encoder.encode(request); channel.writeAndFlush(new TextWebSocketFrame(content)); } } } /** * 获取系统时间 * * @return */ private Long sysTime() { return System.currentTimeMillis(); } }