方案優化：網站實現掃描二維碼關注微信公衆號，自動登錄網站並獲取其信息

時間 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的處理就能夠了

　　　若是以爲寫得還有意思，能夠反手就個推薦。