guyue
2 天以前 b7ff8446a58d59e7df8a588104c231c7af7d8573
src/main/java/com/linghu/service/impl/OrderServiceImpl.java
@@ -1,22 +1,177 @@
package com.linghu.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.linghu.model.entity.Order;
import com.linghu.model.dto.KeywordDto;
import com.linghu.model.dto.OrderDto;
import com.linghu.model.entity.Keyword;
import com.linghu.model.entity.KeywordTask;
import com.linghu.model.entity.Orders;
import com.linghu.model.entity.Question;
import com.linghu.service.KeywordService;
import com.linghu.service.KeywordTaskService;
import com.linghu.service.OrderService;
import com.linghu.mapper.OrderMapper;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
import com.linghu.service.QuestionService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
/**
* @author xy
* @description 针对表【order】的数据库操作Service实现
* @createDate 2025-07-02 16:32:19
*/
 * @author xy
 * @description 针对表【order】的数据库操作Service实现
 * @createDate 2025-07-04 20:17:33
 */
@Service
public class OrderServiceImpl extends ServiceImpl<OrderMapper, Order>
    implements OrderService{
public class OrderServiceImpl extends ServiceImpl<OrderMapper, Orders>
        implements OrderService {
    @Autowired
    private KeywordService keywordService;
    @Autowired
    private QuestionService questionService;
    @Autowired
    private OrderMapper orderMapper;
    @Autowired
    private KeywordTaskService keywordTaskService;
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveOrderWithKeywords(OrderDto orderDto, String order_id) {
        // 如果有关键词,则保存关键词
        if (StringUtils.hasText(orderDto.getKeywords())) {
            String[] keywordArray = orderDto.getKeywords().split("\\n");
            for (String keywordName : keywordArray) {
                if (StringUtils.hasText(keywordName)) {
                    Keyword keyword = new Keyword();
                    keyword.setOrder_id(order_id);
                    keyword.setKeyword_name(keywordName.trim());
                    keyword.setStatus("notSubmitted");
                    // keyword.setNum(1); // 默认采集轮数为1
                    keywordService.save(keyword);
                }
            }
        }
        return true;
    }
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean updateOrderWithKeywords(OrderDto orderDto, Integer currentStatus) {
        // 状态为2或3时禁止修改
        if (currentStatus >= 2) {
            throw new RuntimeException("执行中和已完成状态的订单不可修改");
        }
        // 更新订单基本信息
        if (!this.updateById(orderDto)) {
            return false;
        }
        // 删除旧关键词(当状态为1时允许修改关键词)
        keywordService.lambdaUpdate()
                .eq(Keyword::getOrder_id, orderDto.getOrder_id())
                .eq(Keyword::getStatus, "notSubmitted")
                .remove();
        // 保存新关键词
        if (StringUtils.hasText(orderDto.getKeywords())) {
            String[] keywordArray = orderDto.getKeywords().split("\\n");
            for (String keywordName : keywordArray) {
                if (StringUtils.hasText(keywordName)) {
                    Keyword keyword = new Keyword();
                    keyword.setOrder_id(orderDto.getOrder_id());
                    keyword.setKeyword_name(keywordName.trim());
                    keyword.setStatus("notSubmitted");
                    keywordService.save(keyword);
                }
            }
        }
        return true;
    }
    @Override
    public List<KeywordDto> getKeywordListByOrderId(String orderId) {
        List<Keyword> keywords = keywordService.lambdaQuery()
                .eq(Keyword::getOrder_id, orderId)
                .list();
        // 遍历关键词,获取每个关键词对应的提问词
        List<KeywordDto> keywordDtos = new ArrayList<>();
        for (Keyword keyword : keywords) {
            KeywordDto dto = new KeywordDto();
            BeanUtils.copyProperties(keyword, dto);
            LambdaQueryWrapper<KeywordTask> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(KeywordTask::getKeyword_id, keyword.getKeyword_id());
            List<KeywordTask> keywordTasks = keywordTaskService.list(queryWrapper);
            // 提取 task_id 列表
            List<String> taskIdList = keywordTasks.stream()
                    .map(KeywordTask::getTask_id)
                    .filter(Objects::nonNull)
                    .collect(Collectors.toList());
            dto.setTaskIdList(taskIdList); // 设置 task_id 列表
            // 查询该关键词下的所有提问词
            List<Question> questions = questionService.lambdaQuery()
                    .eq(Question::getKeyword_id, keyword.getKeyword_id())
                    .list();
            dto.setQuestionList(questions);
            keywordDtos.add(dto);
        }
        return keywordDtos;
    }
    @Override
    public Page<String> getClientList(String clientName, Integer pageNum, Integer pageSize) {
        Page<Orders> page = new Page<>(pageNum, pageSize);
        // 构建查询条件(根据客户名称模糊搜索)
        LambdaQueryWrapper<Orders> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.select(Orders::getClient_name) // 只查询客户名称字段
                .eq(Orders::getDel_flag, 0)    // 只查询未删除的订单
                .groupBy(Orders::getClient_name); // 按客户名称分组去重
        if (clientName != null && !clientName.isEmpty()) {
            queryWrapper.like(Orders::getClient_name, clientName);
        }
        // 执行分页查询
        IPage<Orders> orderPage = orderMapper.selectPage(page, queryWrapper);
        // 转换为客户选项列表(统计每个客户的订单数量)
        List<String> clientOptions = orderPage.getRecords().stream()
                .map(order -> {
                    return order.getClient_name();
                })
                .collect(Collectors.toList());
        // 构建结果分页对象
        Page<String> resultPage = new Page<>();
        resultPage.setCurrent(orderPage.getCurrent());
        resultPage.setSize(orderPage.getSize());
        resultPage.setTotal(orderPage.getTotal());
        resultPage.setRecords(clientOptions);
        return resultPage;
    }
}