- // UNKNOWN NODES
- for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
- unknownNodes.add(b.build());
- }
- Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
- instance.addUnknownSchemaNodes(unknownNodes);
+ // AUGMENTATIONS
+ final Set<AugmentationSchema> augments = new HashSet<>();
+ for (AugmentationSchemaBuilder builder : augmentationBuilders) {
+ augments.add(builder.build());
+ }
+ instance.augmentations = ImmutableSet.copyOf(augments);
+
+ // REFINES
+ final Map<SchemaPath, SchemaNode> refineNodes = new HashMap<>();
+ for (SchemaNodeBuilder refineBuilder : refineBuilders) {
+ SchemaNode refineNode = refineBuilder.build();
+ refineNodes.put(refineNode.getPath(), refineNode);
+ }
+ instance.refines = ImmutableMap.copyOf(refineNodes);