Administrator
yesterday fe70ca5db711b458714bfbe4a6eda6cf7d8a0421
src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkItemServiceImpl.java
@@ -2,22 +2,34 @@
import cc.mrbird.febs.ai.entity.AiTalkItem;
import cc.mrbird.febs.ai.mapper.AiTalkItemMapper;
import cc.mrbird.febs.ai.req.talk.ApiTalkItemPageDto;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
import cc.mrbird.febs.ai.res.talk.ApiTalkItemPageVo;
import cc.mrbird.febs.ai.service.AiTalkItemService;
import cc.mrbird.febs.ai.utils.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cn.hutool.core.date.DateTime;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import java.util.List;
@Slf4j
@Service
@RequiredArgsConstructor
public class AiTalkItemServiceImpl extends ServiceImpl<AiTalkItemMapper, AiTalkItem> implements AiTalkItemService {
    private final AiTalkItemMapper aiTalkItemMapper;
    @Override
    public void add(String id, int code, String context, String memberUuid, DateTime date) {
    public void add(String id,String companyId, int code, String context, String memberUuid, DateTime date) {
        AiTalkItem aiTalkItem = new AiTalkItem();
        aiTalkItem.setId(UUID.getSimpleUUIDString());
        aiTalkItem.setCompanyId(companyId);
        aiTalkItem.setMemberId(memberUuid);
        aiTalkItem.setType(code);
        aiTalkItem.setTalkId(id);
@@ -25,4 +37,22 @@
        aiTalkItem.setContext(context);
        this.baseMapper.insert(aiTalkItem);
    }
    @Override
    public FebsResponse historyPage(ApiTalkItemPageDto dto) {
        // 创建分页对象,传入当前页和每页大小
        Page<ApiTalkItemPageVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        Page<ApiTalkItemPageVo> pageListByQuery = aiTalkItemMapper.getPageListByQuery(page, dto);
        return new FebsResponse().success().data(pageListByQuery);
    }
    @Override
    public List<AiTalkItem> getListByTalkId(String talkId) {
        LambdaQueryWrapper<AiTalkItem> queryWrapper = Wrappers.lambdaQuery(AiTalkItem.class);
        queryWrapper.eq(AiTalkItem::getTalkId, talkId);
        queryWrapper.orderByDesc(AiTalkItem::getCreatedTime);
        queryWrapper.last("limit 10");
        List<AiTalkItem> list = aiTalkItemMapper.selectList(queryWrapper);
        return list;
    }
}