@ExcelExport(ExcelBizTypeEnum.USER)
@ApiOperation(notes = "導(dǎo)出用戶數(shù)據(jù)", value = "導(dǎo)出用戶數(shù)據(jù)")
@PostMapping("/export")
public void export(HttpServletResponse response, UserConditionEntity userConditionEntity) {
    // 方法體可以為空，切面會(huì)自動(dòng)處理
}

@Aspect
@Component
public class CommonTaskAspect {
    
    @Before("@annotation(cn.net.susan.annotation.ExcelExport)")
    public void before(JoinPoint joinPoint) throws Throwable {
        // 獲取注解信息
        ExcelBizTypeEnum excelBizTypeEnum = method.getAnnotation(ExcelExport.class).value();
        
        // 創(chuàng)建任務(wù)實(shí)體
        CommonTaskEntity commonTaskEntity = createCommonTaskEntity(excelBizTypeEnum);
        
        // 保存任務(wù)到數(shù)據(jù)庫
        commonTaskMapper.insert(commonTaskEntity);
    }
}

@Component
public class CommonTaskJob extends BaseJob {
    
    @Override
    public JobResult doRun(String params) {
        // 查詢待執(zhí)行任務(wù)
        CommonTaskConditionEntity condition = new CommonTaskConditionEntity();
        condition.setStatusList(Arrays.asList(
            TaskStatusEnum.WAITING.getValue(),
            TaskStatusEnum.RUNNING.getValue()
        ));
        
        List<CommonTaskEntity> tasks = commonTaskMapper.searchByCondition(condition);
        
        // 執(zhí)行任務(wù)
        for (CommonTaskEntity task : tasks) {
            AsyncTaskStrategyContextFactory.getInstance()
                .getStrategy(task.getType())
                .doTask(task);
        }
        
        return JobResult.SUCCESS;
    }
}

@AsyncTask(TaskTypeEnum.EXPORT_EXCEL)
@Service
public class ExcelExportTask implements IAsyncTask {
    
    @Override
    public void doTask(CommonTaskEntity commonTaskEntity) {
        try {
            // 1. 更新任務(wù)狀態(tài)為執(zhí)行中
            commonTaskEntity.setStatus(TaskStatusEnum.RUNNING.getValue());
            commonTaskMapper.update(commonTaskEntity);
            
            // 2. 獲取業(yè)務(wù)類型和請(qǐng)求參數(shù)
            ExcelBizTypeEnum excelBizTypeEnum = getExcelBizTypeEnum(commonTaskEntity.getBizType());
            String requestParam = commonTaskEntity.getRequestParam();
            Object toBean = JSONUtil.toBean(requestParam, aClass);
            
            // 3. 獲取對(duì)應(yīng)的Service并執(zhí)行導(dǎo)出
            String serviceName = this.getServiceName(requestEntity);
            BaseService baseService = (BaseService) SpringBeanUtil.getBean(serviceName);
            String fileName = getFileName(excelBizTypeEnum.getDesc());
            String fileUrl = baseService.export(toBean, fileName, this.getEntityName(requestEntity));
            
            // 4. 更新任務(wù)狀態(tài)為成功
            commonTaskEntity.setFileUrl(fileUrl);
            commonTaskEntity.setStatus(TaskStatusEnum.SUCCESS.getValue());
            
        } catch (Exception e) {
            // 5. 處理失敗情況
            handleTaskFailure(commonTaskEntity, e);
        } finally {
            // 6. 更新任務(wù)記錄并發(fā)送通知
            commonTaskMapper.update(commonTaskEntity);
            sendNotifyMessage(commonTaskEntity);
        }
    }
}

@RocketMQMessageListener(
    topic = "${mall.mgt.excelExportTopic:EXCEL_EXPORT_TOPIC}",
    consumerGroup = "${mall.mgt.excelExportGroup:EXCEL_EXPORT_GROUP}"
)
@Component
public class ExcelExportConsumer implements RocketMQListener<MessageExt> {
    
    @Override
    public void onMessage(MessageExt message) {
        String content = new String(message.getBody());
        CommonNotifyEntity commonNotifyEntity = JSONUtil.toBean(content, CommonNotifyEntity.class);
        pushNotify(commonNotifyEntity);
    }
    
    private void pushNotify(CommonNotifyEntity commonNotifyEntity) {
        // 通過WebSocket推送通知
        WebSocketServer.sendMessage(commonNotifyEntity);
        
        // 更新通知狀態(tài)
        commonNotifyEntity.setIsPush(1);
        commonNotifyMapper.update(commonNotifyEntity);
    }
}

public class AsyncTaskStrategyContextFactory {
    private static Map<Integer, IAsyncTask> asyncTaskMap;
    
    public IAsyncTask getStrategy(Integer taskType) {
        return asyncTaskMap.get(taskType);
    }
}

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface ExcelExport {
    ExcelBizTypeEnum value();
}

private String doExport(V v, String fileName, String clazzName) {
    RequestConditionEntity conditionEntity = (RequestConditionEntity) v;
    
    // 計(jì)算分頁參數(shù)
    int totalCount = getBaseMapper().searchCount(conditionEntity);
    int sheetCount = totalCount % sheetDataSize == 0 ? 
        totalCount / sheetDataSize : totalCount / sheetDataSize + 1;
    
    // 創(chuàng)建ExcelWriter
    ExcelWriter excelWriter = EasyExcel.write(file).build();
    
    // 分頁處理數(shù)據(jù)
    for (int sheetIndex = 1; sheetIndex <= sheetCount; sheetIndex++) {
        List<K> dataEntities = getBaseMapper().searchByCondition(conditionEntity);
        
        // 寫入數(shù)據(jù)到Sheet
        WriteSheet writeSheet = EasyExcel.writerSheet("Sheet" + sheetIndex)
            .head(Class.forName(clazzName)).build();
        excelWriter.write(dataEntities, writeSheet);
        
        conditionEntity.setPageNo(conditionEntity.getPageNo() + 1);
    }
    
    excelWriter.finish();
    return uploadToOSS(file);
}