微信小程序 websocket 实现SpringMVC+Spring+Mybatis

import java.util.Map;

import javax.servlet.http.httpServletrequest;
import org.springframework.http.server.Serverhttprequest;
import org.springframework.http.server.ServerhttpResponse;
import org.springframework.http.server.ServletServerhttprequest;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor;

/**

• 此类用来获取登录用户信息并交由websocket管理
  */
  public class MyWebSockeTinterceptor implements HandshakeInterceptor {

@Override
public Boolean beforeHandshake(Serverhttprequest request,ServerhttpResponse arg1,WebSocketHandler arg2,Map<String,Object> arg3) throws Exception {
// 将Serverhttprequest转换成request请求相关的类，用来获取request域中的用户信息
if (request instanceof ServletServerhttprequest) {
ServletServerhttprequest servletrequest = (ServletServerhttprequest) request;
httpServletrequest httprequest = servletrequest.getServletrequest();

}

System.out.println("连接到我了");

return true;

}

@Override
public void afterHandshake(Serverhttprequest arg0,Exception arg3) {
// TODO Auto-generated method stub

}

}

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.Textmessage;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketmessage;
import org.springframework.web.socket.WebSocketSession;

public class WebSocketPushHandler implements WebSocketHandler {
private static final List users = new ArrayList<>();

// 用户进入系统监听
@Override
public void afterConnectionestablished(WebSocketSession session) throws Exception {
System.out.println("成功进入了系统。。。");
users.add(session);

sendmessagesToUsers(new Textmessage("今天晚上服务器维护,请注意"));

}

//
@Override
public void handlemessage(WebSocketSession session,WebSocketmessage<?> messagE) throws Exception {
// 将消息进行转化，因为是消息是json数据，可能里面包含了发送给某个人的信息，所以需要用json相关的工具类处理之后再封装成Textmessage，
// 我这儿并没有做处理，消息的封装格式一般有{from:xxxx,to:xxxxx,msg:xxxxx}，来自哪里，发送给谁，什么消息等等
// Textmessage msg = (TextmessagE)message.getPayload();
// 给所有用户群发消息
//sendmessagesToUsers(msg);
// 给指定用户群发消息
//sendmessageToUser(userId,msg);

}

// 后台错误信息处理方法
@Override
public void handleTransportError(WebSocketSession session,Throwable exception) throws Exception {

}

// 用户退出后的处理，不如退出之后，要将用户信息从websocket的session中remove掉，这样用户就处于离线状态了，也不会占用系统资源
@Override
public void afterConnectionClosed(WebSocketSession session,CloseStatus closeStatus) throws Exception {
if (session.isOpen()) {
session.close();
}
users.remove(session);
System.out.println("安全退出了系统");

}

@Override
public Boolean supportsPartialmessages() {
return false;
}

/**

• 给所有的用户发送消息
  */
  public void sendmessagesToUsers(Textmessage messagE) {
  for (WebSocketSession user : users) {
  try {
  // isOpen()在线就发送
  if (user.isOpen()) {
  user.sendmessage(messagE);
  }
  } catch (IOException E) {
  e.printStackTrace();
  }
  }
  }

/**

• 发送消息给指定的用户
  */
  public void sendmessageToUser(String userId,Textmessage messagE) {
  for (WebSocketSession user : users) {
  if (user.getAttributes().get("").equals(userId)) {
  try {
  // isOpen()在线就发送
  if (user.isOpen()) {
  user.sendmessage(messagE);
  }
  } catch (IOException E) {
  e.printStackTrace();
  }
  }
  }
  }
  }

import org.springframework.context.Annotation.bean;
import org.springframework.context.Annotation.Configuration;
import org.springframework.web.servlet.config.Annotation.WebMvcConfigurerAdapter;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.Annotation.EnableWebSocket;
import org.springframework.web.socket.config.Annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.Annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {

registry.addHandler(WebSocketPushHandler(),"/webSocketServer.action").addInterceptors(new MyWebSockeTinterceptor()).setAllowedOrigins("*");
registry.addHandler(WebSocketPushHandler(),"/sockjs/webSocketServer.action")
    .addInterceptors(new MyWebSockeTinterceptor()).withSockJS();

}

@Bean
public WebSocketHandler WebSocketPushHandler() {
return new WebSocketPushHandler();
}

}

spring.xml中配置扫描包

/**

• 页面的初始数据
  */
  data: {

},connectWebsocket: function () {
wx.connectSocket({
url: 'ws://localhost:8080/AhutShop/webSocketServer.action',data: {
},header: {
'content-type': 'application/json'
},method: "GET"
})
wx.onSocketOpen(function (res) {
console.log('WebSocket连接已打开！')
})
wx.onSocketError(function (res) {
console.log('WebSocket连接打开失败，请检查！')
})
wx.onSocketmessage(function (res) {
console.log('收到服务器内容：' + res.data)
})
}
})