package com.gk.hotwork.doublePrevention.service.baseService.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.hotwork.Domain.Exception.BusinessException;
import com.gk.hotwork.Domain.Exception.E;
import com.gk.hotwork.doublePrevention.entity.PreventRiskControlMeasure;
import com.gk.hotwork.doublePrevention.entity.PreventTaskUnitAndMeasure;
import com.gk.hotwork.doublePrevention.repository.PreventTaskUnitAndMeasureRepository;
import com.gk.hotwork.doublePrevention.repository.param.PreventTaskUnitAndMeasureParams;
import com.gk.hotwork.doublePrevention.service.baseService.PreventTaskUnitAndMeasureService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

@Service("PreventTaskUnitAndMeasureService")
public class PreventTaskUnitAndMeasureServiceImpl extends ServiceImpl<PreventTaskUnitAndMeasureRepository, PreventTaskUnitAndMeasure> implements PreventTaskUnitAndMeasureService {

    @Autowired
    private PreventTaskUnitAndMeasureRepository preventTaskUnitAndMeasureRepository;

    /**
     * 保存任务单元与管控措施的关系
     * */
    @Override
    public void insert(PreventTaskUnitAndMeasure taskUnitAndMeasure) {
        int result = preventTaskUnitAndMeasureRepository.insert(taskUnitAndMeasure);
        if (result < 1){
            throw new BusinessException(E.ADD_FAIL, "保存任务单元与管控措施的关系失败");
        }
    }
    /**
     * 获取任务单元对应的管控措施
     * */
    @Override
    public List<PreventTaskUnitAndMeasure> getListByUnitId(Long taskUnitId) {
        return preventTaskUnitAndMeasureRepository.getListByUnitId(taskUnitId);
    }
    /**
     * 重置任务单元与管控措施的关系
     * */
    @Override
    public void deleteTaskUnitAndMeasure(PreventTaskUnitAndMeasureParams taskUnitAndMeasureParams) {
        int result = preventTaskUnitAndMeasureRepository.deleteTaskUnitAndMeasure(taskUnitAndMeasureParams);
        if (result < 1){
            throw new BusinessException(E.ADD_FAIL, "保存任务单元与管控措施的关系失败");
        }
    }
    /**
     * 获取任务单元对应的管控措施-by measureId
     * */
    @Override
    public List<PreventTaskUnitAndMeasure>  getListByMeasureId(Long measureId) {
        return preventTaskUnitAndMeasureRepository.getListByMeasureId(measureId);
    }
}