04 xxl-job如何实现任务的分片

@Slf4j
@Component
public class ShardingTaskDemo {

    @Autowired
    private DataService dataService;

    /**
     * 分片广播任务示例：处理100万条数据
     */
    @XxlJob("shardingDataJob")
    public ReturnT<String> shardingDataJob(String param) {
        // 获取分片参数
        int shardIndex = XxlJobHelper.getShardIndex(); // 当前分片序号（从0开始）
        int shardTotal = XxlJobHelper.getShardTotal(); // 总分片数（等于执行器数量）

        // 解析参数中的总数据范围（假设参数为JSON格式：{"startId":1, "endId":1000000}）
        RangeParam range = JSON.parseObject(param, RangeParam.class);

        // 计算当前分片处理的数据范围
        Pair<Integer, Integer> shardRange = calculateRange(range.getStartId(), range.getEndId(), shardIndex, shardTotal);

        // 分批次处理数据（每次处理1000条）
        processDataByBatch(shardRange.getLeft(), shardRange.getRight(), 1000);

        return ReturnT.SUCCESS;
    }

    /**
     * 分片范围计算逻辑
     * 示例：总数据100万，分片数3 → 分片0处理1-333333，分片1处理333334-666666，分片2处理666667-1000000
     */
    private Pair<Integer, Integer> calculateRange(int startId, int endId, int currentShard, int totalShards) {
        int totalData = endId - startId + 1;
        int step = totalData / totalShards;
        int remainder = totalData % totalShards;

        int rangeStart = startId + currentShard * step;
        int rangeEnd = (currentShard == totalShards - 1) ? (rangeStart + step + remainder - 1) : (rangeStart + step - 1);

        return Pair.of(rangeStart, rangeEnd);
    }

    /**
     * 分批次处理数据
     */
    private void processDataByBatch(int startId, int endId, int batchSize) {
        int current = startId;
        while (current <= endId) {
            int batchEnd = Math.min(current + batchSize - 1, endId);
            List<Data> batchData = dataService.queryDataByIdRange(current, batchEnd);
            dataService.processBatch(batchData); // 实际业务处理
            XxlJobHelper.log("处理分片数据：{} - {}", current, batchEnd);
            current = batchEnd + 1;
        }
    }

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    public static class RangeParam {
        private int startId;
        private int endId;
    }
}