package com.whyc.service; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.whyc.mapper.PermitGroupUserMapper; import com.whyc.pojo.db_user.Permitgroup; import com.whyc.pojo.db_user.PermitgroupUsr; 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 getPermitByUser(Integer userId){ if(userId<=100){ //内置用户包含所有得权限 return mapper.getAllPermitList(); }else{ return mapper.getPermitByUser(userId); } } public void update(PermitgroupUsr groupUser, int operationFlag) { if(operationFlag==1) { mapper.insert(groupUser); }else{ UpdateWrapper wrapper = Wrappers.update(); wrapper.eq("permit_group_id",groupUser.getPermitGroupId()).eq("uid",groupUser.getUid()); mapper.delete(wrapper); } } public void updateBatch(List groupUserList, int operationFlag) { if(operationFlag==1) { mapper.insertBatchSomeColumn(groupUserList); }else{ UpdateWrapper wrapper = Wrappers.update(); wrapper.eq("permit_group_id",groupUserList.get(0).getPermitGroupId()); List uIdList = groupUserList.stream().map(e -> e.getUid()).collect(Collectors.toList()); wrapper.in("uid",uIdList); mapper.delete(wrapper); } } public PermitgroupUsr getPermitGroup(Integer uid) { QueryWrapper wrapper = Wrappers.query(); wrapper.select("permit_group_id").eq("uid",uid).last(" limit 1"); return mapper.selectOne(wrapper); } public List getItemList(Integer uId) { return mapper.getItemList(uId); } }