方案优化：网站实现扫描二维码关注微信公众号，自动登陆网站并获取其信息

Posted 2020-10-14 陈明羽

上一篇  《网站实现扫描二维码关注微信公众号，自动登陆网站并获取其信息》 中已经实现用户扫码登陆网站并获取其信息

但是上一篇方案中存在一个问题，也就是文章末尾指出的可以优化的地方（可以点击这个链接去看一下上篇文章）

首先回顾一下上一篇的思路：

 　　1，微信的系统，提供生成带参数的二维码的接口，这个参数就是唯一值（场景值）
 　　2，网站调用微信系统，获取生成的二维码图片
 　　3，用户扫码会直接调用微信服务器，将用户访问微信服务器的信息记录到redis，key就是唯一值（场景值）
 　　4，网站端做轮训去查询redis中是否有这个唯一值的数据，如果有就获取用户信息登录，没有就五秒一次轮训，登录后就不在做轮训（从二维码弹出之后开始做轮训，关闭二维码后停止轮训）

　　 5，这里的唯一值是可以自己定义的，我用的是截取了几位的时间戳 

 上述方案中，实现的难点在于微信服务器不能主动与网站进行通信，所以采用了网页端的轮训去做进行用户状态的查询，但是方案中4，网站端使用轮训去全局的redis缓存中查找用户信息，这样使用轮训无疑就会产生跟多的无用http请求，不仅增加了服务器的压力还浪费资源。

优化方案

　　 简化一下需要完成上述功能的实现要点：

　　　　客户端（网站）与服务端（微信）的通信，并且需要服务器可以主动的推送消息到客户端（网站）

 

　　要是实现客户端与服务器端的通信这个功能，常用的方式有两种：

　　1，轮训（也就是上次方案中使用的方法）

　　2，长连接

 

　我使用的是利用websocket与服务器进行通信，而且websocket使用起来也非常方便，易上手

　优化后的方案：

　　 1，微信的系统，提供生成带参数的二维码的接口，这个参数就是唯一值（场景值）
 　　2，网站调用微信系统，获取生成的二维码图片
 　　3，用户扫码会直接调用微信服务器，然后将获取到的用户信息利用websocket发送给网站端
 　　4，网站接受信息，并显示登陆

 

实现步骤

　　1，首先打开一个页面首先需要与服务器端建立一个socket链接，要想每个页面与服务器端都建立一个链接，每个页面都需要一个唯一标识，这里我用的时间戳

　　使用websocket建立一个连接很容易，代码如下

　　　　

    var timestamp = new Date().getTime() + "";
     timestamp = timestamp.substring(0, timestamp.length-3);   
    var websocket = null;
    //判断当前浏览器是否支持WebSocket
    if (\'WebSocket\' in window) {
        websocket = new WebSocket("ws://xxxx/wx/websocket/"+timestamp);
    }
    else {
        alert(\'当前浏览器  Not support websocket\');
    }

    //连接发生错误的回调方法
    websocket.onerror = function () {

        console.log("WebSocket连接发生错误");
    };

    //连接成功建立的回调方法
    websocket.onopen = function () {
        
        console.log("WebSocket连接成功");
    }

    //接收到消息的回调方法
    websocket.onmessage = function (event) {
                jQuery("#username").html(event.data);
        jQuery("#singup").css("display", "none");
        jQuery("#user").show();
        jQuery("#singout").show();
        layer.closeAll();
    }

    //连接关闭的回调方法
    websocket.onclose = function () {

        console.log("WebSocket连接关闭");
    }

    //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
    window.onbeforeunload = function () {
        closeWebSocket();
    }

    //关闭WebSocket连接
    function closeWebSocket() {
        websocket.close();
    }
        

　　2，服务器端socket的建立，在建立socket链接的时候，将每个页面传过来的唯一标识{sceneId}与会话信息session进行绑定，可以使用map实现，这里用的ConcurrentMap

/**
 * @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
 * 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 */
@Component
@ServerEndpoint("/websocket/{sceneId}")
public class WebSocketController {
    
    //静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
    private static int onlineCount = 0;

    //concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
    //private static CopyOnWriteArraySet<WebSocketController> webSocketSet = new CopyOnWriteArraySet<WebSocketController>();
    private static ConcurrentMap<String,WebSocketController>  webSocketMap = new ConcurrentHashMap<String,WebSocketController>();
    //与某个客户端的连接会话，需要通过它来给客户端发送数据
    private Session session;

    /**
     * 连接建立成功调用的方法
     * @param session  可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
     */
    @OnOpen
    public void onOpen(@PathParam("sceneId") String sceneId,Session session){
        this.session = session;
        webSocketMap.put(sceneId, this);
         //在线数加1
        System.out.println("唯一key为：" + sceneId);
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(@PathParam("sceneId") String sceneId){
        webSocketMap.remove(sceneId);//从map中删除
    }

    /**
     * 收到客户端消息后调用的方法
     * @param message 客户端发送过来的消息
     * @param session 可选的参数
     */
    @OnMessage
    public void onMessage(String message,Session session) {
        System.out.println("来自客户端的消息:" + message);

        JSONObject jsonobject = JSONObject.fromObject(message);
        Hashtable params= (Hashtable)JSONObject.toBean(jsonobject,Hashtable.class);
        
        //群发消息
        WebSocketController webSocketController = webSocketMap.get(params.get("equipmentType"));
        try {
            webSocketController.sendMessage((String)params.get("nickname"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 发生错误时调用
     * @param session
     * @param error
     */
    @OnError
    public void onError(Session session, Throwable error){
        System.out.println("发生错误");
        error.printStackTrace();
    }

    /**
     * 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。
     * @param message
     * @throws IOException
     */
    public void sendMessage(String message) throws IOException{
        this.session.getBasicRemote().sendText(message);
        //this.session.getAsyncRemote().sendText(message);
    }

3，微信端获取到用户信息后，调用socket，通过传过来的sceneId去map中获取session，之后将用户信息发送到网站

// 事件推送
            else if (msgType.equals(MessageUtil.REQ_MESSAGE_TYPE_EVENT)) {
                // 事件类型
                String eventType = requestMap.get("Event");
                // 关注
                if (eventType.equals(MessageUtil.EVENT_TYPE_SUBSCRIBE)) {
                  
                    String requestUrl = "https://api.weixin.qq.com/cgi-bin/user/info?access_token=ACCESS_TOKEN&openid=OPENID";
                    requestUrl = requestUrl.replace("ACCESS_TOKEN", tokenService.getToken())
                            .replace("OPENID", fromUserName);
                    JSONObject jsonObject = CommonUtil.httpsRequest(requestUrl, "GET",null);
                    String nickname = jsonObject.getString("nickname");
                    String address = jsonObject.getString("country")+"-"+jsonObject.getString("province")+"-"+jsonObject.getString("city");
                    String headimgurl = jsonObject.getString("headimgurl");
                    Hashtable params = new Hashtable();
                    params.put("phoneIme", fromUserName);
                    params.put("state", 1);
                    params.put("location", address);
                    params.put("realName", nickname);
                    
                    if(StringUtils.isNotBlank(eventKey)){
                        params.put("equipmentType", eventKey.replace("qrscene_", ""));
                        HttpUtil.postRequest(Constant.getValue("UPDATE_USER"), params);
                        //放到redis缓存中
                        //redisCacheTool.setDataToRedis(eventKey.replace("qrscene_", ""), 3600, params);

                        params.put("nickname", nickname);
                        JSONObject object = JSONObject.fromObject(params);
　　　　　　　　　　　　　　 //调用socket进行通信
                        webSocketController.onMessage(object.toString(), null);
                        LOGGER.info("qrcode redis key ==> "+eventKey.replace("qrscene_", ""));

                    }else{
                        params.put("equipmentType", "0");
                        HttpUtil.postRequest(Constant.getValue("UPDATE_USER"), params);
                    }
                    
                }
                // 取消关注
                else if (eventType.equals(MessageUtil.EVENT_TYPE_UNSUBSCRIBE)) {
                    // TODO 取消订阅后用户不会再收到公众账号发送的消息，因此不需要回复
                    Hashtable params = new Hashtable();
                    params.put("phoneIme", fromUserName);
                    params.put("state", 0);
                    HttpUtil.postRequest(Constant.getValue("UPDATE_USER_ANS"), params);
                    
                }
                // 扫描带参数二维码
                else if (eventType.toLowerCase().equals(MessageUtil.EVENT_TYPE_SCAN)) {
                    // TODO 处理扫描带参数二维码事件
                
                    if(StringUtils.isNotBlank(eventKey)){
                        
                        String requestUrl = "https://api.weixin.qq.com/cgi-bin/user/info?access_token=ACCESS_TOKEN&openid=OPENID";
                        
                        requestUrl = requestUrl.replace("ACCESS_TOKEN", tokenService.getToken())
                                .replace("OPENID", fromUserName);
                        JSONObject jsonObject = CommonUtil.httpsRequest(requestUrl, "GET",null);
                        String nickname = jsonObject.getString("nickname");
                        String address = jsonObject.getString("country")+"-"+jsonObject.getString("province")+"-"+jsonObject.getString("city");
                        String headimgurl = jsonObject.getString("headimgurl");
                        Hashtable params = new Hashtable();
                        params.put("nickname", nickname);
                        params.put("location", address);
                        params.put("equipmentType", eventKey);
                        //放入缓存中
                        //redisCacheTool.setDataToRedis(eventKey, 3600, params);
                        
                        JSONObject object = JSONObject.fromObject(params);

　　　　　　　　　　　　　　 //调用socket进行通信　　

　　　　　　　　　　　　　　 webSocketController.onMessage(object.toString(), null);
                        
                        LOGGER.info("qrcode redis key ==> "+eventKey);
                    }
                    

4， 网站端获取用户信息，进行session的处理就可以了

 

　　　如果觉得写得还有意思，可以反手就个推荐。