Administrator
11 hours ago 05fb9cb0e3d2678e82048a171acbcffbdd702b59
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
package cc.mrbird.febs.ai.service.impl;
 
import cc.mrbird.febs.ai.entity.AiCompany;
import cc.mrbird.febs.ai.entity.AiMemberRole;
import cc.mrbird.febs.ai.entity.AiProductPrompt;
import cc.mrbird.febs.ai.mapper.AiProductPromptMapper;
import cc.mrbird.febs.ai.service.AiProductPromptService;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
 
import java.util.Date;
 
@Slf4j
@Service
@RequiredArgsConstructor
public class AiProductPromptServiceImpl extends ServiceImpl<AiProductPromptMapper, AiProductPrompt> implements AiProductPromptService {
 
    private final AiProductPromptMapper aiProductPromptMapper;
    @Override
    public AiProductPrompt getById(String id) {
        return aiProductPromptMapper.selectById( id);
    }
 
    @Override
    public IPage<AiProductPrompt> listInPage(AiProductPrompt dto, QueryRequest request) {
        Page<AiProductPrompt> page = new Page<>(request.getPageNum(), request.getPageSize());
        LambdaQueryWrapper<AiProductPrompt> query = Wrappers.lambdaQuery(AiProductPrompt.class);
        if (StrUtil.isNotEmpty(dto.getCompanyId())){
            query.eq(AiProductPrompt::getCompanyId, dto.getCompanyId());
            query.or().isNull(AiProductPrompt::getCompanyId);
        }
        query.orderByDesc(AiProductPrompt::getCreatedTime);
        Page<AiProductPrompt> pages = aiProductPromptMapper.selectPage(page, query);
        return pages;
    }
 
    @Override
    public FebsResponse add(AiProductPrompt dto) {
        Date createTime = new Date();
        AiProductPrompt entity = new AiProductPrompt();
        entity.setId(UUID.getSimpleUUIDString());
        entity.setName(dto.getName());
        entity.setCode(dto.getCode());
        entity.setCompanyId(dto.getCompanyId());
        entity.setCreatedTime(createTime);
        aiProductPromptMapper.insert( entity);
        return new FebsResponse().success().message("操作成功");
    }
 
    @Override
    public FebsResponse update(AiProductPrompt dto) {
        String id = dto.getId();
        AiProductPrompt entity = this.getById(id);
        if(ObjectUtil.isNotNull( entity)){
            this.update(null,
                    Wrappers.lambdaUpdate(AiProductPrompt.class)
                            .set(AiProductPrompt::getCode, dto.getCode())
                            .set(AiProductPrompt::getName, dto.getName())
                            .set(AiProductPrompt::getUpdatedTime, new Date())
                            .eq(AiProductPrompt::getId, id));
 
        }
        return new FebsResponse().success().message("操作成功");
    }
}