package com.tianji.learning.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.tianji.learning.domain.po.PointsBoard;
import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.Param;

/**
 * <p>
 * 学霸天梯榜 Mapper 接口
 * </p>
 *
 * @author CamelliaV
 * @since 2024-11-21
 */
public interface PointsBoardMapper extends BaseMapper<PointsBoard> {
    @Insert("CREATE TABLE `${tableName}` (" +
            "id BIGINT NOT NULL AUTO_INCREMENT COMMENT '榜单id'," +
            "user_id BIGINT NOT NULL COMMENT '学生id'," +
            "points INT NOT NULL COMMENT '积分值'," +
            "PRIMARY KEY (`id`) USING BTREE," +
            "INDEX `idx_user_id` (`user_id`) USING BTREE" +
            ")" +
            "COMMENT = '学霸天梯榜'," +
            "COLLATE = 'utf8mb4_0900_ai_ci'," +
            "ENGINE = InnoDB," +
            "ROW_FORMAT = DYNAMIC")
    void createPointsBoardTableBySeason(@Param("tableName") String tableName);
}

package com.tianji.learning.job;

import com.tianji.common.utils.DateUtils;
import com.tianji.learning.constants.LearningConstants;
import com.tianji.learning.constants.RedisConstants;
import com.tianji.learning.domain.po.PointsBoard;
import com.tianji.learning.domain.po.PointsBoardSeason;
import com.tianji.learning.service.IPointsBoardSeasonService;
import com.tianji.learning.service.IPointsBoardService;
import com.tianji.learning.utils.TableInfoContext;
import com.xxl.job.core.context.XxlJobHelper;
import com.xxl.job.core.handler.annotation.XxlJob;
import lombok.RequiredArgsConstructor;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

/**
 * @author CamelliaV
 * @since 2024/11/21 / 21:46
 */
// @Component
@RequiredArgsConstructor
public class PointsBoardPersistHandler {
    private final IPointsBoardService boardService;
    private final IPointsBoardSeasonService seasonService;
    private final StringRedisTemplate redisTemplate;

    // * 拆分成多个XXLJOB然后以子任务关联执行（方便对其中一部分同步Redis数据到数据库做任务分片）
    @XxlJob("createHistoryBoardInDb")
    public void createHistoryBoardInDb() {
        // * 获取上月时间（每次同步上月排行榜数据）
        LocalDate time = LocalDate.now()
                                  .minusMonths(1);
        LocalDate beginDate = DateUtils.getMonthBegin(time);
        LocalDate endDate = beginDate.plusDays(beginDate.lengthOfMonth() - 1);

        // * 查询对应时间的赛季数据
        PointsBoardSeason season = seasonService.lambdaQuery()
                                                .le(PointsBoardSeason::getEndTime, endDate)
                                                .ge(PointsBoardSeason::getBeginTime, beginDate)
                                                .one();
        // * 无数据不同步
        if (season == null) {
            return;
        }
        // * 获取id构造表名在数据库创建表
        Integer seasonId = season.getId();
        String tableName = LearningConstants.POINTS_BOARD_TABLE_PREFIX + seasonId;
        boardService.createPointsBoardTableBySeason(tableName);
        // * 存入ThreadLocal用于动态表名插件
        TableInfoContext.setInfo(tableName);
    }

    @XxlJob("syncHistoryBoardToDb")
    public void syncHistoryBoardToDb() {
        // * 获取上月时间（每次同步上月排行榜数据）
        LocalDate time = LocalDate.now()
                                  .minusMonths(1);
        // * Redis查询上一赛季排行榜
        String key = RedisConstants.POINTS_BOARD_KEY_PREFIX + time.format(DateTimeFormatter.ofPattern("yyyyMM"));
        // * XXLJOB分片
        int shardIndex = XxlJobHelper.getShardIndex();
        int shardTotal = XxlJobHelper.getShardTotal();
        int pageNo = shardIndex + 1;
        int pageSize = LearningConstants.SYNC_POINTS_BOARD_PAGE_SIZE;
        while (true) {
            List<PointsBoard> boardList = new ArrayList<>();
            int start = (pageNo - 1) * pageSize;
            int end = start + pageSize - 1;
            Set<ZSetOperations.TypedTuple<String>> typedTuples = redisTemplate.opsForZSet()
                                                                              .reverseRangeWithScores(key, start, end);
            // * Redis已无排行数据
            if (typedTuples == null) {
                break;
            }
            // * 封装结果
            int rank = start + 1;
            for (ZSetOperations.TypedTuple<String> tuple : typedTuples) {
                String user = tuple.getValue();
                Double points = tuple.getScore();
                PointsBoard item = new PointsBoard();
                // * 数据残缺
                if (points == null || user == null) {
                    continue;
                }
                // * rank作为id
                item.setId((long) rank);
                rank++;
                item.setPoints(points.intValue());
                item.setUserId(Long.valueOf(user));
                boardList.add(item);
            }
            pageNo += shardTotal;
            // * 持久化到数据库
            boardService.saveBatch(boardList);
        }
        // * 移除ThreadLocal
        TableInfoContext.remove();

        // * 存入XXL_JOB计数，防止多次调用unlink（可能导致删除在于同步数据前，致使数据丢失）
        redisTemplate.opsForValue()
                     .increment(RedisConstants.SYNC_BOARD_XXL_JOB_TIMES);
        redisTemplate.opsForValue()
                     .set(RedisConstants.SYNC_BOARD_XXL_SHARD_TOTAL, String.valueOf(shardTotal));
    }

    @XxlJob("removeHistoryBoardInRedis")
    public void removeHistoryBoardInRedis() {
        String timesString = redisTemplate.opsForValue()
                                          .get(RedisConstants.SYNC_BOARD_XXL_JOB_TIMES);
        String totalString = redisTemplate.opsForValue()
                                          .get(RedisConstants.SYNC_BOARD_XXL_SHARD_TOTAL);
        if (timesString != null && timesString.equals(totalString)) {
            LocalDate time = LocalDate.now()
                                      .minusMonths(1);
            String key = RedisConstants.POINTS_BOARD_KEY_PREFIX + time.format(DateTimeFormatter.ofPattern("yyyyMM"));
            // * 异步清除Redis数据
            redisTemplate.unlink(key);
            redisTemplate.delete(RedisConstants.SYNC_BOARD_XXL_JOB_TIMES);
            redisTemplate.delete(RedisConstants.SYNC_BOARD_XXL_SHARD_TOTAL);
        }
    }
}