Examples with HL7Query org.apache.nifi.hl7.query.HL7Query used on opensource projects

Example 1 with HL7Query

use of org.apache.nifi.hl7.query.HL7Query in project nifi by apache.

the class HL7Query method compile.

public static HL7Query compile(final String query) {
    try {
        final CommonTokenStream lexerTokenStream = createTokenStream(query);
        final HL7QueryParser parser = new HL7QueryParser(lexerTokenStream);
        final Tree tree = (Tree) parser.query().getTree();
        return new HL7Query(tree, query);
    } catch (final HL7QueryParsingException e) {
        throw e;
    } catch (final Exception e) {
        throw new HL7QueryParsingException(e);
    }
}

Example 2 with HL7Query

use of org.apache.nifi.hl7.query.HL7Query in project nifi by apache.

the class TestHL7Query method testSelectField.

@Test
@SuppressWarnings({ "unchecked", "rawtypes" })
public void testSelectField() throws HL7Exception, IOException {
    final HL7Query query = HL7Query.compile("SELECT PID.5");
    final HL7Message msg = hypoglycemia;
    final QueryResult result = query.evaluate(msg);
    assertTrue(result.isMatch());
    final List<String> labels = result.getLabels();
    assertEquals(1, labels.size());
    assertEquals(1, result.getHitCount());
    final Object names = result.nextHit().getValue("PID.5");
    assertTrue(names instanceof List);
    final List<Object> nameList = (List) names;
    assertEquals(1, nameList.size());
    final HL7Field nameField = (HL7Field) nameList.get(0);
    assertEquals("SMITH^JOHN", nameField.getValue());
}

Example 3 with HL7Query

use of org.apache.nifi.hl7.query.HL7Query in project nifi by apache.

the class TestHL7Query method testSelectMessage.

@Test
public void testSelectMessage() throws HL7Exception, IOException {
    final HL7Query query = HL7Query.compile("SELECT MESSAGE");
    final HL7Message msg = hypoglycemia;
    final QueryResult result = query.evaluate(msg);
    assertTrue(result.isMatch());
    final List<String> labels = result.getLabels();
    assertEquals(1, labels.size());
    assertEquals("MESSAGE", labels.get(0));
    assertEquals(1, result.getHitCount());
    assertEquals(msg, result.nextHit().getValue("MESSAGE"));
}

Example 4 with HL7Query

use of org.apache.nifi.hl7.query.HL7Query in project nifi by apache.

the class RouteHL7 method onTrigger.

@Override
public void onTrigger(final ProcessContext context, final ProcessSession session) throws ProcessException {
    FlowFile flowFile = session.get();
    if (flowFile == null) {
        return;
    }
    final Charset charset = Charset.forName(context.getProperty(CHARACTER_SET).evaluateAttributeExpressions(flowFile).getValue());
    final byte[] buffer = new byte[(int) flowFile.getSize()];
    session.read(flowFile, new InputStreamCallback() {

        @Override
        public void process(final InputStream in) throws IOException {
            StreamUtils.fillBuffer(in, buffer);
        }
    });
    @SuppressWarnings("resource") final HapiContext hapiContext = new DefaultHapiContext();
    hapiContext.setValidationContext((ca.uhn.hl7v2.validation.ValidationContext) ValidationContextFactory.noValidation());
    final PipeParser parser = hapiContext.getPipeParser();
    final String hl7Text = new String(buffer, charset);
    final HL7Message message;
    try {
        final Message hapiMessage = parser.parse(hl7Text);
        message = new HapiMessage(hapiMessage);
    } catch (final Exception e) {
        getLogger().error("Failed to parse {} as HL7 due to {}; routing to failure", new Object[] { flowFile, e });
        session.transfer(flowFile, REL_FAILURE);
        return;
    }
    final Set<String> matchingRels = new HashSet<>();
    final Map<Relationship, HL7Query> queryMap = queries;
    for (final Map.Entry<Relationship, HL7Query> entry : queryMap.entrySet()) {
        final Relationship relationship = entry.getKey();
        final HL7Query query = entry.getValue();
        final QueryResult result = query.evaluate(message);
        if (result.isMatch()) {
            FlowFile clone = session.clone(flowFile);
            clone = session.putAttribute(clone, "RouteHL7.Route", relationship.getName());
            session.transfer(clone, relationship);
            session.getProvenanceReporter().route(clone, relationship);
            matchingRels.add(relationship.getName());
        }
    }
    session.transfer(flowFile, REL_ORIGINAL);
    getLogger().info("Routed a copy of {} to {} relationships: {}", new Object[] { flowFile, matchingRels.size(), matchingRels });
}

Example 5 with HL7Query

use of org.apache.nifi.hl7.query.HL7Query in project nifi by apache.

the class RouteHL7 method onPropertyModified.

@Override
public void onPropertyModified(final PropertyDescriptor descriptor, final String oldValue, final String newValue) {
    if (!descriptor.isDynamic()) {
        return;
    }
    final Map<Relationship, HL7Query> updatedQueryMap = new HashMap<>(queries);
    final Relationship relationship = new Relationship.Builder().name(descriptor.getName()).build();
    if (newValue == null) {
        updatedQueryMap.remove(relationship);
    } else {
        final HL7Query query = HL7Query.compile(newValue);
        updatedQueryMap.put(relationship, query);
    }
    this.queries = updatedQueryMap;
}