Commit bd42cfc81f4c8641ad7b41f6e24e95f6d2da80f0
Committed by
Andrew Shvayka
1 parent
52d1b756
Changes according to comment
Showing
1 changed file
with
20 additions
and
18 deletions
... | ... | @@ -423,6 +423,25 @@ public class RuleChainController extends BaseController { |
423 | 423 | } |
424 | 424 | |
425 | 425 | private String msgToOutput(TbMsg msg) throws Exception { |
426 | + JsonNode resultNode = convertMsgToOut(msg); | |
427 | + return objectMapper.writeValueAsString(resultNode); | |
428 | + } | |
429 | + | |
430 | + private String msgToOutput(List<TbMsg> msgs) throws Exception { | |
431 | + JsonNode resultNode; | |
432 | + if (msgs.size() > 1) { | |
433 | + resultNode = objectMapper.createArrayNode(); | |
434 | + for (TbMsg msg : msgs) { | |
435 | + JsonNode convertedData = convertMsgToOut(msg); | |
436 | + ((ArrayNode) resultNode).add(convertedData); | |
437 | + } | |
438 | + } else { | |
439 | + resultNode = convertMsgToOut(msgs.get(0)); | |
440 | + } | |
441 | + return objectMapper.writeValueAsString(resultNode); | |
442 | + } | |
443 | + | |
444 | + private JsonNode convertMsgToOut(TbMsg msg) throws Exception{ | |
426 | 445 | ObjectNode msgData = objectMapper.createObjectNode(); |
427 | 446 | if (!StringUtils.isEmpty(msg.getData())) { |
428 | 447 | msgData.set("msg", objectMapper.readTree(msg.getData())); |
... | ... | @@ -430,25 +449,8 @@ public class RuleChainController extends BaseController { |
430 | 449 | Map<String, String> metadata = msg.getMetaData().getData(); |
431 | 450 | msgData.set("metadata", objectMapper.valueToTree(metadata)); |
432 | 451 | msgData.put("msgType", msg.getType()); |
433 | - return objectMapper.writeValueAsString(msgData); | |
452 | + return msgData; | |
434 | 453 | } |
435 | 454 | |
436 | - private String msgToOutput(List<TbMsg> msgs) throws Exception { | |
437 | - ArrayNode resultNode = objectMapper.createArrayNode(); | |
438 | - for (TbMsg msg:msgs) { | |
439 | - ObjectNode msgData = objectMapper.createObjectNode(); | |
440 | - if (!StringUtils.isEmpty(msg.getData())) { | |
441 | - msgData.set("msg", objectMapper.readTree(msg.getData())); | |
442 | - } | |
443 | - Map<String, String> metadata = msg.getMetaData().getData(); | |
444 | - msgData.set("metadata", objectMapper.valueToTree(metadata)); | |
445 | - msgData.put("msgType", msg.getType()); | |
446 | - resultNode.add(msgData); | |
447 | - } | |
448 | - if (resultNode.size() == 1) { | |
449 | - return objectMapper.writeValueAsString(resultNode.get(0)); | |
450 | - } | |
451 | - return objectMapper.writeValueAsString(resultNode); | |
452 | - } | |
453 | 455 | |
454 | 456 | } | ... | ... |