使用线程池模拟处理耗时任务,通过websocket提高用户体验
前言
在文章开始之前,询问一下大家平时工作中后端处理批量任务(耗时任务)的时候,前端是如何告知用户任务的执行情况的?
楼主对这个问题想了下,决定使用websokect将这一过程展现给用户。
于是就有了这篇文章,跟大家一起学习。
WebSocket简单介绍
WebSocket是HTML5开始提供的一种浏览器与服务器间进行全双工通讯的网络技术。 WebSocket通信协议于2011年被IETF定为标准 RFC 6455,WebSocketAPI被W3C定为标准。
在WebSocket API中,浏览器和服务器只需要做一个握手的动作,然后,浏览器和服务器之间就形成了一条快速通道。两者之间就直接可以数据互相传送。
----------以上文字摘自于Wiki对WebSocket的介绍。
简单来说,WebSocket就是浏览器与服务器之间进行通讯的一种技术。
关于WebSocket出现的背景及原理,大家可看wiki中的介绍。
实例讲解
本文讲解的实例使用java语言编写,web服务器使用jetty8。
先看下对应的Servlet和WebSocket代码:
Servlet:
@WebServlet(name = "main", urlPatterns = "/main.do")
public class MainServlet extends WebSocketServlet {
public WebSocket doWebSocketConnect(HttpServletRequest httpServletRequest, String s) {
return new TaskWebSocket();
}
}
WebSocket使用Task对象模拟耗时任务,用线程池进行处理:
public class TaskWebSocket implements WebSocket.OnTextMessage {
private Connection conn;
private ExecutorService executorService;
private CompletionService completionService;
private Random random = new Random();
public TaskWebSocket() {
executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
completionService = new ExecutorCompletionService<String>(executorService);
}
public void onMessage(String s) {
try {
conn.sendMessage("开始执行任务");
int taskNum = 4;
for(int i = 0; i < taskNum; i ++) {
Task task = new Task(random.nextInt(20), conn, String.valueOf(i + 1));
completionService.submit(task);
}
for(int i = 0; i < taskNum; i ++) {
String result = completionService.take().get();
conn.sendMessage(result);
}
conn.sendMessage("任务执行完毕");
conn.close();
} catch (Exception e) {
e.printStackTrace();
System.err.println("error");
}
}
public void onOpen(Connection connection) {
System.out.println("on Open");
this.conn = connection;
}
public void onClose(int i, String s) {
System.out.println("on Close");
}
}
class Task implements Callable<String> {
private int sleepSec;
private WebSocket.Connection conn;
private String name;
public Task(int sleepSec, WebSocket.Connection conn, String name) {
this.sleepSec = sleepSec;
this.conn = conn;
this.name = name;
}
public String call() throws Exception {
conn.sendMessage("任务(" + name + ")开始执行, 该任务大概会执行" + sleepSec + "秒");
Thread.sleep(sleepSec * 1000);
return "任务" + name + "执行完成";
}
}
效果图:
前端js的代码就不贴了,用的是原生的websocket。
代码下载地址: https://github.com/fangjian0423/jetty8-websocket-demo
参考资料
http://zh.wikipedia.org/wiki/WebSocket
http://redstarofsleep.iteye.com/blog/1307608
http://www.ibm.com/developerworks/cn/web/1112_huangxa_websocket/