feat(ai): 添加公司编码字段并更新相关逻辑

Administrator

2 days ago 7aec9e836c4a81e111c3721f8f40a850b708a102

feat(ai): 添加公司编码字段并更新相关逻辑

- 在 AiCompany 实体中新增 code 字段
- 更新 ApiCompanyVo以包含公司编码信息
- 在会员角色服务中设置公司编码
- 优化小程序登录时根据公司编码查询公司ID的逻辑
- 注入 AiCompanyMapper 以支持公司编码查询
- 使用公司编码替代直接使用 companyId 参数

 src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java

@@ -12,4 +12,9 @@
     * 公司名称
     */
    private String name;

    /**
     * 公司名称
     */
    private String code;
}

 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;

 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);
            }
        }

 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);