package com.gkhy.exam.noncoalmine.service.impl;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.extension.conditions.update.UpdateChainWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.exam.noncoalmine.entity.WorkRegistration;
import com.gkhy.exam.noncoalmine.mapper.WorkRegistrationMapper;
import com.gkhy.exam.noncoalmine.model.addForm.WorkRegistrationAddForm;
import com.gkhy.exam.noncoalmine.model.modForm.WorkRegistrationModForm;
import com.gkhy.exam.noncoalmine.model.query.WorkRegistrationQuery;
import com.gkhy.exam.noncoalmine.model.vo.WorkRegistrationVO;
import com.gkhy.exam.noncoalmine.service.WorkRegistrationService;
import com.ruoyi.common.utils.bean.BeanUtils;
import com.ruoyi.system.domain.bo.SysOperateTypeWholeNameBO;
import com.ruoyi.system.service.SysOperateTypeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.stream.Collectors;

/**
 * (WorkRegistration)表服务实现类
 *
 * @author makejava
 * @since 2023-09-11 16:59:58
 */
@Service("workRegistrationService")
public class WorkRegistrationServiceImpl extends ServiceImpl<WorkRegistrationMapper, WorkRegistration> implements WorkRegistrationService {
	@Autowired
	private WorkRegistrationMapper workRegistrationMapper;
	@Autowired
	private SysOperateTypeService sysOperateTypeService;

	@Override
	public List<WorkRegistrationVO> listByPage(WorkRegistrationQuery query) {
		List<WorkRegistration> workRegistrations = workRegistrationMapper.listByPage(query);
		List<WorkRegistrationVO> list = workRegistrations.stream().map(workRegistration -> {
			WorkRegistrationVO workRegistrationVO = new WorkRegistrationVO();
			BeanUtils.copyProperties(workRegistration, workRegistrationVO);
			SysOperateTypeWholeNameBO operateTypeWholeName = sysOperateTypeService.getOperateTypeWholeNameById(workRegistration.getOperateTypeId());
			workRegistrationVO.setJobCategory(operateTypeWholeName.getJobCategory());
			workRegistrationVO.setQualificationType(operateTypeWholeName.getQualificationType());
			workRegistrationVO.setOperationItems(operateTypeWholeName.getOperationItems());
			return workRegistrationVO;
		}).collect(Collectors.toList());
		return list;
	}

	/**
	 * 新增
	 * @param addForm
	 * @return
	 */
	@Override
	public int add(WorkRegistrationAddForm addForm) {
		WorkRegistration workRegistration = new WorkRegistration();
		BeanUtils.copyProperties(addForm,workRegistration);
		workRegistration.setDelFlag((byte) 0);
		return baseMapper.insert(workRegistration);
	}
	/**
	 * 修改
	 * @param modForm
	 * @return
	 */
	@Override
	public int mod(WorkRegistrationModForm modForm) {
		WorkRegistration workRegistration = new WorkRegistration();
		BeanUtils.copyProperties(modForm,workRegistration);
		return baseMapper.updateById(workRegistration);
	}

	/**
	 * 根据身份证获取做业数据
	 */
	public List<WorkRegistrationVO> getByIdCard(String idCard, byte isCm){
		List<WorkRegistration> workRegistrations = baseMapper.selectList(new LambdaQueryWrapper<WorkRegistration>()
				.eq(WorkRegistration::getDelFlag, (byte)0)
				.eq(WorkRegistration::getIdCard, idCard)
				.eq(WorkRegistration::getIsCm,isCm));
		List<WorkRegistrationVO> list = workRegistrations.stream().map(workRegistration -> {
			WorkRegistrationVO workRegistrationVO = new WorkRegistrationVO();
			BeanUtils.copyProperties(workRegistration, workRegistrationVO);
			SysOperateTypeWholeNameBO operateTypeWholeName = sysOperateTypeService.getOperateTypeWholeNameById(workRegistration.getOperateTypeId());
			workRegistrationVO.setJobCategory(operateTypeWholeName.getJobCategory());
			workRegistrationVO.setQualificationType(operateTypeWholeName.getQualificationType());
			workRegistrationVO.setOperationItems(operateTypeWholeName.getOperationItems());
			return workRegistrationVO;
		}).collect(Collectors.toList());
		return list;
	}

	@Override
	public void delBatch(List<Long> workIds) {
		UpdateWrapper<WorkRegistration> updateWrapper = new UpdateWrapper<>();
		updateWrapper.in("work_id",workIds)
				.set("del_flag",(byte)2);
		this.update(updateWrapper);
	}
}