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.TaskRisk;
import com.gk.hotwork.Mapper.TaskRiskMapper;
import com.gk.hotwork.Service.TaskRiskService;
import org.springframework.stereotype.Service;

@Service("TaskRiskService")
public class TaskRiskServiceImpl extends ServiceImpl<TaskRiskMapper, TaskRisk> implements TaskRiskService {


    @Override
    public TaskRisk getTaskRiskByCode(String code) {
        LambdaQueryWrapper<TaskRisk>wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskRisk::getTaskcode,code);
        return baseMapper.selectOne(wrapper);
    }

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

    @Override
    public void deleteByTaskCode(String code) {
        TaskRisk taskRisk = getTaskRiskByCode(code);
        if (taskRisk != null){
            taskRisk.setFlag((byte)1);
            baseMapper.updateById(taskRisk);
        }

    }
}