UploadData.java 文件源码

/**
 * 执行数据上报
 * 
 * @author gaoxianglong
 * @throws Exception
 */
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class, isolation = Isolation.DEFAULT)
public void upload(String taskId, Map<String, CountDownLatch> countDownLatchMap,
        Map<String, List<ResultBean>> resultMap) throws Exception {
    CountDownLatch latch = countDownLatchMap.get(taskId);
    try {
        /* 等待指定场景的压测数据 */
        latch.await();
        List<ResultBean> results = resultMap.get(taskId);
        /* 统计压测结果 */
        ReportPO reportPO = ResultStatistics.result(results);
        if (null != reportPO) {
            /* 新增压测结果信息 */
            reportDao.insertReport(reportPO);
            /* 更改场景状态为未开始 */
            reportDao.updateScene(reportPO.getSceneId(), 0);
            log.info("senceId为[" + reportPO.getSceneId() + "]的压测结果已经收集完成并成功上报");
        }
    } finally {
        /* 资源回收 */
        countDownLatchMap.remove(taskId);
        resultMap.remove(taskId);
    }
}