package com.gk.hotwork.Service.ServiceImpl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.hotwork.Domain.TaskEquipment;
import com.gk.hotwork.Mapper.TaskEquipmentMapper;
import com.gk.hotwork.Service.TaskEquipmentService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

@Service("taskEquipmentService")
public class TaskEquipmentServiceImpl extends ServiceImpl<TaskEquipmentMapper, TaskEquipment> implements TaskEquipmentService {

    @Override
    public List<TaskEquipment> getListByTaskCode(String code) {
        LambdaQueryWrapper<TaskEquipment>wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEquipment::getTaskcode,code);
        return baseMapper.selectList(wrapper);
    }

    @Override
    public void removeByTaskCode(String code) {
        LambdaQueryWrapper<TaskEquipment>wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEquipment::getTaskcode,code);
        baseMapper.delete(wrapper);
    }

    @Override
    public void removeByTaskCodeByConstruction(String code) {
        LambdaQueryWrapper<TaskEquipment>wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEquipment::getTaskcode,code);
        List<String>list = new ArrayList<>();
        list.add("有毒有害防护设备");
        list.add("个人防护设备");
        list.add("安全措施设备及工器具");
        wrapper.in(TaskEquipment::getType,list);
        baseMapper.delete(wrapper);
    }
}