package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.whyc.mapper.PermitGroupUserMapper;
|
import com.whyc.pojo.PermitGroup;
|
import com.whyc.pojo.PermitGroupUser;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class PermitGroupUserService {
|
|
@Resource
|
private PermitGroupUserMapper mapper;
|
|
public List<PermitGroup> getPermitByUser(Long userId){
|
return mapper.getPermitByUser(userId);
|
}
|
|
public void update(PermitGroupUser groupUser,int operationFlag) {
|
if(operationFlag==1) {
|
mapper.insert(groupUser);
|
}else{
|
UpdateWrapper<PermitGroupUser> wrapper = Wrappers.update();
|
wrapper.eq("permit_group_id",groupUser.getPermitGroupId()).eq("uId",groupUser.getUId());
|
mapper.delete(wrapper);
|
}
|
}
|
|
public void updateBatch(List<PermitGroupUser> groupUserList, int operationFlag) {
|
if(operationFlag==1) {
|
mapper.insertBatchSomeColumn(groupUserList);
|
}else{
|
UpdateWrapper<PermitGroupUser> wrapper = Wrappers.update();
|
wrapper.eq("permit_group_id",groupUserList.get(0).getPermitGroupId());
|
List<Integer> uIdList = groupUserList.stream().map(e -> e.getUId()).collect(Collectors.toList());
|
wrapper.in("uId",uIdList);
|
mapper.delete(wrapper);
|
}
|
}
|
}
|