java后台耗时任务多线程返回结果

Question

@RequestMapping(value = "ehr", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON)
    @ResponseBody
    public User getUser() {
        User u = new User();
        Future<User> future = poolTaskExecutor.submit(ehrDownloadTask);
        boolean flag = future.isDone();
        while (!flag) {
            u.setMsg("未完成");
            return u;
        }
        try {
            u = future.get();
        } catch (Exception e) {
            e.printStackTrace();
        }
        System.out.println("任务完成！");
        return u;
    }
    

假设我我请求一个耗时任务放到线程池里执行，我想如果没执行完返回前台未完成，前台可以刷新页面显示完没完成，但是一刷新，我就又新起了一个线程去执行，所以一直返回未完成。请教下这个怎么写？

Answer 1

写个简单实现：

    private static Map<String,Future<User>> futures = new HashMap<>();
    
    @RequestMapping(value = "ehr", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON)
    @ResponseBody
    public User getUser(HttpServletRequest request, HttpServletResponse response) {
        String sessionID = request.getSession().getId();

        User u = new User();
        u.setMsg("未完成");
        if(futures.containsKey(sessionID)) {
            Future<User> userFuture = futures.get(sessionID);
            if(userFuture.isDone()) {
                try {
                    u = userFuture.get();
                } catch (InterruptedException | ExecutionException e) {
                    e.printStackTrace();
                }
                futures.remove(sessionID);
            }
        } else {
            Future<User> future = poolTaskExecutor.submit(ehrDownloadTask);
            futures.put(sessionID,future);
        }
        return u;
    }

前台轮训直到正确返回。
Key你可以用我这样的SessionID，也可以自己根据逻辑生成
如果是分布式Tomcat可以用Redis代替HashMap

Answer 2

谢邀
linkse7en 的方案基本没问题，但是需要注意一点 线程安全 HashMap不具备线程安全，这么写会出问题，建议使用ConcurrentHashMap
其次，你这个可以考虑用定时任务处理表里的记录，tableA.doneStatus这种 这个时候前端就走查询逻辑就好了，一般的业务场景还是系统自动触发处理，用户查询了你再处理，我的感觉还是不太好
以上