add websocket

Helius

2021-05-27 182c01b109ffb3b4be248c2e128835ace70eeae8

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package com.xcong.excoin.netty.logic;
 
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import cn.hutool.crypto.asymmetric.RSA;
import com.alibaba.fastjson.JSONObject;
import com.xcong.excoin.common.contants.AppContants;
import com.xcong.excoin.configurations.properties.SecurityProperties;
import com.xcong.excoin.netty.bean.ChatRequest;
import com.xcong.excoin.netty.bean.RequestBean;
import com.xcong.excoin.netty.bean.ResponseBean;
import com.xcong.excoin.netty.common.ChannelManager;
import com.xcong.excoin.netty.common.NettyTools;
import com.xcong.excoin.rabbit.producer.ChatProducer;
import com.xcong.excoin.utils.RedisUtils;
import io.netty.channel.Channel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
 
 
/**
 * @author wzy
 * @email wangdoubleone@gmail.com
 * @date 2019-05-09
 */
@Slf4j
@Component
public class WebSocketLogic {
 
    @Autowired
    private RedisUtils redisUtils;
    @Autowired
    private ChatProducer chatProducer;
    @Autowired
    private SecurityProperties securityProperties;
 
    public void authCheck(RequestBean requestBean) {
        Channel channel = ChannelManager.findWebSocketChannel(requestBean.getChannelId());
        ResponseBean responseBean = new ResponseBean();
        responseBean.setType(requestBean.getType());
        responseBean.setStatus(1);
 
//        String bearerToken = requestBean.getData().toString();
//        String rsaToken = bearerToken.replace(AppContants.TOKEN_START_WITH, "");
//        RSA rsa = new RSA(securityProperties.getPrivateKey(), null);
//        String[] tokens = StrUtil.split(rsa.decryptStr(rsaToken, KeyType.PrivateKey), "_");
//
//        Long memberId = Long.parseLong(tokens[0]);
 
 
        Long memberId = Long.parseLong(requestBean.getData().toString());
 
        channel.writeAndFlush(NettyTools.webSocketBytes(JSONObject.toJSONString(responseBean)));
        ChannelManager.addWsChannel(channel, memberId);
        String value = redisUtils.getString(AppContants.MSG_NOTICE + memberId);
        if (StrUtil.isNotBlank(value)) {
            responseBean.setType(3);
            responseBean.setData(Integer.parseInt(value));
            channel.writeAndFlush(NettyTools.webSocketBytes(JSONObject.toJSONString(responseBean)));
 
            redisUtils.del(AppContants.MSG_NOTICE + memberId);
        }
    }
 
    public void sendMsg(RequestBean requestBean) {
        String chatStr = requestBean.getData().toString();
        ChatRequest chat = JSONObject.parseObject(chatStr, ChatRequest.class);
 
        // 判断是否在线
        Channel targetChannel = ChannelManager.findWsChannel(Long.parseLong(chat.getTo()));
        if (targetChannel != null) {
            targetChannel.writeAndFlush(NettyTools.webSocketBytes(JSONObject.toJSONString(ResponseBean.ok(chat))));
 
            chatProducer.sendMsgHistory(chat);
        } else {
            // 在redis中保存用户未在线时，给该用户发送的消息条数
            String key = AppContants.MSG_NOTICE + chat.getTo();
            String value = redisUtils.getString(key);
            if (StrUtil.isEmpty(value)) {
                redisUtils.set(key , 1);
            } else {
                redisUtils.set(key, Integer.parseInt(value) + 1);
            }
        }
    }
}