|
|
|
@ -2,11 +2,10 @@ package com.thing.websocket; |
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollectionUtil; |
|
|
|
import cn.hutool.core.map.MapUtil; |
|
|
|
|
|
|
|
import com.google.common.collect.Lists; |
|
|
|
import com.alibaba.fastjson.JSONObject; |
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper; |
|
|
|
import com.fasterxml.jackson.databind.node.ObjectNode; |
|
|
|
import com.google.gson.Gson; |
|
|
|
import com.google.gson.JsonElement; |
|
|
|
import com.google.gson.JsonObject; |
|
|
|
import com.google.gson.reflect.TypeToken; |
|
|
|
import com.thing.common.core.event.QueueSocketEvent; |
|
|
|
import com.thing.common.core.message.MessageData; |
|
|
|
@ -15,10 +14,7 @@ import com.thing.thing.tskv.dto.ThingAttrDTO; |
|
|
|
import com.thing.thing.tskv.dto.TsKvDTO; |
|
|
|
import com.thing.thing.tskv.service.TskvService; |
|
|
|
import com.thing.websocket.data.SocketDataCache; |
|
|
|
|
|
|
|
import lombok.extern.slf4j.Slf4j; |
|
|
|
|
|
|
|
import org.apache.commons.lang3.StringUtils; |
|
|
|
import org.springframework.beans.factory.annotation.Autowired; |
|
|
|
import org.springframework.context.event.EventListener; |
|
|
|
import org.springframework.stereotype.Component; |
|
|
|
@ -27,7 +23,6 @@ import java.lang.reflect.Type; |
|
|
|
import java.util.ArrayList; |
|
|
|
import java.util.List; |
|
|
|
import java.util.Map; |
|
|
|
import java.util.Set; |
|
|
|
import java.util.stream.Collectors; |
|
|
|
|
|
|
|
/** |
|
|
|
@ -54,9 +49,12 @@ public class QueueSocketEventListener { |
|
|
|
//超级API |
|
|
|
if (MapUtil.isNotEmpty(webSocketServer.API_SOCKET_MAP)) { |
|
|
|
webSocketServer.API_SOCKET_MAP.forEach((sessionId, thingList) -> { |
|
|
|
JsonObject jsonObject = getApiMap(lastAttrValMap, thingList); |
|
|
|
if (jsonObject != null && !jsonObject.isJsonNull() && !jsonObject.entrySet().isEmpty()) { |
|
|
|
webSocketServer.sendMessage(sessionId, new MessageData<>().data(jsonObject)); |
|
|
|
JSONObject jsonObject = getApiMap(lastAttrValMap, thingList); |
|
|
|
if(null != jsonObject){ |
|
|
|
JSONObject result = jsonObject.getJSONObject("result"); // 获取 result |
|
|
|
if(result.containsKey("values")){ |
|
|
|
webSocketServer.sendMessage(sessionId, new MessageData<>().data(jsonObject)); |
|
|
|
} |
|
|
|
} |
|
|
|
}); |
|
|
|
} |
|
|
|
@ -120,33 +118,44 @@ public class QueueSocketEventListener { |
|
|
|
private final Gson gson = new Gson(); |
|
|
|
private final Type listType = new TypeToken<List<QueueMsgDTO>>() {}.getType(); |
|
|
|
|
|
|
|
private JsonObject getApiMap(Map<String, List<QueueMsgDTO>> lastAttrValMap, JsonObject jsonObject) { |
|
|
|
// 使用 Gson 将第一个 JsonObject 转换为 JsonElement |
|
|
|
JsonElement jsonElement = gson.toJsonTree(jsonObject); |
|
|
|
// 将 JsonElement 转换为第二个 JsonObject |
|
|
|
JsonObject dataJson = jsonElement.getAsJsonObject(); |
|
|
|
JsonObject result = dataJson.get("result").getAsJsonObject(); |
|
|
|
JsonObject info = result.get("info").getAsJsonObject(); |
|
|
|
List<QueueMsgDTO> queueMsgDTOList = Lists.newArrayList(); |
|
|
|
for (String thingCode : info.keySet()) { |
|
|
|
JsonObject object = info.get(thingCode).getAsJsonObject(); |
|
|
|
Set<String> thingAttrList = object.get("attrs").getAsJsonObject().keySet(); |
|
|
|
List<QueueMsgDTO> queueMsgDTOS = lastAttrValMap.get(thingCode); |
|
|
|
if(CollectionUtil.isNotEmpty(thingAttrList) && CollectionUtil.isNotEmpty(queueMsgDTOS)){ |
|
|
|
List<QueueMsgDTO> collect = queueMsgDTOS.stream() |
|
|
|
.filter(s -> thingAttrList.contains(s.getAttrKey()) && StringUtils.equals(thingCode, s.getThingCode())) |
|
|
|
.collect(Collectors.toList()); |
|
|
|
queueMsgDTOList.addAll(collect); |
|
|
|
} |
|
|
|
private JSONObject getApiMap(Map<String, List<QueueMsgDTO>> lastAttrValMap, JSONObject jsonObject) { |
|
|
|
// 获取 "result" 对象 |
|
|
|
JSONObject result = jsonObject.getJSONObject("result"); // 获取 result |
|
|
|
if (result == null) { |
|
|
|
return null; // 如果 "result" 为 null，返回空 JSON 对象 |
|
|
|
} |
|
|
|
if(CollectionUtil.isEmpty(queueMsgDTOList)){ |
|
|
|
return new JsonObject(); |
|
|
|
JSONObject info = result.getJSONObject("info"); // 获取 info |
|
|
|
if (info == null) { |
|
|
|
return null; // 如果 "info" 为 null，返回空 JSON 对象 |
|
|
|
} |
|
|
|
List<QueueMsgDTO> queueMsgDTOList = new ArrayList<>(); |
|
|
|
info.forEach((k,v)->{ |
|
|
|
ObjectMapper objectMapper = new ObjectMapper(); |
|
|
|
ObjectNode objectNode = objectMapper.createObjectNode(); |
|
|
|
String string = info.getString(k); |
|
|
|
JSONObject jsonObject1 = JSONObject.parseObject(string); |
|
|
|
String thingCode = jsonObject1.getString("entityCode"); |
|
|
|
boolean b = lastAttrValMap.containsKey(thingCode); |
|
|
|
if(b){ |
|
|
|
JSONObject attrs = jsonObject1.getJSONObject("attrs"); |
|
|
|
List<QueueMsgDTO> queueMsgDTOS = lastAttrValMap.get(thingCode); |
|
|
|
if (CollectionUtil.isNotEmpty(queueMsgDTOS)) { |
|
|
|
List<QueueMsgDTO> collect = queueMsgDTOS.stream() |
|
|
|
.filter(s -> attrs.keySet().contains(s.getAttrKey()) && s.getThingCode().equals(thingCode)) |
|
|
|
.toList(); |
|
|
|
queueMsgDTOList.addAll(collect); |
|
|
|
} |
|
|
|
} |
|
|
|
}); |
|
|
|
|
|
|
|
if (queueMsgDTOList.isEmpty()) { |
|
|
|
result.remove("values"); |
|
|
|
jsonObject.put("result",result); |
|
|
|
return null; // 如果没有符合条件的 QueueMsgDTO，返回空 JSON 对象 |
|
|
|
} |
|
|
|
// 使用 TypeToken 获取 List<QueueMsgDTO> 的类型 |
|
|
|
// Type listType = new TypeToken<List<QueueMsgDTO>>() {}.getType(); |
|
|
|
// 将 List<QueueMsgDTO> 转换为 JsonElement |
|
|
|
JsonElement jsonElement1 = gson.toJsonTree(queueMsgDTOList, listType); |
|
|
|
result.add("values",jsonElement1); |
|
|
|
return dataJson; |
|
|
|
result.put("values", queueMsgDTOList); // 将 values 放入 result 中 |
|
|
|
result.put("info", info); // 将 values 放入 result 中 |
|
|
|
jsonObject.put("result",result); |
|
|
|
return jsonObject; // 返回修改后的 dataJson |
|
|
|
} |
|
|
|
} |
lishuai
1 year ago