From a0a82fb8a2a96221de10b1355836b16bd94f77c3 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 04 Feb 2026 11:33:01 +0800
Subject: [PATCH] fix(ai): 解决产品提示实体公司ID设置问题
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPromptServiceImpl.java | 12 +++++++++---
1 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPromptServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPromptServiceImpl.java
index 50b8a98..05d4db1 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPromptServiceImpl.java
+++ b/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("操作成功");
--
Gitblit v1.9.1