AI对接

xuhy

22 小时以前 4955cdc73d9beb5733aa2c0a578c14798394fa61

 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TCrmSupplierServiceImpl.java

@@ -1,5 +1,6 @@
package com.ruoyi.system.service.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
@@ -9,15 +10,20 @@
import com.ruoyi.system.dto.TCrmSupplierDTO;
import com.ruoyi.system.mapper.TCrmSupplierMapper;
import com.ruoyi.system.model.TCrmSupplier;
import com.ruoyi.system.model.TCrmSupplierToWarehouse;
import com.ruoyi.system.query.TCrmSupplierQuery;
import com.ruoyi.system.service.ISysUserService;
import com.ruoyi.system.service.TCrmSupplierService;
import com.ruoyi.system.service.TCrmSupplierToWarehouseService;
import com.ruoyi.system.vo.TCrmSupplierVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
@@ -32,11 +38,26 @@

    @Autowired
    private ISysUserService sysUserService;
    @Autowired
    private TCrmSupplierToWarehouseService crmSupplierToWarehouseService;

    @Override
    public PageInfo<TCrmSupplierVO> pageList(TCrmSupplierQuery query) {
        PageInfo<TCrmSupplierVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
        List<TCrmSupplierVO> list = this.baseMapper.pageList(query,pageInfo);
        if(CollectionUtils.isEmpty(list)){
            return pageInfo;
        }
        List<String> supplierIds = list.stream().map(TCrmSupplierVO::getId).collect(Collectors.toList());
        List<TCrmSupplierToWarehouse> crmSupplierToWarehouses = crmSupplierToWarehouseService.list(Wrappers.lambdaQuery(TCrmSupplierToWarehouse.class)
                .eq(TCrmSupplierToWarehouse::getSupplierId, supplierIds));
        for (TCrmSupplierVO crmSupplierVO : list) {
            List<TCrmSupplierToWarehouse> supplierToWarehouses = crmSupplierToWarehouses.stream().filter(crmSupplierToWarehouse -> crmSupplierToWarehouse.getSupplierId().equals(crmSupplierVO.getId())).collect(Collectors.toList());
            if(!CollectionUtils.isEmpty(supplierToWarehouses)){
                List<String> warehouseIds = supplierToWarehouses.stream().map(TCrmSupplierToWarehouse::getWarehouseId).collect(Collectors.toList());
                crmSupplierVO.setWarehouseIds(warehouseIds);
            }
        }
        pageInfo.setRecords(list);
        return pageInfo;
    }
@@ -49,6 +70,18 @@
            return R.fail(dto.getAccount()+"-账号已存在");
        }
        this.save(dto);
        // 添加供应商和仓库关联关系
        TCrmSupplierToWarehouse crmSupplierToWarehouse = new TCrmSupplierToWarehouse();
        List<String> warehouseIds = dto.getWarehouseIds();
        if(!CollectionUtils.isEmpty(warehouseIds)){
            List<TCrmSupplierToWarehouse> crmSupplierToWarehouses = new ArrayList<>();
            for (String warehouseId : warehouseIds) {
                crmSupplierToWarehouse.setSupplierId(dto.getId());
                crmSupplierToWarehouse.setWarehouseId(warehouseId);
                crmSupplierToWarehouses.add(crmSupplierToWarehouse);
            }
            crmSupplierToWarehouseService.saveBatch(crmSupplierToWarehouses);
        }
        // 添加账号
        SysUser user = new SysUser();
        user.setUserName(dto.getAccount());
@@ -74,6 +107,20 @@
        }
        SysUser user = sysUserService.selectUserById(dto.getUserId());
        this.updateById(dto);
        // 删除供应商和仓库关联关系
        crmSupplierToWarehouseService.remove(Wrappers.lambdaQuery(TCrmSupplierToWarehouse.class).eq(TCrmSupplierToWarehouse::getSupplierId, dto.getId()));
        // 添加供应商和仓库关联关系
        List<String> warehouseIds = dto.getWarehouseIds();
        if(!CollectionUtils.isEmpty(warehouseIds)){
            List<TCrmSupplierToWarehouse> crmSupplierToWarehouses = new ArrayList<>();
            for (String warehouseId : warehouseIds) {
                TCrmSupplierToWarehouse crmSupplierToWarehouse = new TCrmSupplierToWarehouse();
                crmSupplierToWarehouse.setSupplierId(dto.getId());
                crmSupplierToWarehouse.setWarehouseId(warehouseId);
                crmSupplierToWarehouses.add(crmSupplierToWarehouse);
            }
            crmSupplierToWarehouseService.saveBatch(crmSupplierToWarehouses);
        }
        if(Objects.nonNull(user)){
            // 修改账号
            user.setPhonenumber(dto.getPhone());