feat(ai): 实现公司成员申请列表过滤及同意/解绑功能

Administrator

2025-10-20 79cd87004c473dc9af6f3495892a932d72d039dd

 src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyMemberApplyServiceImpl.java

@@ -2,16 +2,21 @@

import cc.mrbird.febs.ai.entity.AiCompany;
import cc.mrbird.febs.ai.entity.AiCompanyMemberApply;
import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
import cc.mrbird.febs.ai.entity.AiMember;
import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
import cc.mrbird.febs.ai.mapper.AiCompanyMemberApplyMapper;
import cc.mrbird.febs.ai.service.AiCompanyMemberApplyService;
import cc.mrbird.febs.ai.service.AiMemberService;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.mall.entity.MallMember;
import cc.mrbird.febs.mall.mapper.MallMemberMapper;
import cn.hutool.core.collection.CollUtil;
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;
@@ -31,12 +36,17 @@

    private final MallMemberMapper mallMemberMapper;
    private final AiMemberService aiMemberService;
    private final AiCompanyMapper aiCompanyMapper;

    @Override
    public IPage<AiCompanyMemberApply> listInPage(AiCompanyMemberApply dto, QueryRequest request) {

        Page<AiCompanyMemberApply> page = new Page<>(request.getPageNum(), request.getPageSize());
        LambdaQueryWrapper<AiCompanyMemberApply> query = Wrappers.lambdaQuery(AiCompanyMemberApply.class);
        if (StrUtil.isNotEmpty(dto.getCompanyId())){
            AiCompany aiCompany = aiCompanyMapper.selectById(dto.getCompanyId());
            query.eq(AiCompanyMemberApply::getCompanyCode, aiCompany.getCode());
        }
        query.orderByDesc(AiCompanyMemberApply::getCreatedTime);
        Page<AiCompanyMemberApply> pages = this.baseMapper.selectPage(page, query);
        List<AiCompanyMemberApply> records = pages.getRecords();
@@ -65,7 +75,40 @@

    @Override
    public FebsResponse agree(AiCompanyMemberApply dto) {
        return null;
        String id = dto.getId();
        AiCompanyMemberApply aiCompanyMemberApply = this.getById(id);
        if (ObjectUtil.isEmpty(aiCompanyMemberApply)){
            throw new RuntimeException("申请不存在");
        }
        if (0 != aiCompanyMemberApply.getState()){
            throw new RuntimeException("申请已处理");
        }
        String companyCode = aiCompanyMemberApply.getCompanyCode();
        AiCompany aiCompany = aiCompanyMapper.selectOne(
                Wrappers.lambdaQuery(AiCompany.class)
                        .eq(AiCompany::getCode, companyCode)
                        .last("limit 1")
        );
        
        if (ObjectUtil.isEmpty(aiCompany)){
            throw new FebsException("公司编码不存在");
        }

        String memberUuid = aiCompanyMemberApply.getMemberId();
        String companyId = aiCompany.getId();

        mallMemberMapper.update(null,
                Wrappers.lambdaUpdate(MallMember.class)
                        .set(MallMember::getCompanyId, companyId)
                        .eq(MallMember::getMemberUuid, memberUuid)
        );

        aiMemberService.update(null,
                Wrappers.lambdaUpdate(AiMember.class)
                        .set(AiMember::getCompanyId, companyId)
                        .eq(AiMember::getId, memberUuid)
        );
        return new FebsResponse().success();
    }

    @Override
@@ -88,7 +131,40 @@

    @Override
    public FebsResponse unbind(AiCompanyMemberApply dto) {
        return null;
        String id = dto.getId();
        AiCompanyMemberApply aiCompanyMemberApply = this.getById(id);
        if (ObjectUtil.isEmpty(aiCompanyMemberApply)){
            throw new RuntimeException("申请不存在");
        }
        if (0 != aiCompanyMemberApply.getState()){
            throw new RuntimeException("申请已处理");
        }
        String companyCode = aiCompanyMemberApply.getCompanyCode();
        AiCompany aiCompany = aiCompanyMapper.selectOne(
                Wrappers.lambdaQuery(AiCompany.class)
                        .eq(AiCompany::getCode, companyCode)
                        .last("limit 1")
        );

        if (ObjectUtil.isEmpty(aiCompany)){
            throw new FebsException("公司编码不存在");
        }

        String memberUuid = aiCompanyMemberApply.getMemberId();
        String companyId = aiCompany.getId();

        mallMemberMapper.update(null,
                Wrappers.lambdaUpdate(MallMember.class)
                        .set(MallMember::getCompanyId, null)
                        .eq(MallMember::getMemberUuid, memberUuid)
        );

        aiMemberService.update(null,
                Wrappers.lambdaUpdate(AiMember.class)
                        .set(AiMember::getCompanyId, null)
                        .eq(AiMember::getId, memberUuid)
        );
        return new FebsResponse().success();
    }

}