|
@@ -2,6 +2,7 @@ package com.fjhx.wms.service.warehouse.impl;
|
|
|
|
|
|
import cn.hutool.core.bean.BeanUtil;
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
import com.fjhx.wms.entity.warehouse.dto.WarehouseDto;
|
|
@@ -10,14 +11,16 @@ import com.fjhx.wms.entity.warehouse.po.Warehouse;
|
|
|
import com.fjhx.wms.entity.warehouse.vo.WarehouseVo;
|
|
|
import com.fjhx.wms.mapper.warehouse.WarehouseMapper;
|
|
|
import com.fjhx.wms.service.warehouse.WarehouseService;
|
|
|
+import com.ruoyi.common.core.domain.entity.SysDept;
|
|
|
import com.ruoyi.common.utils.SecurityUtils;
|
|
|
import com.ruoyi.common.utils.wrapper.IWrapper;
|
|
|
+import com.ruoyi.system.service.ISysDeptService;
|
|
|
import com.ruoyi.system.utils.UserUtil;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
-import java.util.Arrays;
|
|
|
-import java.util.Map;
|
|
|
-import java.util.Objects;
|
|
|
+import java.util.*;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
|
|
|
/**
|
|
@@ -31,6 +34,9 @@ import java.util.Objects;
|
|
|
@Service
|
|
|
public class WarehouseServiceImpl extends ServiceImpl<WarehouseMapper, Warehouse> implements WarehouseService {
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private ISysDeptService sysDeptService;
|
|
|
+
|
|
|
@Override
|
|
|
public Page<WarehouseVo> getPage(WarehouseSelectDto dto) {
|
|
|
IWrapper<Warehouse> wrapper = getWrapper();
|
|
@@ -41,12 +47,34 @@ public class WarehouseServiceImpl extends ServiceImpl<WarehouseMapper, Warehouse
|
|
|
Long companyId = SecurityUtils.getCompanyId();
|
|
|
if (!Objects.equals(companyId, 100L)) {
|
|
|
wrapper.eq(Warehouse::getCompanyId, companyId);
|
|
|
+ } else {
|
|
|
+ wrapper.eq(Warehouse::getCompanyId, dto.getCompanyId());
|
|
|
}
|
|
|
|
|
|
wrapper.orderByDesc("w", Warehouse::getId);
|
|
|
Page<WarehouseVo> page = this.baseMapper.getPage(dto.getPage(), wrapper);
|
|
|
+ List<WarehouseVo> records = page.getRecords();
|
|
|
+
|
|
|
+ if (ObjectUtil.isEmpty(records)) {
|
|
|
+ return page;
|
|
|
+ }
|
|
|
+
|
|
|
//赋值仓管员名称
|
|
|
- UserUtil.assignmentNickName(page.getRecords(), Warehouse::getKeeperId, WarehouseVo::setKeeperName);
|
|
|
+ UserUtil.assignmentNickName(records, Warehouse::getKeeperId, WarehouseVo::setKeeperName);
|
|
|
+
|
|
|
+ //获取归属公司信息
|
|
|
+ Map<Long, String> sysDeptMap = new HashMap<>();
|
|
|
+ List<Long> companyIds = records.stream().map(Warehouse::getCompanyId).collect(Collectors.toList());
|
|
|
+ if (ObjectUtil.isNotEmpty(companyIds)) {
|
|
|
+ List<SysDept> deptList = sysDeptService.list(Wrappers.<SysDept>query().in("dept_id", companyIds));
|
|
|
+ sysDeptMap = deptList.stream().collect(Collectors.toMap(SysDept::getDeptId, SysDept::getDeptName));
|
|
|
+ }
|
|
|
+
|
|
|
+ for (WarehouseVo record : records) {
|
|
|
+ //赋值归属公司信息
|
|
|
+ record.setCompanyName(sysDeptMap.get(record.getCompanyId()));
|
|
|
+ }
|
|
|
+
|
|
|
return page;
|
|
|
}
|
|
|
|