Administrator
10 hours ago 45391bbd1117ddac9dd91d8b2223381176713c37
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPromptServiceImpl.java
@@ -10,6 +10,7 @@
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;
@@ -36,7 +37,10 @@
    public IPage<AiProductPrompt> listInPage(AiProductPrompt dto, QueryRequest request) {
        Page<AiProductPrompt> page = new Page<>(request.getPageNum(), request.getPageSize());
        LambdaQueryWrapper<AiProductPrompt> query = Wrappers.lambdaQuery(AiProductPrompt.class);
        query.eq(AiProductPrompt::getCompanyId, dto.getCompanyId());
        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;
@@ -48,8 +52,10 @@
        AiProductPrompt entity = new AiProductPrompt();
        entity.setId(UUID.getSimpleUUIDString());
        entity.setName(dto.getName());
        entity.setCode(ShareCodeUtil.toSerialCode(1L));
        entity.setCompanyId(dto.getCompanyId());
        entity.setCode(dto.getCode());
        if (StrUtil.isNotEmpty(dto.getCompanyId())){
            entity.setCompanyId(dto.getCompanyId());
        }
        entity.setCreatedTime(createTime);
        aiProductPromptMapper.insert( entity);
        return new FebsResponse().success().message("操作成功");