package com.ruoyi.system.service.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.system.mapper.TQuestionnaireMapper;
import com.ruoyi.system.mapper.TQuestionnaireRecordMapper;
import com.ruoyi.system.mapper.TQuestionnaireStemMapper;
import com.ruoyi.system.model.TAssistedDining;
import com.ruoyi.system.model.TQuestionnaire;
import com.ruoyi.system.model.TQuestionnaireRecord;
import com.ruoyi.system.model.TQuestionnaireStem;
import com.ruoyi.system.query.TQuestionnaireQuery;
import com.ruoyi.system.service.TQuestionnaireService;
import com.ruoyi.system.vo.TQuestionnaireVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 问卷调查 服务实现类
 * </p>
 *
 * @author xiaochen
 * @since 2025-11-06
 */
@Service
public class TQuestionnaireServiceImpl extends ServiceImpl<TQuestionnaireMapper, TQuestionnaire> implements TQuestionnaireService {

    @Autowired
    private TQuestionnaireStemMapper questionnaireStemMapper;
    @Autowired
    private TQuestionnaireRecordMapper questionnaireRecordMapper;

    @Override
    public PageInfo<TQuestionnaireVO> pageList(TQuestionnaireQuery query) {
        PageInfo<TQuestionnaireVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
        List<TQuestionnaireVO> list = this.baseMapper.pageList(query,pageInfo);

        if(CollectionUtils.isEmpty(list)){
            return pageInfo;
        }
        List<String> ids = list.stream().map(TQuestionnaireVO::getId).collect(Collectors.toList());
        List<TQuestionnaireStem> questionnaireStemList = questionnaireStemMapper.selectList(Wrappers.lambdaQuery(TQuestionnaireStem.class)
                .in(TQuestionnaireStem::getQuestionnaireId, ids));
        List<TQuestionnaireRecord> questionnaireRecordList = questionnaireRecordMapper.selectList(Wrappers.lambdaQuery(TQuestionnaireRecord.class)
                .in(TQuestionnaireRecord::getQuestionnaireId, ids));
        for (TQuestionnaireVO questionnaireVO : list) {
            List<TQuestionnaireStem> questionnaireStems = questionnaireStemList.stream().filter(questionnaireStem -> questionnaireVO.getId().equals(questionnaireStem.getQuestionnaireId())).collect(Collectors.toList());
            questionnaireVO.setOptionNum(questionnaireStems.size());
            List<TQuestionnaireRecord> questionnaireRecords = questionnaireRecordList.stream().filter(questionnaireRecord -> questionnaireVO.getId().equals(questionnaireRecord.getQuestionnaireId())).collect(Collectors.toList());
            questionnaireVO.setTotalNum(questionnaireRecords.size());
        }
        pageInfo.setRecords(list);
        return pageInfo;
    }
}