001/*
002 The contents of this file are subject to the Mozilla Public License Version 1.1
003 (the "License"); you may not use this file except in compliance with the License.
004 You may obtain a copy of the License at http://www.mozilla.org/MPL/
005 Software distributed under the License is distributed on an "AS IS" basis,
006 WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the
007 specific language governing rights and limitations under the License.
008
009 The Original Code is "VisitorMessageRule.java".  Description:
010 "A message rule that runs a ValidatingMessageVisitor over the message"
011
012 The Initial Developer of the Original Code is University Health Network. Copyright (C)
013 2013.  All Rights Reserved.
014
015 Contributor(s): ______________________________________.
016
017 Alternatively, the contents of this file may be used under the terms of the
018 GNU General Public License (the "GPL"), in which case the provisions of the GPL are
019 applicable instead of those above.  If you wish to allow use of your version of this
020 file only under the terms of the GPL and not to allow others to use your version
021 of this file under the MPL, indicate your decision by deleting  the provisions above
022 and replace  them with the notice and other provisions required by the GPL License.
023 If you do not delete the provisions above, a recipient may use your version of
024 this file under either the MPL or the GPL.
025 */
026
027package ca.uhn.hl7v2.validation.builder.support;
028
029import ca.uhn.hl7v2.HL7Exception;
030import ca.uhn.hl7v2.model.Message;
031import ca.uhn.hl7v2.model.MessageVisitorFactory;
032import ca.uhn.hl7v2.model.MessageVisitors;
033import ca.uhn.hl7v2.validation.ValidationException;
034import ca.uhn.hl7v2.validation.impl.AbstractMessageRule;
035
036/**
037 * A message rule that runs a {@link ValidatingMessageVisitor} over the message,
038 * collecting exceptions while traversing
039 */
040public class VisitorMessageRule extends AbstractMessageRule {
041
042    private final MessageVisitorFactory<? extends ValidatingMessageVisitor> visitorFactory;
043
044    public VisitorMessageRule(MessageVisitorFactory<? extends ValidatingMessageVisitor> visitorFactory) {
045        if (visitorFactory == null) throw new IllegalArgumentException("Visitor factory must not be null");
046        this.visitorFactory = visitorFactory;
047    }
048
049    public ValidationException[] apply(Message message) {
050        try {
051            ValidatingMessageVisitor visitor = visitorFactory.create();
052            MessageVisitors.visit(message, visitor);
053            return visitor.getValidationExceptions();
054        } catch (HL7Exception e) {
055            return new ValidationException[] {
056                new ValidationException(e)
057            };
058        }
059    }
060}