lxw
2023-08-15 160e150009b51a39fa95d9462c3798ba28d51a09
src/main/java/com/whyc/service/PermitGroupUserService.java
@@ -1,5 +1,7 @@
package com.whyc.service;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
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;
@@ -9,6 +11,7 @@
import javax.annotation.Resource;
import java.util.List;
import java.util.stream.Collectors;
@Service
public class PermitGroupUserService {
@@ -29,4 +32,26 @@
            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);
        }
    }
    public PermitGroupUser getPermitGroup(Long uId) {
        QueryWrapper<PermitGroupUser> wrapper = Wrappers.query();
        wrapper.select("permit_group_id").eq("uId",uId).last(" limit 1");
        return mapper.selectOne(wrapper);
    }
    public List<String> getItemList(Long uId) {
        return mapper.getItemList(uId);
    }
}