View Javadoc
1   /*
2    * Portions of this software was developed by employees of the National Institute
3    * of Standards and Technology (NIST), an agency of the Federal Government and is
4    * being made available as a public service. Pursuant to title 17 United States
5    * Code Section 105, works of NIST employees are not subject to copyright
6    * protection in the United States. This software may be subject to foreign
7    * copyright. Permission in the United States and in foreign countries, to the
8    * extent that NIST may hold copyright, to use, copy, modify, create derivative
9    * works, and distribute this software and its documentation without fee is hereby
10   * granted on a non-exclusive basis, provided that this notice and disclaimer
11   * of warranty appears in all copies.
12   *
13   * THE SOFTWARE IS PROVIDED 'AS IS' WITHOUT ANY WARRANTY OF ANY KIND, EITHER
14   * EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY
15   * THAT THE SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF
16   * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND FREEDOM FROM
17   * INFRINGEMENT, AND ANY WARRANTY THAT THE DOCUMENTATION WILL CONFORM TO THE
18   * SOFTWARE, OR ANY WARRANTY THAT THE SOFTWARE WILL BE ERROR FREE.  IN NO EVENT
19   * SHALL NIST BE LIABLE FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO, DIRECT,
20   * INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES, ARISING OUT OF, RESULTING FROM,
21   * OR IN ANY WAY CONNECTED WITH THIS SOFTWARE, WHETHER OR NOT BASED UPON WARRANTY,
22   * CONTRACT, TORT, OR OTHERWISE, WHETHER OR NOT INJURY WAS SUSTAINED BY PERSONS OR
23   * PROPERTY OR OTHERWISE, AND WHETHER OR NOT LOSS WAS SUSTAINED FROM, OR AROSE OUT
24   * OF THE RESULTS OF, OR USE OF, THE SOFTWARE OR SERVICES PROVIDED HEREUNDER.
25   */
26  
27  package gov.nist.secauto.metaschema.databind.model;
28  
29  import gov.nist.secauto.metaschema.core.metapath.MetapathException;
30  import gov.nist.secauto.metaschema.core.model.constraint.ICardinalityConstraint;
31  import gov.nist.secauto.metaschema.core.model.constraint.IConstraint.ISource;
32  import gov.nist.secauto.metaschema.core.model.constraint.IIndexConstraint;
33  import gov.nist.secauto.metaschema.core.model.constraint.IModelConstrained;
34  import gov.nist.secauto.metaschema.core.model.constraint.IUniqueConstraint;
35  import gov.nist.secauto.metaschema.databind.model.annotations.AssemblyConstraints;
36  import gov.nist.secauto.metaschema.databind.model.annotations.ValueConstraints;
37  
38  import java.util.Arrays;
39  import java.util.LinkedList;
40  import java.util.List;
41  import java.util.stream.Collectors;
42  
43  import edu.umd.cs.findbugs.annotations.NonNull;
44  import edu.umd.cs.findbugs.annotations.Nullable;
45  
46  /**
47   * Support for constraints on Module assembly bound objects.
48   */
49  class AssemblyConstraintSupport
50      extends ValueConstraintSupport
51      implements IModelConstrained {
52    @NonNull
53    private final List<IIndexConstraint> indexConstraints;
54    @NonNull
55    private final List<IUniqueConstraint> uniqueConstraints;
56    @NonNull
57    private final List<ICardinalityConstraint> cardinalityConstraints;
58  
59    public AssemblyConstraintSupport() {
60      this.indexConstraints = new LinkedList<>();
61      this.uniqueConstraints = new LinkedList<>();
62      this.cardinalityConstraints = new LinkedList<>();
63    }
64  
65    @SuppressWarnings("null")
66    public AssemblyConstraintSupport(
67        @Nullable ValueConstraints valueAnnotation,
68        @Nullable AssemblyConstraints assemblyAnnotation,
69        @NonNull ISource source) {
70      super(valueAnnotation, source);
71      if (assemblyAnnotation == null) {
72        this.indexConstraints = new LinkedList<>();
73        this.uniqueConstraints = new LinkedList<>();
74        this.cardinalityConstraints = new LinkedList<>();
75      } else {
76        try {
77          indexConstraints = Arrays.stream(assemblyAnnotation.index())
78              .map(annotation -> ConstraintFactory.newIndexConstraint(annotation, source))
79              .collect(Collectors.toCollection(LinkedList::new));
80  
81          uniqueConstraints = Arrays.stream(assemblyAnnotation.isUnique())
82              .map(annotation -> ConstraintFactory.newUniqueConstraint(annotation, source))
83              .collect(Collectors.toCollection(LinkedList::new));
84  
85          cardinalityConstraints = Arrays.stream(assemblyAnnotation.hasCardinality())
86              .map(annotation -> ConstraintFactory.newCardinalityConstraint(annotation, source))
87              .collect(Collectors.toCollection(LinkedList::new));
88        } catch (MetapathException ex) {
89          throw new MetapathException(
90              String.format("Unable to compile a Metapath in '%s'. %s", source.getSource(), ex.getLocalizedMessage()),
91              ex);
92        }
93      }
94      getConstraints().addAll(indexConstraints);
95      getConstraints().addAll(uniqueConstraints);
96      getConstraints().addAll(cardinalityConstraints);
97    }
98  
99    @Override
100   public List<IIndexConstraint> getIndexConstraints() {
101     return indexConstraints;
102   }
103 
104   @Override
105   public List<IUniqueConstraint> getUniqueConstraints() {
106     return uniqueConstraints;
107   }
108 
109   @Override
110   public List<ICardinalityConstraint> getHasCardinalityConstraints() {
111     return cardinalityConstraints;
112   }
113 
114   @Override
115   public void addConstraint(@NonNull IIndexConstraint constraint) {
116     getConstraints().add(constraint);
117     indexConstraints.add(constraint);
118   }
119 
120   @Override
121   public void addConstraint(@NonNull IUniqueConstraint constraint) {
122     getConstraints().add(constraint);
123     uniqueConstraints.add(constraint);
124   }
125 
126   @Override
127   public void addConstraint(@NonNull ICardinalityConstraint constraint) {
128     getConstraints().add(constraint);
129     cardinalityConstraints.add(constraint);
130   }
131 }