ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

基于WebSocket实现简单的扫码登录

2021-06-18 12:03:04  阅读:114  来源: 互联网

标签:map 扫码 WebSocket String 登录 session websocket public


基于WebSocket实现简单的扫码登录

    WebSocket是一种在单个TCP连接上进行全双工通信的协议。WebSocket通信协议于2011年被IETF定为标准RFC 6455,并由RFC7936补充规范。WebSocket API也被W3C定为标准。
    WebSocket使得客户端和服务器之间的数据交换变得更加简单,允许服务端主动向客户端推送数据。在WebSocket API中,浏览器和服务器只需要完成一次握手,两者之间就直接可以创建持久性的连接,并进行双向数据传输。
	因为是第一次接触,所以画了一个简单的流程图
	![流程图](https://www.icode9.com/i/ll/?i=2021061709594618.png?,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzMTU5NzYy,size_16,color_FFFFFF,t_70#pic_center)

开始编码

1、导包

		<dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-websocket</artifactId>
        </dependency>

2、开启WebSocket功能
发现网上很多文章都缺少这一步

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}

3、创建WebSocket端点

@Component
@ServerEndpoint(value = "/websocket")
public class WebSocketWxLogin {
​
​
    //用来存放每个连接对应的WebSocketWxLogin对象,适用于同时与多个客户端通信
    public static CopyOnWriteArraySet<WebSocketWxLogin> webSocketSet = new CopyOnWriteArraySet<WebSocketWxLogin>();
    //其中Key为连接唯一标识
    public static ConcurrentHashMap<String, WebSocketWxLogin> webSocketMap = new ConcurrentHashMap<String, WebSocketWxLogin>();
​
    private Session session;
​
​
    /**
     * 建立连接成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session)throws IOException {
        this.session = session;
        //我这里使用时间戳作为唯一id
        long l = System.currentTimeMillis();
        String str = l+"";
        String sessionId = TripleDES.encode(str);
        webSocketSet.add(this); 
        webSocketMap.put(sessionId,this);
        Map map = new HashMap();
        map.put("code",400);
        map.put("msg","获取成功!");
        map.put("sessionId",sessionId);
        String s = JSONObject.toJSONString(map);
        session.getBasicRemote().sendText(s);
    }
​
    /**
     * 关闭连接调用的方法
     */
    @OnClose
    public void onClose(Session closeSession){
        System.out.println(session);
        webSocketMap.remove(session);
        webSocketSet.remove(this);
    }
​
     /**
     * 向用户推送消息
     */
    public void sendMessage(String message) throws IOException {
        this.session.getBasicRemote().sendText(message);
    }
}
​

4、手机端接口

@PostMapping("login")
    @ApiOperation(value = "扫码登录", notes = "扫码登录")
    private Object login(String sessionId) {
        Assert.notNull(sessionId,"缺少参数");
        WebSocketWxLogin o1 = WebSocketWxLogin.webSocketMap.get(sessionId);
        Assert.notNull(o1,"二维码已过期");
        Assert.notNull(getUser(),"未登录!");
        String token = getRequest().getSession().getId();
        redisUtil.set(String.format(RedisParam.KEY_MOBILE_USER_LOGIN, token), getUser(), RedisParam.LOGIN_EXPIRE_TIME);
        Map map = new HashMap();
        map.put("code",401);
        map.put("msg","登录成功!");
        map.put("token",token);
        String s = JSONObject.toJSONString(map);
        try {
            o1.sendLogin(s);
        }catch (Exception e){
            return RespEntity.errorResp("登录失败!");
        }
        return RespEntity.successResp("登录成功!",token);
    }

5、前端

<script type="text/javascript">
var websocket  = new WebSocket("ws:127.0.0.1:8009/websocket");
//ws:服务器+端口/后端的端点
   //接收到消息的回调方法
    websocket.onmessage = function (event) {
        console.log(event.data);
    }
​
​
    //连接关闭的回调方法
    websocket.onclose = function () {
        console.log("WebSocket连接关闭");
    }
​
​
    //监听窗口关闭事件,当窗口关闭时,主动去关闭websocket连接,确保断开时关闭连接
    window.onbeforeunload = function () {
        closeSocket();
    }
​
    //关闭WebSocket连接
    function closeSocket() {
        websocket.close();
    }
</script>

代码可直接食用,在合适的地方添加业务代码即可。

第一次写博客,有什么不对的地方感谢大家指出,多谢!

标签:map,扫码,WebSocket,String,登录,session,websocket,public
来源: https://blog.csdn.net/qq_43159762/article/details/117703839

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有