fix(yinhe): 修复公司级别限制logo的比较逻辑

Administrator

5 hours ago 248f8116f4ef74f63429c54915498eba6634bb61

 src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java

@@ -86,6 +86,13 @@
            yhMemberVo.setCompanyId(yhAiCompany.getId());
            yhMemberVo.setCompanyName(yhAiCompany.getName());
            yhMemberVo.setPhone(yhAiMember.getPhone());

            String levelCode = yhAiCompany.getLevelCode();
            YHSysCompanyLevel yhSysCompanyLevel = yhSysCompanyLevelMapper.selectOne(
                    Wrappers.lambdaQuery(YHSysCompanyLevel.class)
                            .eq(YHSysCompanyLevel::getCode, levelCode)
            );
            yhMemberVo.setLimitLogo(yhSysCompanyLevel.getLimitLogo());
        }
        return new FebsResponse().success().data(yhMemberVo);
    }
@@ -162,7 +169,6 @@
        LambdaQueryWrapper<YHAiAgent> query = Wrappers.lambdaQuery(YHAiAgent.class);

        boolean defaultCompany = isDefaultCompany(companyId);
        log.info("开始查询智能体列表{}",defaultCompany);
        if (!defaultCompany){

            List<YHAiAgentLink> yhAiAgentLinks = yhAiAgentLinkMapper.selectList(
@@ -213,7 +219,6 @@


        boolean defaultCompany = isDefaultCompany(companyId);
        log.info("开始查询智能体{}",defaultCompany);
        if (!defaultCompany){
            List<YHAiAgentLink> yhAiAgentLinks = yhAiAgentLinkMapper.selectList(
                    Wrappers.lambdaQuery(YHAiAgentLink.class)
@@ -544,7 +549,38 @@
        YhAiMember yhAiMember = yhAiMemberMapper.selectOne(Wrappers.lambdaQuery(YhAiMember.class).eq(YhAiMember::getMemberUuid, memberUuid));
        String companyId = yhAiMember.getCompanyId();



        List<YHBannerVo> vos = new ArrayList<>();

        YHAiCompany yhAiCompany = yhAiCompanyMapper.selectById(companyId);
        String levelCode = yhAiCompany.getLevelCode();
        if (StrUtil.isNotEmpty(levelCode)){
            YHSysCompanyLevel sysCompanyLevel = yhSysCompanyLevelMapper.selectOne(
                    Wrappers.lambdaQuery(YHSysCompanyLevel.class)
                            .eq(YHSysCompanyLevel::getCode, levelCode)
            );

            Integer limitLogo = sysCompanyLevel.getLimitLogo();
            if (1 == limitLogo){
                List<YHAiBanner> list = yhAiBannerMapper.selectList(
                        Wrappers.lambdaQuery(YHAiBanner.class)
                                .eq(YHAiBanner::getState, 1)
                                .isNull(YHAiBanner::getCompanyId)
                                .orderByDesc(YHAiBanner::getSort)
                );
                if (CollUtil.isNotEmpty(list)){
                    for (YHAiBanner aiBanner : list){
                        YHBannerVo vo = new YHBannerVo();
                        vo.setName(aiBanner.getName());
                        vo.setImageUrl(aiBanner.getImageUrl());
                        vo.setSort(aiBanner.getSort());
                        vos.add(vo);
                    }
                }
                return new FebsResponse().success().data(vos);
            }
        }
        List<YHAiBanner> list = yhAiBannerMapper.selectList(
                Wrappers.lambdaQuery(YHAiBanner.class)
                        .eq(YHAiBanner::getState, 1)
@@ -571,6 +607,32 @@

        YHLogoVo yhLogoVo = new YHLogoVo();

        YHAiCompany yhAiCompany = yhAiCompanyMapper.selectById(companyId);
        String levelCode = yhAiCompany.getLevelCode();
        if (StrUtil.isNotEmpty(levelCode)){
            YHSysCompanyLevel sysCompanyLevel = yhSysCompanyLevelMapper.selectOne(
                    Wrappers.lambdaQuery(YHSysCompanyLevel.class)
                            .eq(YHSysCompanyLevel::getCode, levelCode)
            );

            Integer limitLogo = sysCompanyLevel.getLimitLogo();
            if (1 == limitLogo){
                List<YHAiLogo> yhAiLogos = yhAiLogoMapper.selectList(
                        Wrappers.lambdaQuery(YHAiLogo.class)
                                .eq(YHAiLogo::getState, 1)
                                .isNull(YHAiLogo::getCompanyId)
                );
                if (CollUtil.isNotEmpty(yhAiLogos)){
                    YHAiLogo yhAiLogo = yhAiLogos.get(0);
                    yhLogoVo.setName(yhAiLogo.getName());
                    yhLogoVo.setImageUrl(yhAiLogo.getImageUrl());
                    yhLogoVo.setDesription(yhAiLogo.getDesription());
                }

                return new FebsResponse().success().data(yhLogoVo);
            }
        }

        List<YHAiLogo> yhAiLogos = yhAiLogoMapper.selectList(
                Wrappers.lambdaQuery(YHAiLogo.class)
                        .eq(YHAiLogo::getState, 1)