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

时间 2019-11-09

原文原文链接

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

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

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

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

　　 5，这里的惟一值是能够本身定义的，我用的是截取了几位的时间戳

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

优化方案

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

　　　　客户端（网站）与服务端（微信）的通讯，而且须要服务器能够主动的推送消息到客户端（网站）api

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

　　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的处理就能够了

　　　若是以为写得还有意思，能够反手就个推荐。