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/mall/service/impl/ApiMallMemberServiceImpl.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

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