如何将FlowFile的属性放入其JSON内容?

时间:2017-09-14 11:18:27

标签: python json apache-nifi

我使用ExecuteScript处理器和Python语言编写脚本。

我想将FlowFile的两个属性(eventidreason)作为参数:值对传递到其JSON内容中。 eventid的值是字符串,而reason的值是整数。我尝试使用flowFile.getAttribute('eventid'),但它失败了。

正确的方法是什么?

def process(self, inputStream, outputStream):
        text = IOUtils.toString(inputStream, StandardCharsets.UTF_8)
        obj = json.loads(text)
        dt = datetime.now().strftime('%Y-%m-%dT%H:%M:%S.%f')

        newObj = {
            "EventId": str(parse(flowFile.getAttribute('eventid'))),
            "EventType": self.getEventType(dt,obj),
            "EventReason": flowFile.getAttribute('reason')
        }
        outputStream.write(bytearray(json.dumps(newObj, indent=4).encode('utf-8')))

flowFile = session.get()
if (flowFile != None):
    flowFile = session.write(flowFile, ModJSON())
session.transfer(flowFile, REL_SUCCESS)
session.commit()

1 个答案:

答案 0 :(得分:2)

您可以将EvaluateJsonPath与Destination设置为流文件属性,将Return Type设置为JSON。然后,您可以为每个JSON路径添加属性以提取,如:

eventid = $.eventid