package com.zt.life.core.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.constant.Layer;
|
import com.zt.common.constant.Status;
|
import com.zt.common.db.query.QueryFilter;
|
import com.zt.common.entity.MapData;
|
import com.zt.common.exception.RenException;
|
import com.zt.common.service.BaseService;
|
import com.zt.common.utils.CacheUtils;
|
import com.zt.common.utils.CommonUtils;
|
import com.zt.common.utils.TreeUtils;
|
import com.zt.core.context.UserContext;
|
import com.zt.core.sys.model.SysDept;
|
import com.zt.core.sys.service.ISysDeptService;
|
import com.zt.life.core.constant.Cache;
|
import com.zt.life.core.constant.ProjectStatus;
|
import com.zt.life.core.dao.ProductDao;
|
import com.zt.life.core.enums.CompanyType;
|
import com.zt.life.core.model.Product;
|
import org.apache.commons.lang.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.cache.annotation.CacheEvict;
|
import org.springframework.cache.annotation.Cacheable;
|
import org.springframework.stereotype.Service;
|
import org.springframework.ui.Model;
|
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* 产品结构树，id为数据主键，productId为产品主键
|
*
|
* 本类隐藏id，手工将productId作为id返回给外部使用
|
*
|
* @author zt generator
|
* @since 1.0.0 2020-07-15
|
*/
|
@Service
|
public class ProductService extends BaseService<ProductDao, Product> {
|
|
@Value("${zt.oss.local-area}")
|
private String localArea;
|
|
@Autowired
|
private ISysDeptService sysDeptService;
|
|
|
|
public List<Product> getLevelList(String l1) {
|
List<Product> productList = baseDao.getLevelList(l1);
|
return productList;
|
}
|
|
public List<Product> page(QueryFilter queryFilter) {
|
return query(queryFilter);
|
}
|
|
|
/**
|
* 查询所有，包含历史版本
|
*
|
* @return
|
*/
|
public List<Product> getAll() {
|
// 这里手动缓存，不然在当前方法内调用缓存无效
|
List<Product> all = (List<Product>) CacheUtils.get(Cache.PRODUCT, "all");
|
if (all == null) {
|
all = baseDao.getAll();
|
CacheUtils.put(Cache.PRODUCT, "all", all);
|
}
|
return all.stream().map(product -> {
|
try {
|
Product p = product.clone();
|
p.setId(p.getProductId()); // 将productId设置给id
|
return p;
|
} catch (CloneNotSupportedException e) {
|
}
|
return null;
|
}).collect(Collectors.toList());
|
}
|
public List<Product> getShipList(){
|
String area = null;
|
if (localArea.equals("qd") || localArea.equals("sy"))
|
area = localArea;
|
List<Product> list = baseDao.getShipList(area);
|
return list;
|
}
|
public List<Product> getSubList(String pid){
|
List<Product> list = baseDao.getSubList(pid);
|
return list;
|
}
|
public List<Product> getListByIds(String ids){
|
List<Product> list = baseDao.getListByIds(ids);
|
return list;
|
}
|
/**
|
* 获取树，懒加载
|
*
|
* @param parentId
|
* @return
|
*/
|
public List<Product> getProductTree(Long parentId) {
|
List<Product> list = new ArrayList<>();
|
if (parentId == 0L) { // 首先只查询两级
|
list.addAll(TreeUtils.build(this.getWithDescendant(0L)));
|
list.stream().forEach(product -> product.getChildren().stream().forEach(product1 -> {
|
product1.setHasChildren(product1.getChildren().size() > 0);
|
product1.getChildren().clear();
|
}));
|
} else {
|
list.addAll(this.getChildren(parentId));
|
Map<Long, Boolean> childrenMap = new HashMap<>();
|
this.getAll().stream().forEach(product -> {
|
if (!childrenMap.containsKey(product.getParentProductId())) {
|
childrenMap.put(product.getParentProductId(), true);
|
}
|
});
|
list.stream().forEach(product -> product.setHasChildren(childrenMap.containsKey(product.getId())));
|
}
|
return list;
|
}
|
|
/**
|
* 产品结构树
|
*
|
* @param parentIds
|
* @param showLevel
|
* @return
|
*/
|
public List<Product> getProductTree2(String parentIds, String showLevel) {
|
/* Product currentUserTProduct = null;// 当前用户所属的T
|
long companyId = UserContext.getUser().getCompanyId();
|
// companyId = 1304648154755698690L;
|
if (companyId > 0) {
|
SysDept dept = sysDeptService.get(companyId);
|
if (dept == null) {
|
throw new RenException("用户所属单位不存在！");
|
}
|
if (CompanyType.T.getValue().equals(dept.getNature())) {// T用户登录
|
List<Product> list = this.getAll().stream().filter(product -> dept.getCode().equals(product.getCode())).collect(Collectors.toList());
|
if (list.size() > 0) {
|
currentUserTProduct = list.get(0);
|
}
|
}
|
}
|
List<Product> list = new ArrayList<>();
|
if (StringUtils.isBlank(parentIds) || "0".equals(parentIds)) {
|
final Product productTmp = currentUserTProduct;
|
if (productTmp == null) {
|
list.addAll(this.getWithDescendant(0L));
|
} else {
|
list.addAll(this.getAll().stream().filter(product -> productTmp.getParentProductId().equals(product.getProductId())).collect(Collectors.toList()));
|
list.addAll(this.getWithDescendant(productTmp.getProductId()));
|
}
|
} else {
|
String[] ids = parentIds.split(",");
|
for (String id : ids) {
|
try {
|
list.addAll(this.getWithDescendant(Long.parseLong(id)));
|
} catch (NumberFormatException e) {
|
}
|
}
|
}*/
|
|
List<Product> list = new ArrayList<>();
|
list.addAll(this.getWithDescendant(0L));
|
|
if (StringUtils.isBlank(showLevel)) {
|
showLevel = "equipment";
|
}
|
String levels= "";
|
/* if ("side".equals(showLevel)){
|
levels = "'model','side'";
|
}
|
else if ("system1".equals(showLevel)){
|
levels = "'model','side','system1'";
|
}
|
else if ("system2".equals(showLevel)) {
|
levels = "'model','side','system1','system2'";
|
}
|
else if ("equipment".equals(showLevel)) {
|
levels = "'model','side','system1','system2','equipment'";
|
}
|
List<Product> list = baseDao.getProduct(levels);*/
|
//int level = Integer.parseInt(showLevel.substring(1));
|
Integer level = Layer.get(showLevel);
|
list = list
|
.stream()
|
.filter(product -> Layer.get(product.getLevel()) <= level).collect(Collectors.toList());
|
List<Product> trees = TreeUtils.build(list);
|
return trees;
|
}
|
|
public List<Product> getProductTree(String parentIds, String showLevel, String pageCode) {
|
|
/* if (StringUtils.isBlank(showLevel)) {
|
showLevel = "equipment";
|
}*/
|
/* String levels= "";
|
if ("side".equals(showLevel)){
|
levels = "'model','side'";
|
}
|
else if ("system1".equals(showLevel)){
|
levels = "'model','side','system1'";
|
}
|
else if ("system2".equals(showLevel)) {
|
levels = "'model','side','system1','system2'";
|
}
|
else if ("equipment".equals(showLevel)) {
|
levels = "'model','side','system1','system2','equipment'";
|
}
|
List<Product> list = baseDao.getProduct(levels);*/
|
//int level = Integer.parseInt(showLevel.substring(1));
|
|
String msg = "";
|
Date beginDate = new Date();
|
List<Product> projectList = getProjectByProductId(null);
|
msg = "工程查询时间：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
|
System.out.println(msg);
|
Product newestProjectProduct = getProductIdAndProjectId();
|
msg = "工程查询时间2：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
|
System.out.println(msg);
|
|
List<Product> modelList = baseDao.getProduct("'model'",null);
|
msg = "产品节点查询时间：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
|
System.out.println(msg);
|
int i=0;
|
for (Product model : modelList) {
|
model.setProductIds(model.getId().toString()+","+newestProjectProduct.getProductId().toString());
|
model.setProjectId(newestProjectProduct.getId());
|
|
List<Product> sideList = baseDao.getProduct("'side'",model.getId());
|
msg = "产品节点查询时间"+i+"：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
|
System.out.println(msg);
|
model.setChildren(sideList);
|
if (!"lxProjectList".equals(pageCode)) {
|
for (Product side : sideList) {
|
side.setChildren(projectList.stream().filter(item -> side.getId().equals(item.getProductId())).collect(Collectors.toList()));
|
}
|
}
|
i++;
|
}
|
msg = "产品节点查询时间(总)：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
|
System.out.println(msg);
|
return modelList;
|
}
|
|
@CacheEvict(value = Cache.PRODUCT, allEntries = true)
|
public void insert(Product entity) {
|
if (entity.getId() == null || entity.getId() == 0L) {
|
entity.setId(IdWorker.getId());
|
entity.setProductId(entity.getId());// 初始productId和id一致
|
}
|
if (entity.getPid() == null) {
|
entity.setPid(Constant.Sys.DEPT_ROOT);
|
}
|
Product parent = getByProductId(entity.getPid());
|
|
if (parent != null) {
|
entity.setParentProductIds(parent.getParentProductIds() + "," + entity.getId());
|
} else {
|
entity.setParentProductIds(Constant.Sys.DEPT_ROOT + "");
|
}
|
entity.setStatus(Status.ENABLE.getValue());
|
entity.setVersion(1);
|
setProductId(entity,parent);
|
baseDao.insert(entity);
|
}
|
|
@Override
|
@CacheEvict(value = Cache.PRODUCT, allEntries = true)
|
public void update(Product entity) {
|
// 这里id为productId，需要转换一下
|
Product p = getByProductId(entity.getId());
|
if (p == null) {
|
throw new RenException("产品节点不存在");
|
}
|
entity.setId(p.getId());// 设置为原始id，通过id保存
|
Product parent = getByProductId(entity.getPid());
|
setProductId(entity,parent);
|
super.update(entity);
|
}
|
|
@CacheEvict(value = Cache.PRODUCT, allEntries = true)
|
public void deleteLogic(Long[] ids) {
|
for (int i = 0; i < ids.length; i++) {
|
Long id = ids[i];
|
List<Product> subList = getChildren(id);
|
if (subList.size() > 0) {
|
throw new RenException("请先删除下级节点!");
|
}
|
Product p = getByProductId(id);
|
deleteLogic(p);
|
}
|
}
|
|
@Override
|
@Cacheable(value = Cache.PRODUCT, key = "'id:' + #id")
|
public Product get(Long id) {
|
Product product = this.getByProductId(id);
|
product.setId(product.getProductId());
|
return product;
|
}
|
|
@Cacheable(value = Cache.PRODUCT, key = "'product:' + #id")
|
public Product getByProductId(Long id) {
|
if(id == null || id == 0) {
|
return new Product();
|
}
|
Product bean = baseDao.selectOne(new QueryWrapper<Product>().lambda().eq(Product::getProductId, id)
|
.eq(Product::getStatus, Status.ENABLE.getValue()));
|
if(bean == null) {
|
bean = new Product();
|
}
|
return bean;
|
}
|
|
/**
|
* 通过id查找所有祖先，按照祖先的层级排序
|
*
|
* @param id
|
* @return
|
*/
|
public List<Product> getAncestor(Long id) {
|
return TreeUtils.getAncestor(getAll(), id);
|
}
|
|
/**
|
* 获取自己和所有下级
|
*
|
* @param id
|
* @return
|
*/
|
public List<Product> getWithDescendant(Long id) {
|
return TreeUtils.getWithDescendant(getAll(), id);
|
}
|
|
@Cacheable(value = Cache.PRODUCT, key = "'withdescendant:' + #id")
|
public List<Long> getWithDescendantIds(Long id) {
|
return this.getWithDescendant(id).stream().map(dept -> dept.getId()).collect(Collectors.toList());
|
}
|
|
/**
|
* 获取下级
|
*
|
* @param id
|
* @return
|
*/
|
public List<Product> getDescendant(Long id) {
|
return TreeUtils.getDescendant(getAll(), id);
|
}
|
|
/**
|
* 获取某产品的子节点
|
*
|
* @param id
|
* @return
|
*/
|
public List<Product> getChildren(Long id) {
|
return getAll().stream().filter(dept -> dept.getPid().equals(id)).collect(Collectors.toList());
|
}
|
|
/**
|
* 改换装，升级产品节点版本
|
*
|
* @param product
|
*/
|
public void replace(Product product) {
|
if (product.getProductId() == null) {
|
throw new RenException("缺少产品ProductId");
|
}
|
Product db = this.getByProductId(product.getProductId());
|
if (db == null) {
|
throw new RenException("产品节点不存在");
|
}
|
// 设置原来的版本无效
|
db.setStatus(Status.DISABLE.getValue());
|
baseDao.updateById(db);
|
|
// 添加新版本
|
product.setVersion(db.getVersion() + 1);
|
product.setStatus(Status.ENABLE.getValue());
|
baseDao.insert(product);
|
}
|
|
public void setProductId(Product product, Product parent) {
|
String level = product.getLevel();
|
String parentLevel = parent.getLevel();
|
// 型号
|
if (StringUtils.equals("L1",level)) {
|
|
}
|
// X号
|
if (StringUtils.equals("L2",level)) {
|
product.setShipId(product.getId());
|
product.setShipProductId(product.getId());
|
}
|
// 系统
|
if (StringUtils.equals("L3",level)) {
|
// 系统
|
if (StringUtils.equals("L2",parentLevel)) {
|
product.setShipId(parent.getShipId());
|
product.setShipProductId(parent.getShipProductId());
|
product.setSysId(product.getId());
|
product.setSysProductId(product.getId());
|
}
|
// 子系统
|
if (StringUtils.equals("L3",parentLevel)) {
|
product.setShipId(parent.getShipId());
|
product.setShipProductId(parent.getShipProductId());
|
product.setSysId(parent.getId());
|
product.setSysProductId(parent.getId());
|
product.setSubSysId(product.getId());
|
product.setSubSysProductId(product.getId());
|
}
|
}
|
// 设备
|
if (StringUtils.equals("L4",level)) {
|
product.setDeviceId(product.getId());
|
product.setDeviceProductId(product.getId());
|
// 父节点为系统
|
if (StringUtils.equals("L3",parentLevel)) {
|
Product sup_parent = getByProductId(parent.getPid());
|
String supParentLevel = sup_parent.getLevel();
|
// 父节点的父节点为X号,则父节点为系统
|
if (StringUtils.equals("L2",supParentLevel)) {
|
product.setShipId(sup_parent.getShipId());
|
product.setShipProductId(sup_parent.getShipProductId());
|
product.setSysId(sup_parent.getId());
|
product.setSysProductId(sup_parent.getId());
|
}
|
// 父节点的父节点为系统,则父节点为子系统
|
if (StringUtils.equals("L3",supParentLevel)) {
|
product.setShipId(sup_parent.getShipId());
|
product.setShipProductId(sup_parent.getShipProductId());
|
product.setSysId(sup_parent.getSysId());
|
product.setSysProductId(parent.getSysProductId());
|
product.setSubSysId(sup_parent.getId());
|
product.setSubSysProductId(sup_parent.getId());
|
}
|
|
}
|
// 父节点为设备,沿用父节点系统/子系统编号
|
if (StringUtils.equals("L4",parentLevel)) {
|
product.setShipId(parent.getShipId());
|
product.setShipProductId(parent.getShipProductId());
|
product.setSysId(parent.getSysId());
|
product.setSysProductId(parent.getSysProductId());
|
product.setSubSysId(product.getSubSysId());
|
product.setSubSysProductId(product.getSubSysProductId());
|
}
|
}
|
}
|
public Map<String, String> getMapAllNodeByShipId(Boolean idToNmae, Long shipId) {
|
Map<String, String> result = new HashMap<>();
|
List<MapData> list = baseDao.getAllNodeByShipId(shipId);
|
for (MapData item:list) {
|
if (idToNmae)
|
result.put(item.getId().toString(), item.getName());
|
else
|
result.put(item.getName(), item.getId().toString());
|
}
|
return result;
|
}
|
|
public Long getIdByName(String name) {
|
return baseDao.getIdByName(name);
|
}
|
|
public Long getIdByNameByProductId(String name,Long productId) {
|
return baseDao.getIdByNameByProductId(name,productId);
|
}
|
|
public List<Product> getProjectByProductId(Long sideId) {
|
return baseDao.getProjectByProductId(sideId);
|
}
|
|
public Product getProductIdAndProjectId() {
|
return baseDao.getProductIdAndProjectId();
|
}
|
}
|
