1 files changed, 44 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/document/TemporaryImportedFields.java b/config-model/src/main/java/com/yahoo/schema/document/TemporaryImportedFields.java
new file mode 100644
index 00000000000..7ad4feb6d32
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/schema/document/TemporaryImportedFields.java
@@ -0,0 +1,44 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.schema.document;
+
+import com.yahoo.schema.Schema;
+
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+/**
+ * A set of fields that are imported from referenced document types.
+ *
+ * This is temporary AST structure that only refers to the imported fields by name.
+ *
+ * @author geirst
+ */
+public class TemporaryImportedFields {
+
+    private final Schema owner;
+    private final Map<String, TemporaryImportedField> fields = new LinkedHashMap<>();
+
+    public TemporaryImportedFields(Schema owner) {
+        this.owner = owner;
+    }
+
+    public void add(TemporaryImportedField importedField) {
+        fields.put(importedField.fieldName(), importedField);
+    }
+
+    public boolean hasField(String fieldName) {
+        return fields.get(fieldName) != null;
+    }
+
+    public Map<String, TemporaryImportedField> fields() {
+        if (owner.inherited().isEmpty()) return Collections.unmodifiableMap(fields);
+        if (owner.inherited().get().temporaryImportedFields().isEmpty()) return Collections.unmodifiableMap(fields);
+
+        var allFields = new HashMap<>(owner.inherited().get().temporaryImportedFields().get().fields());
+        allFields.putAll(fields);
+        return allFields;
+    }
+
+}