package com.zt.modules.sys.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.zt.common.service.BaseService;
import com.zt.modules.sys.dao.SysJobUserDao;
import com.zt.modules.sys.dto.JobDto;
import com.zt.modules.sys.model.SysJobUser;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

/**
 * SYS_JOB_USER
 *
 * @author zt generator
 * @since 1.0.0 2020-06-11
 */
@Service
public class SysJobUserService extends BaseService<SysJobUserDao, SysJobUser> {

    @Transactional(rollbackFor = Exception.class)
    public void saveOrUpdate(Long userId, List<Long> jobIdList) {
        List<SysJobUser> list = jobIdList.stream().map(jobId -> new SysJobUser(jobId, userId))
                .collect(Collectors.toList());
        this.saveRelatedDatas("user_id", userId, "job_id", list);
    }

    public List<JobDto> getUserJobs(Long userId) {
        return baseDao.getUserJobs(userId);
    }

    public void deleteByJobId(Long jobId) {
        baseDao.delete(new QueryWrapper<SysJobUser>().lambda().eq(SysJobUser::getJobId, jobId));
    }
}