使用线程池模拟处理耗时任务，通过websocket提高用户体验

使用线程池模拟处理耗时任务，通过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/