java spring整合websocket教程(ssm框架,原创复制即可用)_太平洋学习网|一个最全的javaweb,js,css,html5,csdn,android,linux的学习网站。

在javaweb项目开发中，我们经常可能需要websocket来实现人员互动，消息提醒等功能，但我们如何在ssm框架(spring+springmvc+mybatis)的spring中整合websocket呢？本人刚刚开发了一个消息提醒功能，就是在oa系统中，当申请人有一条申请记录时，会马上js弹窗消息告诉审批人员他有一条待办事项。

（请在ssm框架搭建好运行正常的情况下使用websocket整合教程），总共7个步骤，步骤7和最后面的红色字体备注的方式可以任选一种。

步骤一：ssm框架中使用到了maven项目，我们在pom.xml中添加引入jar包，还有json转换相关的jar包，因为websocket消息是通过json数据传输的，json相关的jar包有很多，我相信你的项目中已经有了，本人用的是jackson相关的jar包。

<!-- Jackson -->
<dependency>
	<groupId>com.fasterxml.jackson.core</groupId>
	<artifactId>jackson-databind</artifactId>
	<version>${jackson.version}</version>
</dependency>

<dependency>
	<groupId>com.fasterxml.jackson.core</groupId>
	<artifactId>jackson-core</artifactId>
	<version>${jackson.version}</version>
</dependency>

<dependency>
	<groupId>com.fasterxml.jackson.core</groupId>
	<artifactId>jackson-annotations</artifactId>
	<version>${jackson.version}</version>
</dependency>
<dependency>
	<groupId>javax.websocket</groupId>
	<artifactId>javax.websocket-api</artifactId>
	<version>1.1</version>
</dependency>
<dependency>
  <groupId>org.springframework</groupId>
  <artifactId>spring-messaging</artifactId>
  <version>4.0.5.RELEASE</version>
</dependency>
<dependency>
  <groupId>org.springframework</groupId>
  <artifactId>spring-websocket</artifactId>
  <version>4.0.5.RELEASE</version>
</dependency>

步骤二：websocket相关jar包导入之后，开始创建websocket配置相关的类，如我的WebSocketConfig.java，这一个类似于servlet项目中的@ServerEndpoint注解，它的作用就是在html页面中可以直接访问这个类配置的网址，从而进行相关逻辑的处理，而不是经过controller层来管理，有了这个类，WebSocketConfig就相当于是controller层了。

@Configuration
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer{
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(WebSocketPushHandler(),"/webSocketServer").addInterceptors(new MyWebSocketInterceptor());
        registry.addHandler(WebSocketPushHandler(), "/sockjs/webSocketServer").addInterceptors(new MyWebSocketInterceptor())
                .withSockJS();
    }
    @Bean	
    public WebSocketHandler WebSocketPushHandler(){
        return new WebSocketPushHandler();
    }
}

上面的类创建好之后，需要在springmvc的spring-context.xml配置文件中加入包扫描，如果你的ssm框架中已经有包扫描，就不需要再添加包扫描了，比如WebSocketConfig.java类在项目包com.baidu.webSocketDemo.xxxxx下，你只需要扫描基本包就好了，如下：

<context:component-scan base-package="com.baidu.webSocketDemo"/>

@Configuration注解该类，等价于XML中配置beans标签，@Bean相当于spring xml里面的bean标签，@EnableWebSocket这个注解用于springmvc对websocket的支持。

其中WebSocketPushHandler是websocket消息处理类，包括窗口打开，关闭，信息发送，错误信息监听等功能。MyWebSocketInterceptor是登录信息拦截处理的类，都是自定义的，名字可以自己取，后面会有代码演示。

1：/webSocketServer这个请求网址是普通浏览器访问所用，使用大多数浏览器支持websocket

2：/sockjs/webSocketServer 是适应IE低版本浏览器所用的，因为IE11以下的浏览器都不支持websocket，所以需要前台jsp页面用这个网址访问注册对应的注册器。

步骤三：在web.xml中的servlet和filter中加入支持异步处理的代码，因为websocket页面是异步处理信息的。

<async-supported>true</async-supported>

servlet中加入：

<servlet>
	<servlet-name>oa</servlet-name>
	<servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
	<init-param>
		<param-name>contextConfigLocation</param-name>
		<param-value>classpath:spring/spring-mvc.xml</param-value>
	</init-param>
	<load-on-startup>1</load-on-startup>
	<async-supported>true</async-supported>
</servlet>

所有的filter中也要加入，如下：

<filter>
	<filter-name>shiroFilter</filter-name>
	<filter-class>org.springframework.web.filter.DelegatingFilterProxy</filter-class>
	<async-supported>true</async-supported>
	<init-param>
		<param-name>targetFilterLifecycle</param-name>
		<param-value>true</param-value>
	</init-param>
</filter>

经过上面的步骤，spring websocket的配置整合基本上已经ok，下面就要开始写处理类相关信息。

步骤四：MyWebSocketInterceptor拦截用户登录信息，并将用户登录信息交给websocket的WebSocketSession来管理，因为这样websocket就可以知道用户是否在线，是否不在线了，而且还能给别的用户发送消息，非常的方便。

/**
 * 此类用来获取登录用户信息并交由websocket管理
 * @author ts
 *
 */
public class MyWebSocketInterceptor implements HandshakeInterceptor{
		
	@Override
	public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
			Map<String, Object> attributes) throws Exception {
		//将ServerHttpRequest转换成request请求相关的类，用来获取request域中的用户信息	
		if (request instanceof ServletServerHttpRequest) {
			ServletServerHttpRequest servletRequest  = (ServletServerHttpRequest) request;
			HttpServletRequest httpRequest = servletRequest.getServletRequest();
			//Constants.CURRENT_USER这个是我定义的常量，是request域的key，通过key就可以获取到用户信息了
			TsUser user = (TsUser)httpRequest.getAttribute(Constants.CURRENT_USER);
			//Constants.CURRENT_WEBSOCKET_USER也是常量，用来存储WebsocketSession的key值
			attributes.put(Constants.CURRENT_WEBSOCKET_USER,user.getUserid());
		}
		return true;
	}
	@Override
	public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
			Exception exception) {		
	}
}

步骤五：WebSocketPushHandler消息处理类，这个类主要就是用来做消息的逻辑处理的，代码如下。

/**
 * 消息处理类
 * @author ts
 *
 */
public class WebSocketPushHandler implements WebSocketHandler{
	
	private static final List<WebSocketSession> users = new ArrayList<>();
	
	//用户进入系统监听
	@Override
	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
		System.out.println("成功进入了系统。。。");
		users.add(session);
	}

	//
	@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(Constants.CURRENT_WEBSOCKET_USER).equals(userId)){
				try {
				    //isOpen()在线就发送
					if(user.isOpen()){
						user.sendMessage(message);
					}
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
}

步骤六：jsp页面创建websocket对象，实现页面信息的实时接收。

下载sockjs.min.js,点击进去:https://cdn.jsdelivr.net/sockjs/1/sockjs.min.js,然后右键“另存为”就下载下来了。

引入socketjs文件，这个非常重要，不然IE低版本就不支持websocket功能了，提高浏览器的兼容性。

<script type="text/javascript" src="${ctx}/asset/lib/socketjs/sockjs.min.js"></script>

<script type="text/javascript">
	var  ws;
	window.onload=connect;
	function connect(){
		var ws;
		if ('WebSocket' in window) {
			ws = new WebSocket("ws://192.168.1.117:8080/xf_oa_new/webSocketServer");
		} else if ('MozWebSocket' in window) {
			ws = new MozWebSocket("ws://192.168.1.117:8080/xf_oa_new/webSocketServer");
		} else {
			//如果是低版本的浏览器，则用SockJS这个对象，对应了后台“sockjs/webSocketServer”这个注册器，
			//它就是用来兼容低版本浏览器的
			ws = new SockJS("http://192.168.1.117:8080/xf_oa_new/sockjs/webSocketServer");
		}
		ws.onopen = function (evnt) {
			
		};
		ws.onmessage = function (evnt) {
			eval("var dataObj="+event.data);
			if(dataObj != undefined){	
				$(dataObj.data).each(function(i,o){
					$("#msgName").text(o.msgName);
					$("#amount").text(o.amount);
					$("#msg a:first").attr("_href",core.getRootPath()+o.toUrl).text(o.taskName);
					$("#msgId").val(o.id);
					setTimeout("tips_pop()",1000); 
				});
			}
			
		};
		ws.onerror = function (evnt) {
		};
		ws.onclose = function (evnt) {
		}
	}
	
	function  send(){
		var value= $("#msg").val();
		ws.send(value);
	}
</script>

步骤七：我们ssm框架的controller层或者service层该如何用它来发送消息呢？通常我们使用一个方法，都是用@Autowired注解注入service的，而我们这里因为WebSocketPushHandler并不是service层方法，而是一个和controller层一样的控制器，这个时候我们就需要new对象来引用WebSocketPushHandler这个类里面的发送信息的方法了。

首先我在controller层注入对象实例。

/**
 * 注入发送消息类实例
 * @return
 */
@Bean
public WebSocketPushHandler getWebSocketPushHandler() {
	return new WebSocketPushHandler();
}

然后对信息进行封装。

/**
 * 根据用户获取消息内容
 */
public TextMessage getMsgByUserId(String userId){
	Map<String, Object> resultMap = tcService.selectTask(userId);
	ObjectMapper mapper = new ObjectMapper();
	String msg = "";
	try {
		msg = mapper.writeValueAsString(resultMap);
	} catch (JsonProcessingException e) {
		e.printStackTrace();
	}
	return new TextMessage(msg);
}

然后在controller方法中引入发送消息给某个人的方法。

//前台通知处理人处理任务
getWebSocketPushHandler().sendMessageToUser(info.getUserId(), this.getMsgByUserId(info.getUserId()));

特别注意：步骤七，其实我们发送消息也不需要再controller层做，因为jsp页面已经将消息交给了WebSocketConfig类，而我们前面已经在WebSocketConfig.java文件中定义了url对应的的WebSocketPushHandler消息处理类，因此我们所有群发消息，对某个用户发送消息，都可以在前面WebSocketPushHandler这个消息处理类中的handleMessage方法中判断是否群发消息，是否对某人发送消息。步骤七可以增加程序的扩展性，它可以在任何类中使用发送消息的方法。