上一篇 《网站实现扫描二维码关注微信公众号,自动登陆网站并获取其信息》 中已经实现用户扫码登陆网站并获取其信息
但是上一篇方案中存在一个问题,也就是文章末尾指出的可以优化的地方(可以点击这个链接去看一下上篇文章)
首先回顾一下上一篇的思路:
上述方案中,实现的难点在于微信服务器不能主动与网站进行通信,所以采用了网页端的轮训去做进行用户状态的查询,但是方案中4,网站端使用轮训去全局的redis缓存中查找用户信息,这样使用轮训无疑就会产生跟多的无用http请求,不仅增加了服务器的压力还浪费资源。
简化一下需要完成上述功能的实现要点:
客户端(网站)与服务端(微信)的通信,并且需要服务器可以主动的推送消息到客户端(网站)
要是实现客户端与服务器端的通信这个功能,常用的方式有两种:
1,轮训(也就是上次方案中使用的方法)
2,长连接
我使用的是利用websocket与服务器进行通信,而且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);
- }
来源: http://www.cnblogs.com/cmyxn/p/7814120.html