iailab-module-ai/iailab-module-ai-biz/src/main/java/com/iailab/module/ai/dal/mysql/chat/AiChatMessageMapper.java

@@ -7,6 +7,7 @@
import com.iailab.framework.mybatis.core.mapper.BaseMapperX;
import com.iailab.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.iailab.module.ai.controller.admin.chat.vo.message.AiChatMessagePageReqVO;
import com.iailab.module.ai.controller.admin.chat.vo.message.AiChatMessageReqVO;
import com.iailab.module.ai.dal.dataobject.chat.AiChatMessageDO;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.apache.ibatis.annotations.Mapper;
@@ -24,6 +25,13 @@
@Mapper
public interface AiChatMessageMapper extends BaseMapperX<AiChatMessageDO> {

    default PageResult<AiChatMessageDO> selectPageByConversationId(AiChatMessageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<AiChatMessageDO>()
                .eq(AiChatMessageDO::getConversationId, reqVO.getConversationId())
                .betweenIfPresent(AiChatMessageDO::getCreateTime, reqVO.getCreateTime())
                .orderByDesc(AiChatMessageDO::getId));
    }

    default List<AiChatMessageDO> selectListByConversationId(Long conversationId) {
        return selectList(new LambdaQueryWrapperX<AiChatMessageDO>()
                .eq(AiChatMessageDO::getConversationId, conversationId)