|
@@ -1,9 +1,20 @@
|
|
|
package com.persagy.proxy.adm.service.impl;
|
|
|
|
|
|
+import cn.hutool.core.collection.CollUtil;
|
|
|
+import com.persagy.proxy.adm.model.AdmEquipment;
|
|
|
import com.persagy.proxy.adm.model.AdmShaft;
|
|
|
+import com.persagy.proxy.adm.request.AdmQueryCriteria;
|
|
|
+import com.persagy.proxy.adm.request.AdmResponse;
|
|
|
+import com.persagy.proxy.adm.service.IAdmEquipmentService;
|
|
|
import com.persagy.proxy.adm.service.IAdmShaftService;
|
|
|
+import com.persagy.proxy.adm.utils.AdmContextUtil;
|
|
|
+import com.persagy.proxy.common.entity.InstanceUrlParam;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
/**
|
|
|
* ADM 竖井 实现类
|
|
|
* @author lvxy
|
|
@@ -11,4 +22,28 @@ import org.springframework.stereotype.Service;
|
|
|
*/
|
|
|
@Service
|
|
|
public class AdmShaftServiceImpl extends AbstractAdmBaseServiceImpl<AdmShaft> implements IAdmShaftService {
|
|
|
+ @Autowired
|
|
|
+ private IAdmEquipmentService equipmentService;
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void ensureSpecialValues(InstanceUrlParam context, List<AdmShaft> vos) {
|
|
|
+ if(CollUtil.isEmpty(vos)) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ //收集设备对象
|
|
|
+ List<AdmEquipment> equipmentList = new ArrayList<>();
|
|
|
+ vos.stream().forEach(admShaft -> {
|
|
|
+ if(CollUtil.isNotEmpty(admShaft.getEquipmentList())){
|
|
|
+ equipmentList.addAll(admShaft.getEquipmentList());
|
|
|
+ }
|
|
|
+ });
|
|
|
+ equipmentService.ensureSpecialValues(AdmContextUtil.toDmpContext(), equipmentList);
|
|
|
+ }
|
|
|
+ @Override
|
|
|
+ public AdmResponse doQuery(InstanceUrlParam context, AdmQueryCriteria request, Class<AdmShaft> clazz) {
|
|
|
+ AdmResponse response = super.doQuery(context, request, clazz);
|
|
|
+ // 系统需要补充分类名称属性
|
|
|
+ ensureSpecialValues(context, (List<AdmShaft>) response.getContent());
|
|
|
+ return response;
|
|
|
+ }
|
|
|
}
|