1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition.parser;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* Class resolving some inheritance relationships.
*
* @author arnej27959
**/
public class InheritanceResolver {
private final Map<String, ParsedSchema> parsedSchemas;
private final Map<String, ParsedDocument> parsedDocs = new HashMap<>();
public InheritanceResolver(Map<String, ParsedSchema> parsedSchemas) {
this.parsedSchemas = parsedSchemas;
}
private void inheritanceCycleCheck(ParsedSchema schema, List<String> seen) {
String name = schema.name();
if (seen.contains(name)) {
seen.add(name);
throw new IllegalArgumentException("Inheritance cycle for schemas: " +
String.join(" -> ", seen));
}
seen.add(name);
for (ParsedSchema parent : schema.getResolvedInherits()) {
inheritanceCycleCheck(parent, seen);
}
seen.remove(name);
}
private void resolveSchemaInheritance() {
for (ParsedSchema schema : parsedSchemas.values()) {
for (String inherit : schema.getInherited()) {
var parent = parsedSchemas.get(inherit);
if (parent == null) {
throw new IllegalArgumentException("schema " + schema.name() + " inherits from unavailable schema " + inherit);
}
schema.resolveInherit(inherit, parent);
}
}
}
private void checkSchemaCycles() {
List<String> seen = new ArrayList<>();
for (ParsedSchema schema : parsedSchemas.values()) {
inheritanceCycleCheck(schema, seen);
}
}
private void resolveDocumentInheritance() {
for (ParsedSchema schema : parsedSchemas.values()) {
if (! schema.hasDocument()) {
// TODO: is schema without a document even valid?
continue;
}
ParsedDocument doc = schema.getDocument();
var old = parsedDocs.put(doc.name(), doc);
assert(old == null);
}
for (ParsedDocument doc : parsedDocs.values()) {
for (String inherit : doc.getInherited()) {
var parentDoc = parsedDocs.get(inherit);
if (parentDoc == null) {
throw new IllegalArgumentException("document " + doc.name() + " inherits from unavailable document " + inherit);
}
doc.resolveInherit(inherit, parentDoc);
}
}
}
private void inheritanceCycleCheck(ParsedDocument document, List<String> seen) {
String name = document.name();
if (seen.contains(name)) {
seen.add(name);
throw new IllegalArgumentException("Inheritance cycle for documents: " +
String.join(" -> ", seen));
}
seen.add(name);
for (ParsedDocument parent : document.getResolvedInherits()) {
inheritanceCycleCheck(parent, seen);
}
seen.remove(name);
}
private void checkDocumentCycles() {
List<String> seen = new ArrayList<>();
for (ParsedDocument doc : parsedDocs.values()) {
inheritanceCycleCheck(doc, seen);
}
}
public void resolveInheritance() {
resolveSchemaInheritance();
resolveDocumentInheritance();
checkSchemaCycles();
checkDocumentCycles();
}
}
|
