package com.iailab.module.model.mcs.pre.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.pojo.SortingField;
import com.iailab.framework.common.service.impl.BaseServiceImpl;
import com.iailab.framework.common.util.object.BeanUtils;
import com.iailab.module.model.mcs.pre.dao.MmPredictInfluenceFactorResultDao;
import com.iailab.module.model.mcs.pre.entity.MmPredictInfluenceFactorResultEntity;
import com.iailab.module.model.mcs.pre.service.MmPredictInfluenceFactorResultService;
import com.iailab.module.model.mcs.pre.vo.MmPredictInfluenceFactorResultListReqVO;
import com.iailab.module.model.mcs.pre.vo.MmPredictInfluenceFactorResultPageReqVO;
import com.iailab.module.model.mcs.pre.vo.MmPredictInfluenceFactorResultVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

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

@Slf4j
@Service
public class MmPredictInfluenceFactorResultServiceImpl extends BaseServiceImpl<MmPredictInfluenceFactorResultDao, MmPredictInfluenceFactorResultEntity> implements MmPredictInfluenceFactorResultService {
    @Override
    public PageResult<MmPredictInfluenceFactorResultVO> page(MmPredictInfluenceFactorResultPageReqVO params) {
        SortingField sortingField = new SortingField("time",SortingField.ORDER_DESC);
        List<SortingField> sortingFields = new ArrayList<>();
        sortingFields.add(sortingField);
        QueryWrapper<MmPredictInfluenceFactorResultEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("factor_id",params.getFactorId());
        queryWrapper.ge(params.getStartTime() != null,"time",params.getStartTime());
        queryWrapper.le(params.getEndTime() != null,"time",params.getEndTime());

        PageResult<MmPredictInfluenceFactorResultEntity> result = baseDao.selectPage(params, sortingFields, queryWrapper);
        return new PageResult<>(BeanUtils.toBean(result.getList(),MmPredictInfluenceFactorResultVO.class), result.getTotal());
    }

    @Override
    public void insert(List<MmPredictInfluenceFactorResultEntity> resultList) {
        baseDao.insertOrUpdate(resultList);
    }

    @Override
    public Map<String,List<MmPredictInfluenceFactorResultVO>> getResultList(MmPredictInfluenceFactorResultListReqVO params) {
        List<MmPredictInfluenceFactorResultVO> list = baseDao.getResultList(params);
        Map<String, List<MmPredictInfluenceFactorResultVO>> result = list.stream().collect(Collectors.groupingBy(MmPredictInfluenceFactorResultVO::getOutputId));
        return result;
    }
}