|
@@ -4,14 +4,18 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
import org.dromara.ai.domain.Conversation;
|
|
|
+import org.dromara.ai.domain.VO.ConversationVO;
|
|
|
import org.dromara.ai.mapper.ConversationMapper;
|
|
|
import org.dromara.ai.service.ConversationService;
|
|
|
import org.dromara.common.core.domain.R;
|
|
|
import org.dromara.common.mybatis.core.page.PageQuery;
|
|
|
import org.dromara.common.mybatis.core.page.TableDataInfo;
|
|
|
import org.dromara.common.satoken.utils.LoginHelper;
|
|
|
+import org.springframework.beans.BeanUtils;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
@Service
|
|
|
public class ConversationServiceImpl extends ServiceImpl<ConversationMapper, Conversation> implements ConversationService {
|
|
|
@Override
|
|
@@ -19,6 +23,14 @@ public class ConversationServiceImpl extends ServiceImpl<ConversationMapper, Con
|
|
|
Long userId = LoginHelper.getUserId();
|
|
|
LambdaQueryWrapper<Conversation> queryWrapper = new LambdaQueryWrapper<Conversation>().eq(Conversation::getUserId, userId).orderByDesc(Conversation::getTimestamp);
|
|
|
Page<Conversation> page = page(pageQuery.build(), queryWrapper);
|
|
|
- return R.ok(TableDataInfo.build(page));
|
|
|
+ List<ConversationVO> list = page.getRecords().stream().map(conversation -> {
|
|
|
+ ConversationVO vo = new ConversationVO();
|
|
|
+ BeanUtils.copyProperties(conversation, vo);
|
|
|
+ vo.setConversationId(vo.getId());
|
|
|
+ return vo;
|
|
|
+ }).toList();
|
|
|
+ TableDataInfo<ConversationVO> tableDataInfo = TableDataInfo.build(list);
|
|
|
+ tableDataInfo.setTotal(page.getTotal());
|
|
|
+ return R.ok(tableDataInfo);
|
|
|
}
|
|
|
}
|