上一篇 《网站实现扫描二维码关注微信公众号，自动登陆网站并获取其信息》 中已经实现用户扫码登陆网站并获取其信息
但是上一篇方案中存在一个问题，也就是文章末尾指出的可以优化的地方（可以点击这个链接去看一下上篇文章）
首先回顾一下上一篇的思路：
1，微信的系统，提供生成带参数的二维码的接口，这个参数就是唯一值（场景值）
2，网站调用微信系统，获取生成的二维码图片
3，用户扫码会直接调用微信服务器，将用户访问微信服务器的信息记录到redis，key就是唯一值（场景值）
4，网站端做轮训去查询redis中是否有这个唯一值的数据，如果有就获取用户信息登录，没有就五秒一次轮训，登录后就不在做轮训（从二维码弹出之后开始做轮训，关闭二维码后停止轮训）
2，网站调用微信系统，获取生成的二维码图片
3，用户扫码会直接调用微信服务器，将用户访问微信服务器的信息记录到redis，key就是唯一值（场景值）
4，网站端做轮训去查询redis中是否有这个唯一值的数据，如果有就获取用户信息登录，没有就五秒一次轮训，登录后就不在做轮训（从二维码弹出之后开始做轮训，关闭二维码后停止轮训）
5，这里的唯一值是可以自己定义的，我用的是截取了几位的时间戳
上述方案中，实现的难点在于微信服务器不能主动与网站进行通信，所以采用了网页端的轮训去做进行用户状态的查询，但是方案中4，网站端使用轮训去全局的redis缓存中查找用户信息，这样使用轮训无疑就会产生跟多的无用http请求，不仅增加了服务器的压力还浪费资源。
优化方案
简化一下需要完成上述功能的实现要点：
客户端（网站）与服务端（微信）的通信，并且需要服务器可以主动的推送消息到客户端（网站）
要是实现客户端与服务器端的通信这个功能，常用的方式有两种：
1，轮训（也就是上次方案中使用的方法）
2，长连接
我使用的是利用websocket与服务器进行通信，而且websocket使用起来也非常方便，易上手
优化后的方案：
1，微信的系统，提供生成带参数的二维码的接口，这个参数就是唯一值（场景值）
2，网站调用微信系统，获取生成的二维码图片
3，用户扫码会直接调用微信服务器，然后将获取到的用户信息利用websocket发送给网站端
4，网站接受信息，并显示登陆
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的处理就可以了
如果觉得写得还有意思，可以反手就个推荐。