package com.qianwen.smartman.common.websocket.fms;
|
|
import cn.hutool.core.collection.ListUtil;
|
import com.alibaba.fastjson.JSONObject;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
import com.qianwen.smartman.common.cache.ParamCache;
|
import com.qianwen.smartman.common.constant.FmsConstant;
|
import com.qianwen.core.tool.utils.Func;
|
import com.qianwen.core.websocket.config.WebSocketMessageSender;
|
import com.qianwen.core.websocket.distribute.MessageDO;
|
import com.qianwen.core.websocket.distribute.RedisMessageDistributor;
|
import com.qianwen.core.websocket.handler.JsonMessageHandler;
|
import com.qianwen.smartman.modules.cps.service.IWarehouseAreaService;
|
import com.qianwen.smartman.modules.cps.service.IWorkstationService;
|
import com.qianwen.smartman.modules.fms.entity.FmsRealTimeTray;
|
import com.qianwen.smartman.modules.fms.service.IFmsRealTimeTrayService;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.socket.WebSocketSession;
|
|
@Component
|
public class FmsVisualDataMessageHandler implements JsonMessageHandler<FmsVisualDataRequestMessage> {
|
private static String fmsCarryTest = " {\"id\": 333, \"code\": \"003\", \"name\": \"搬运\", \"color\": \"#3e607c\", \"workBenchList\": [{\"name\": \"10\", \"id\": \"bench3\", \"color\": \"#0FC015\"}] }";
|
private static String fmsManualStationTest = "{\"id\": 444, \"code\": \"004\", \"name\": \"人工卸货站\", \"color\": \"#3e607c\", \"workBenchList\": [{\"name\": \"18\", \"id\": \"bench4\", \"color\": \"#0FC015\"}] }";
|
private final IWarehouseAreaService warehouseAreaService;
|
private final IWorkstationService workstationService;
|
private final IFmsRealTimeTrayService realTimeTrayService;
|
private final FmsVisualDataFilter fmsVisualDataFilter;
|
private final RedisMessageDistributor messageDistributor;
|
|
public FmsVisualDataMessageHandler(final IWarehouseAreaService warehouseAreaService, final IWorkstationService workstationService, final IFmsRealTimeTrayService realTimeTrayService, final FmsVisualDataFilter fmsVisualDataFilter, final RedisMessageDistributor messageDistributor) {
|
this.warehouseAreaService = warehouseAreaService;
|
this.workstationService = workstationService;
|
this.realTimeTrayService = realTimeTrayService;
|
this.fmsVisualDataFilter = fmsVisualDataFilter;
|
this.messageDistributor = messageDistributor;
|
}
|
|
public void handle(WebSocketSession session, FmsVisualDataRequestMessage message) {
|
WebSocketMessageSender.send(session, getResponseMessage(message));
|
}
|
|
public String type() {
|
return "fmsVisualData";
|
}
|
|
public Class<FmsVisualDataRequestMessage> getMessageClass() {
|
return FmsVisualDataRequestMessage.class;
|
}
|
|
public void pushData() {
|
this.fmsVisualDataFilter.dataMap.forEach((sessionId, message) -> {
|
MessageDO messageDO = new MessageDO();
|
messageDO.setSessionIds(ListUtil.toList(new String[]{sessionId}));
|
messageDO.setNeedBroadcast(false);
|
messageDO.setMessageText(JSONObject.toJSONString(getResponseMessage(message)));
|
this.messageDistributor.distribute(messageDO);
|
});
|
}
|
|
public FmsVisualDataResponseMessage getResponseMessage(FmsVisualDataRequestMessage message) {
|
List<String> fmsAreaListStr = message.getFmsAreaList();
|
List<String> fmsCarryListStr = message.getFmsCarryList();
|
List<String> fmsWorkstationListStr = message.getFmsWorkstationList();
|
List<String> fmsManualStationList = message.getFmsManualStationList();
|
FmsVisualDataResponseMessage responseMessage = new FmsVisualDataResponseMessage();
|
List<FmsRealTimeTray> realTimeTrayList = this.realTimeTrayService.list();
|
List<String> positionList = (List) realTimeTrayList.stream().map((v0) -> {
|
return v0.getCurrentPosition();
|
}).filter(currentPosition -> {
|
return Func.isNotEmpty(currentPosition);
|
}).collect(Collectors.toList());
|
Map<String, FmsRealTimeTray> realTimeTrayMap = (Map) realTimeTrayList.stream().filter(c -> {
|
return Func.isNotEmpty(c.getTrayCode()) && Func.isNotEmpty(c.getCurrentPosition());
|
}).collect(Collectors.toMap((v0) -> {
|
return v0.getTrayCode();
|
}, r -> {
|
return r;
|
}));
|
if (Func.isNotEmpty(fmsAreaListStr)) {
|
responseMessage.setFmsAreaList(this.warehouseAreaService.assembleAreaData(fmsAreaListStr, realTimeTrayMap));
|
}
|
if (Func.isNotEmpty(fmsCarryListStr)) {
|
responseMessage.setFmsCarryList(this.workstationService.assemblyWorkstationData(fmsCarryListStr, realTimeTrayMap, realTimeTrayList, positionList));
|
}
|
if (Func.isNotEmpty(fmsWorkstationListStr)) {
|
responseMessage.setFmsWorkstationList(this.workstationService.assemblyWorkstationData(fmsWorkstationListStr, realTimeTrayMap, realTimeTrayList, positionList));
|
}
|
if (Func.isNotEmpty(fmsManualStationList)) {
|
responseMessage.setFmsManualStationList(this.workstationService.assemblyWorkstationData(fmsManualStationList, realTimeTrayMap, realTimeTrayList, positionList));
|
}
|
responseMessage.setIsAuto(ParamCache.getValue(FmsConstant.PARAM_KEY));
|
return responseMessage;
|
}
|
}
|
