关注微信公众号 太平洋学习网 扫描二维码
最新消息:关注【太平洋学习网】微信公众号,可以获取全套资料,【全套Java基础27天】【JavaEE就业视频4个月】【Android就业视频4个月】

java spring整合websocket教程(ssm框架,原创复制即可用)

Javaweb admin 浏览 评论

在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方法中判断是否群发消息,是否对某人发送消息。步骤七可以增加程序的扩展性,它可以在任何类中使用发送消息的方法。

来源网站:太平洋学习网,转载请注明出处:http://www.tpyyes.com/a/javaweb/2017/0119/41.html

    与本文相关的文章