package com.ycwl.basic.puzzle.mapper;

import com.ycwl.basic.puzzle.entity.PuzzleGenerationRecordEntity;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.util.List;

/**
 * 拼图生成记录Mapper接口
 *
 * @author Claude
 * @since 2025-01-17
 */
@Mapper
public interface PuzzleGenerationRecordMapper {

    /**
     * 根据ID查询记录
     */
    PuzzleGenerationRecordEntity getById(@Param("id") Long id);

    /**
     * 查询用户的生成记录列表
     */
    List<PuzzleGenerationRecordEntity> listByUserId(@Param("userId") Long userId,
                                                     @Param("limit") Integer limit);

    /**
     * 查询人脸ID的生成记录列表
     */
    List<PuzzleGenerationRecordEntity> listByFaceId(@Param("faceId") Long faceId);


    /**
     * 统计人脸ID的生成记录数量
     */
    int countByFaceId(@Param("faceId") Long faceId);

    /**
     * 插入记录
     */
    int insert(PuzzleGenerationRecordEntity entity);

    /**
     * 更新记录
     */
    int update(PuzzleGenerationRecordEntity entity);

    /**
     * 更新为成功状态
     */
    int updateSuccess(@Param("id") Long id,
                     @Param("resultImageUrl") String resultImageUrl,
                     @Param("resultFileSize") Long resultFileSize,
                     @Param("resultWidth") Integer resultWidth,
                     @Param("resultHeight") Integer resultHeight,
                     @Param("generationDuration") Integer generationDuration);

    /**
     * 更新为失败状态
     */
    int updateFail(@Param("id") Long id,
                   @Param("errorMessage") String errorMessage);

    /**
     * 根据内容哈希查询历史记录(用于去重)
     * 查询条件: template_id + content_hash + scenic_id + status=1
     * 返回最新的成功记录
     *
     * @param templateId  模板ID
     * @param contentHash 内容哈希
     * @param scenicId    景区ID
     * @return 历史记录,如果不存在返回null
     */
    PuzzleGenerationRecordEntity findByContentHash(@Param("templateId") Long templateId,
                                                    @Param("contentHash") String contentHash,
                                                    @Param("scenicId") Long scenicId);

    /**
     * 根据人脸ID和模板ID查询最近的成功记录
     * 用于素材版本缓存命中时快速返回历史结果
     *
     * @param faceId     人脸ID
     * @param templateId 模板ID
     * @return 最近的成功记录，如果不存在返回null
     */
    PuzzleGenerationRecordEntity findLatestSuccessByFaceAndTemplate(@Param("faceId") Long faceId,
                                                                      @Param("templateId") Long templateId);
}
