summaryrefslogtreecommitdiffstats
path: root/config-model/src/main/java/com/yahoo/searchdefinition/parser/ParsedDocument.java
blob: ed975238067853f923e7322baadfe87c6f21a7fe (plain) (blame)
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
// 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.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

/**
 * This class holds the extracted information after parsing a
 * "document" block in a schema (.sd) file, using simple data
 * structures as far as possible.  Do not put advanced logic here!
 * @author arnej27959
 **/
public class ParsedDocument extends ParsedBlock {
    private final List<String> inherited = new ArrayList<>();
    private final Map<String, ParsedDocument> resolvedInherits = new LinkedHashMap();
    private final Map<String, ParsedField> docFields = new LinkedHashMap<>();
    private final Map<String, ParsedStruct> docStructs = new LinkedHashMap<>();
    private final Map<String, ParsedAnnotation> docAnnotations = new LinkedHashMap<>();

    public ParsedDocument(String name) {
        super(name, "document");
    }

    List<String> getInherited() { return List.copyOf(inherited); }
    List<ParsedAnnotation> getAnnotations() { return List.copyOf(docAnnotations.values()); }
    List<ParsedDocument> getResolvedInherits() { return List.copyOf(resolvedInherits.values()); }
    List<ParsedField> getFields() { return List.copyOf(docFields.values()); }
    List<ParsedStruct> getStructs() { return List.copyOf(docStructs.values()); }
    ParsedStruct getStruct(String name) { return docStructs.get(name); }
    ParsedAnnotation getAnnotation(String name) { return docAnnotations.get(name); }

    List<String> getReferencedDocuments() {
        var result = new ArrayList<String>();
        for (var field : docFields.values()) {
            var type = field.getType();
            if (type.getVariant() == ParsedType.Variant.DOC_REFERENCE) {
                var docType = type.getReferencedDocumentType();
                assert(docType.getVariant() == ParsedType.Variant.DOCUMENT);
                result.add(docType.name());
            }
        }
        return result;
    }

    void inherit(String other) { inherited.add(other); }

    void addField(ParsedField field) {
        String fieldName = field.name();
        verifyThat(! docFields.containsKey(fieldName), "already has field", fieldName);
        docFields.put(fieldName, field);
    }

    void addStruct(ParsedStruct struct) {
        String sName = struct.name();
        verifyThat(! docStructs.containsKey(sName), "already has struct", sName);
        docStructs.put(sName, struct);
        struct.tagOwner(name());
    }

    void addAnnotation(ParsedAnnotation annotation) {
        String annName = annotation.name();
        verifyThat(! docAnnotations.containsKey(annName), "already has annotation", annName);
        docAnnotations.put(annName, annotation);
        annotation.tagOwner(name());
        annotation.getStruct().ifPresent(s -> s.tagOwner(name()));
    }

    public String toString() { return "document " + name(); }

    void resolveInherit(String name, ParsedDocument parsed) {
        verifyThat(inherited.contains(name), "resolveInherit for non-inherited name", name);
        verifyThat(name.equals(parsed.name()), "resolveInherit name mismatch for", name);
        verifyThat(! resolvedInherits.containsKey(name), "double resolveInherit for", name);
        resolvedInherits.put(name, parsed);
    }

    void resolveReferenced(ParsedDocument parsed) {
        // TODO - not really inheritance:
        var old = resolvedInherits.put(parsed.name(), parsed);
        assert(old == null || old == parsed);
    }

}