feat(ai): 添加 AI 用户答题记录和角色选择相关实体、Mapper 和 Service

Administrator

2025-07-30 ccefddf89bae188faad02ed845c5f9756f121591

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package cc.mrbird.febs.ai.service.impl;
 
import cc.mrbird.febs.ai.entity.AiMemberRole;
import cc.mrbird.febs.ai.mapper.AiMemberRoleMapper;
import cc.mrbird.febs.ai.service.AiMemberRoleService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.util.List;
 
/**
 * AI用户选择角色表 Service实现类
 *
 * @author yourname
 * @date 2025-07-29
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class AiMemberRoleServiceImpl extends ServiceImpl<AiMemberRoleMapper, AiMemberRole> implements AiMemberRoleService {
 
    private final AiMemberRoleMapper aiMemberRoleMapper;
 
    @Override
    public AiMemberRole getById(String id) {
        return this.getById(id);
    }
 
    @Override
    public List<AiMemberRole> getByCompanyId(String companyId) {
        LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberRole::getCompanyId, companyId);
        queryWrapper.orderByDesc(AiMemberRole::getCreatedTime);
        return this.list(queryWrapper);
    }
 
    @Override
    public AiMemberRole getByCode(String code) {
        LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberRole::getCode, code);
        return this.getOne(queryWrapper);
    }
 
    @Override
    public List<AiMemberRole> getByName(String name) {
        LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberRole::getName, name);
        queryWrapper.orderByDesc(AiMemberRole::getCreatedTime);
        return this.list(queryWrapper);
    }
 
    @Override
    public List<AiMemberRole> getByState(Integer state) {
        LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberRole::getState, state);
        queryWrapper.orderByDesc(AiMemberRole::getCreatedTime);
        return this.list(queryWrapper);
    }
 
    @Override
    public List<AiMemberRole> getByCompanyIdAndState(String companyId, Integer state) {
        LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(AiMemberRole::getCompanyId, companyId);
        queryWrapper.eq(AiMemberRole::getState, state);
        queryWrapper.orderByDesc(AiMemberRole::getCreatedTime);
        return this.list(queryWrapper);
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveRole(AiMemberRole aiMemberRole) {
        try {
            return this.save(aiMemberRole);
        } catch (Exception e) {
            log.error("保存角色失败: ", e);
            return false;
        }
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveBatchRoles(List<AiMemberRole> roles) {
        try {
            return this.saveBatch(roles);
        } catch (Exception e) {
            log.error("批量保存角色失败: ", e);
            return false;
        }
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean updateRole(AiMemberRole aiMemberRole) {
        try {
            return this.updateById(aiMemberRole);
        } catch (Exception e) {
            log.error("更新角色失败: ", e);
            return false;
        }
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean deleteById(String id) {
        try {
            return this.removeById(id);
        } catch (Exception e) {
            log.error("删除角色失败: ", e);
            return false;
        }
    }
 
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean deleteByCompanyId(String companyId) {
        try {
            LambdaQueryWrapper<AiMemberRole> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.eq(AiMemberRole::getCompanyId, companyId);
            return this.remove(queryWrapper);
        } catch (Exception e) {
            log.error("根据公司ID删除角色失败: ", e);
            return false;
        }
    }
}