From 7aec9e836c4a81e111c3721f8f40a850b708a102 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 25 Sep 2025 10:01:16 +0800
Subject: [PATCH] feat(ai): 添加公司编码字段并更新相关逻辑
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberRoleServiceImpl.java | 1 +
src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java | 5 +++++
src/main/java/cc/mrbird/febs/ai/res/company/ApiCompanyVo.java | 2 ++
src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java | 12 +++++++++++-
4 files changed, 19 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java b/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
index aa09742..78aee0e 100644
--- a/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
+++ b/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
@@ -12,4 +12,9 @@
* 公司名称
*/
private String name;
+
+ /**
+ * 公司名称
+ */
+ private String code;
}
diff --git a/src/main/java/cc/mrbird/febs/ai/res/company/ApiCompanyVo.java b/src/main/java/cc/mrbird/febs/ai/res/company/ApiCompanyVo.java
index 7339719..04ef8c7 100644
--- a/src/main/java/cc/mrbird/febs/ai/res/company/ApiCompanyVo.java
+++ b/src/main/java/cc/mrbird/febs/ai/res/company/ApiCompanyVo.java
@@ -7,6 +7,8 @@
@ApiModel(value = "ApiCompanyVo", description = "参数")
public class ApiCompanyVo {
+ private String code;
+
private String name;
private String id;
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberRoleServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberRoleServiceImpl.java
index f4027f5..2b073b9 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberRoleServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberRoleServiceImpl.java
@@ -112,6 +112,7 @@
ApiCompanyVo apiCompanyVo = new ApiCompanyVo();
apiCompanyVo.setId(aiCompany.getId());
apiCompanyVo.setName(aiCompany.getName());
+ apiCompanyVo.setCode(aiCompany.getCode());
list.add(apiCompanyVo);
}
}
diff --git a/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
index 2709a51..72e82f2 100644
--- a/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
@@ -1,6 +1,8 @@
package cc.mrbird.febs.mall.service.impl;
+import cc.mrbird.febs.ai.entity.AiCompany;
import cc.mrbird.febs.ai.entity.AiMember;
+import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
import cc.mrbird.febs.ai.service.AiMemberService;
import cc.mrbird.febs.ai.utils.UUID;
@@ -71,6 +73,7 @@
@RequiredArgsConstructor
public class ApiMallMemberServiceImpl extends ServiceImpl<MallMemberMapper, MallMember> implements IApiMallMemberService {
+ private final AiCompanyMapper aiCompanyMapper;
private final AiMemberMapper aiMemberMapper;
private final MallMemberWalletMapper mallMemberWalletMapper;
private final ICommonService commonService;
@@ -737,7 +740,14 @@
AiMember aiMember = new AiMember();
aiMember.setId(UUID.getSimpleUUIDString());
if (StrUtil.isNotEmpty(apiXcxLoginDto.getCompanyId())){
- aiMember.setCompanyId(apiXcxLoginDto.getCompanyId());
+ AiCompany aiCompany = aiCompanyMapper.selectOne(
+ Wrappers.lambdaQuery(AiCompany.class)
+ .eq(AiCompany::getCode, apiXcxLoginDto.getCode())
+ .last("limit 1")
+ );
+ if (ObjectUtil.isNotEmpty(aiCompany)){
+ aiMember.setCompanyId(aiCompany.getId());
+ }
}
aiMember.setCreatedTime(new Date());
aiMemberMapper.insert(aiMember);
--
Gitblit v1.9.1