Examples with StructDataType com.yahoo.document.StructDataType used on opensource projects

Example 1 with StructDataType

use of com.yahoo.document.StructDataType in project vespa by vespa-engine.

the class DocumentModelBuilder method convert.

private NewDocumentType convert(SDDocumentType sdoc) {
    Map<AnnotationType, String> annotationInheritance = new HashMap<>();
    Map<StructDataType, String> structInheritance = new HashMap<>();
    NewDocumentType dt = new NewDocumentType(new NewDocumentType.Name(sdoc.getName()), sdoc.getDocumentType().getHeaderType(), sdoc.getDocumentType().getBodyType(), sdoc.getFieldSets(), convertDocumentReferencesToNames(sdoc.getDocumentReferences()));
    for (SDDocumentType n : sdoc.getInheritedTypes()) {
        NewDocumentType.Name name = new NewDocumentType.Name(n.getName());
        NewDocumentType inherited = model.getDocumentManager().getDocumentType(name);
        if (inherited != null) {
            dt.inherit(inherited);
        }
    }
    for (SDDocumentType type : sdoc.getTypes()) {
        if (type.isStruct()) {
            handleStruct(dt, type);
        } else {
            throw new IllegalArgumentException("Data type '" + sdoc.getName() + "' is not a struct => tostring='" + sdoc.toString() + "'.");
        }
    }
    for (AnnotationType annotation : sdoc.getAnnotations()) {
        dt.add(annotation);
    }
    for (AnnotationType annotation : sdoc.getAnnotations()) {
        SDAnnotationType sa = (SDAnnotationType) annotation;
        if (annotation.getInheritedTypes().isEmpty() && (sa.getInherits() != null)) {
            annotationInheritance.put(annotation, sa.getInherits());
        }
        if (annotation.getDataType() == null) {
            if (sa.getSdDocType() != null) {
                StructDataType s = handleStruct(dt, sa.getSdDocType());
                annotation.setDataType(s);
                if ((sa.getInherits() != null)) {
                    structInheritance.put(s, "annotation." + sa.getInherits());
                }
            } else if (sa.getInherits() != null) {
                StructDataType s = new StructDataType("annotation." + annotation.getName());
                if (anyParentsHavePayLoad(sa, sdoc)) {
                    annotation.setDataType(s);
                    addType(dt, s);
                }
                structInheritance.put(s, "annotation." + sa.getInherits());
            }
        }
    }
    for (Map.Entry<AnnotationType, String> e : annotationInheritance.entrySet()) {
        e.getKey().inherit(dt.getAnnotationType(e.getValue()));
    }
    for (Map.Entry<StructDataType, String> e : structInheritance.entrySet()) {
        StructDataType s = (StructDataType) dt.getDataType(e.getValue());
        if (s != null) {
            e.getKey().inherit(s);
        }
    }
    handleStruct(dt, sdoc.getDocumentType().getHeaderType());
    handleStruct(dt, sdoc.getDocumentType().getBodyType());
    extractDataTypesFromFields(dt, sdoc.fieldSet());
    return dt;
}

Example 2 with StructDataType

use of com.yahoo.document.StructDataType in project vespa by vespa-engine.

the class IndexSchemaTestCase method requireThatArrayOfStructIsFlattened.

@Test
public void requireThatArrayOfStructIsFlattened() {
    StructDataType type = new StructDataType("my_struct");
    type.addField(new Field("my_byte", DataType.BYTE));
    type.addField(new Field("my_double", DataType.DOUBLE));
    type.addField(new Field("my_float", DataType.FLOAT));
    type.addField(new Field("my_int", DataType.INT));
    type.addField(new Field("my_long", DataType.LONG));
    type.addField(new Field("my_raw", DataType.RAW));
    type.addField(new Field("my_string", DataType.STRING));
    type.addField(new Field("my_uri", DataType.URI));
    assertFlat(new Field("foo", DataType.getArray(type)), new Field("foo.my_byte", DataType.getArray(DataType.BYTE)), new Field("foo.my_double", DataType.getArray(DataType.DOUBLE)), new Field("foo.my_float", DataType.getArray(DataType.FLOAT)), new Field("foo.my_int", DataType.getArray(DataType.INT)), new Field("foo.my_long", DataType.getArray(DataType.LONG)), new Field("foo.my_raw", DataType.getArray(DataType.RAW)), new Field("foo.my_string", DataType.getArray(DataType.STRING)), new Field("foo.my_uri", DataType.getArray(DataType.URI)));
}

Example 3 with StructDataType

use of com.yahoo.document.StructDataType in project vespa by vespa-engine.

the class IndexSchemaTestCase method requireThatArrayOfStructWithArrayFieldIsFlattened.

@Test
public void requireThatArrayOfStructWithArrayFieldIsFlattened() {
    StructDataType type = new StructDataType("my_struct");
    type.addField(new Field("my_byte", DataType.getArray(DataType.BYTE)));
    type.addField(new Field("my_double", DataType.getArray(DataType.DOUBLE)));
    type.addField(new Field("my_float", DataType.getArray(DataType.FLOAT)));
    type.addField(new Field("my_int", DataType.getArray(DataType.INT)));
    type.addField(new Field("my_long", DataType.getArray(DataType.LONG)));
    type.addField(new Field("my_raw", DataType.getArray(DataType.RAW)));
    type.addField(new Field("my_string", DataType.getArray(DataType.STRING)));
    type.addField(new Field("my_uri", DataType.getArray(DataType.URI)));
    assertFlat(new Field("foo", DataType.getArray(type)), new Field("foo.my_byte", DataType.getArray(DataType.getArray(DataType.BYTE))), new Field("foo.my_double", DataType.getArray(DataType.getArray(DataType.DOUBLE))), new Field("foo.my_float", DataType.getArray(DataType.getArray(DataType.FLOAT))), new Field("foo.my_int", DataType.getArray(DataType.getArray(DataType.INT))), new Field("foo.my_long", DataType.getArray(DataType.getArray(DataType.LONG))), new Field("foo.my_raw", DataType.getArray(DataType.getArray(DataType.RAW))), new Field("foo.my_string", DataType.getArray(DataType.getArray(DataType.STRING))), new Field("foo.my_uri", DataType.getArray(DataType.getArray(DataType.URI))));
}

Example 4 with StructDataType

use of com.yahoo.document.StructDataType in project vespa by vespa-engine.

the class IndexSchemaTestCase method requireThatStructIsFlattened.

@Test
public void requireThatStructIsFlattened() {
    StructDataType type = new StructDataType("my_struct");
    type.addField(new Field("my_byte", DataType.BYTE));
    type.addField(new Field("my_double", DataType.DOUBLE));
    type.addField(new Field("my_float", DataType.FLOAT));
    type.addField(new Field("my_int", DataType.INT));
    type.addField(new Field("my_long", DataType.LONG));
    type.addField(new Field("my_raw", DataType.RAW));
    type.addField(new Field("my_string", DataType.STRING));
    type.addField(new Field("my_uri", DataType.URI));
    assertFlat(new Field("foo", type), new Field("foo.my_byte", DataType.BYTE), new Field("foo.my_double", DataType.DOUBLE), new Field("foo.my_float", DataType.FLOAT), new Field("foo.my_int", DataType.INT), new Field("foo.my_long", DataType.LONG), new Field("foo.my_raw", DataType.RAW), new Field("foo.my_string", DataType.STRING), new Field("foo.my_uri", DataType.URI));
}

Example 5 with StructDataType

use of com.yahoo.document.StructDataType in project vespa by vespa-engine.

the class IndexSchemaTestCase method requireThatArrayOfArrayOfStructIsFlattened.

@Test
public void requireThatArrayOfArrayOfStructIsFlattened() {
    StructDataType type = new StructDataType("my_struct");
    type.addField(new Field("my_byte", DataType.BYTE));
    type.addField(new Field("my_double", DataType.DOUBLE));
    type.addField(new Field("my_float", DataType.FLOAT));
    type.addField(new Field("my_int", DataType.INT));
    type.addField(new Field("my_long", DataType.LONG));
    type.addField(new Field("my_raw", DataType.RAW));
    type.addField(new Field("my_string", DataType.STRING));
    type.addField(new Field("my_uri", DataType.URI));
    assertFlat(new Field("foo", DataType.getArray(DataType.getArray(type))), new Field("foo.my_byte", DataType.getArray(DataType.getArray(DataType.BYTE))), new Field("foo.my_double", DataType.getArray(DataType.getArray(DataType.DOUBLE))), new Field("foo.my_float", DataType.getArray(DataType.getArray(DataType.FLOAT))), new Field("foo.my_int", DataType.getArray(DataType.getArray(DataType.INT))), new Field("foo.my_long", DataType.getArray(DataType.getArray(DataType.LONG))), new Field("foo.my_raw", DataType.getArray(DataType.getArray(DataType.RAW))), new Field("foo.my_string", DataType.getArray(DataType.getArray(DataType.STRING))), new Field("foo.my_uri", DataType.getArray(DataType.getArray(DataType.URI))));
}