whycxzp
2023-03-28 aaa2e601eb2a6806e761d4424fcaaf43a95fdcfd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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;
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);
        }
    }
 
    public int getPermitGroupId(Long uId) {
        QueryWrapper<PermitGroupUser> wrapper = Wrappers.query();
        wrapper.select("permit_group_id").eq("uId",uId).last(" limit 1");
        return mapper.selectOne(wrapper).getPermitGroupId();
    }
 
    public List<String> getItemList(Long uId) {
        return mapper.getItemList(uId);
    }
}