Examples with ARecord org.apache.asterix.om.base.ARecord used on opensource projects

Example 21 with ARecord

use of org.apache.asterix.om.base.ARecord in project asterixdb by apache.

the class ARecordSerializerDeserializerTest method test.

@Test
public void test() {
    // Generates types.
    ARecordType addrRecordType = SerializerDeserializerTestUtils.generateAddressRecordType();
    ARecordType employeeType = SerializerDeserializerTestUtils.generateEmployeeRecordType(addrRecordType);
    //Generates records.
    ARecord[] records = SerializerDeserializerTestUtils.generateRecords(addrRecordType, employeeType);
    ARecordSerializerDeserializer serde = new ARecordSerializerDeserializer(employeeType);
    // Run four test threads to serialize/deserialize records concurrently.
    SerializerDeserializerTestUtils.concurrentSerDeTestRun(serde, records);
}

Example 22 with ARecord

use of org.apache.asterix.om.base.ARecord in project asterixdb by apache.

the class SerializerDeserializerTestUtils method generateRecords.

public static ARecord[] generateRecords(ARecordType addrRecordType, ARecordType employeeType) {
    AOrderedListType addrListType = new AOrderedListType(addrRecordType, "address_history");
    ARecord addr11 = new ARecord(addrRecordType, new IAObject[] { new AString("120 San Raman Street"), new AString("Irvine"), new AString("CA"), new AInt16((short) 95051), new AInterval(0, 100, (byte) 0) });
    ARecord addr12 = new ARecord(addrRecordType, new IAObject[] { new AString("210 University Drive"), new AString("Philadelphia"), new AString("PA"), new AInt16((short) 10086), new AInterval(100, 300, (byte) 0) });
    ARecord addr21 = new ARecord(addrRecordType, new IAObject[] { new AString("1 College Street"), new AString("Seattle"), new AString("WA"), new AInt16((short) 20012), new AInterval(400, 500, (byte) 0) });
    ARecord addr22 = new ARecord(addrRecordType, new IAObject[] { new AString("20 Lindsay Avenue"), new AString("Columbus"), new AString("OH"), new AInt16((short) 30120), new AInterval(600, 900, (byte) 0) });
    ARecord addr31 = new ARecord(addrRecordType, new IAObject[] { new AString("200 14th Avenue"), new AString("Long Island"), new AString("NY"), new AInt16((short) 95011), new AInterval(12000, 14000, (byte) 0) });
    // With nested open field addr31.
    ARecord addr32 = new ARecord(addrRecordType, new IAObject[] { new AString("51 8th Street"), new AString("Orlando"), new AString("FL"), new AInt16((short) 49045), new AInterval(190000, 200000, (byte) 0) });
    ARecord record1 = new ARecord(employeeType, new IAObject[] { new AInt64(0L), new AString("Tom"), new AOrderedList(addrListType, Arrays.asList(new IAObject[] { addr11, addr12 })) });
    ARecord record2 = new ARecord(employeeType, new IAObject[] { new AInt64(1L), new AString("John"), new AOrderedList(addrListType, Arrays.asList(new IAObject[] { addr21, addr22 })) });
    ARecord record3 = new ARecord(employeeType, new IAObject[] { new AInt64(2L), new AString("Lindsay"), new AOrderedList(addrListType, Arrays.asList(new IAObject[] { addr31, addr32 })) });
    // With nested open field addr41.
    ARecord record4 = new ARecord(employeeType, new IAObject[] { new AInt64(3L), new AString("Joshua"), new AOrderedList(addrListType, Arrays.asList(new IAObject[] {})) });
    ARecord[] records = new ARecord[] { record1, record2, record3, record4 };
    return records;
}

Example 23 with ARecord

use of org.apache.asterix.om.base.ARecord in project asterixdb by apache.

the class ARecordSerializerDeserializer method deserialize.

@Override
public ARecord deserialize(DataInput in) throws HyracksDataException {
    try {
        boolean isExpanded = isExpandedRecord(in);
        IAObject[] schemaFields = getValuesForSchemaFields(in);
        if (isExpanded) {
            int numberOfOpenFields = in.readInt();
            String[] fieldNames = new String[numberOfOpenFields];
            IAType[] fieldTypes = new IAType[numberOfOpenFields];
            IAObject[] openFields = new IAObject[numberOfOpenFields];
            for (int i = 0; i < numberOfOpenFields; i++) {
                in.readInt();
                in.readInt();
            }
            for (int i = 0; i < numberOfOpenFields; i++) {
                fieldNames[i] = AStringSerializerDeserializer.INSTANCE.deserialize(in).getStringValue();
                openFields[i] = AObjectSerializerDeserializer.INSTANCE.deserialize(in);
                fieldTypes[i] = openFields[i].getType();
            }
            ARecordType openPartRecType = new ARecordType(null, fieldNames, fieldTypes, true);
            if (numberOfSchemaFields > 0) {
                ARecordType mergedRecordType = mergeRecordTypes(this.recordType, openPartRecType);
                IAObject[] mergedFields = mergeFields(schemaFields, openFields);
                return new ARecord(mergedRecordType, mergedFields);
            } else {
                return new ARecord(openPartRecType, openFields);
            }
        } else {
            return new ARecord(this.recordType, schemaFields);
        }
    } catch (IOException e) {
        throw new HyracksDataException(e);
    }
}

Example 24 with ARecord

use of org.apache.asterix.om.base.ARecord in project asterixdb by apache.

the class NodeGroupTupleTranslator method getMetadataEntityFromTuple.

@Override
public NodeGroup getMetadataEntityFromTuple(ITupleReference frameTuple) throws HyracksDataException {
    byte[] serRecord = frameTuple.getFieldData(NODEGROUP_PAYLOAD_TUPLE_FIELD_INDEX);
    int recordStartOffset = frameTuple.getFieldStart(NODEGROUP_PAYLOAD_TUPLE_FIELD_INDEX);
    int recordLength = frameTuple.getFieldLength(NODEGROUP_PAYLOAD_TUPLE_FIELD_INDEX);
    ByteArrayInputStream stream = new ByteArrayInputStream(serRecord, recordStartOffset, recordLength);
    DataInput in = new DataInputStream(stream);
    ARecord nodeGroupRecord = recordSerDes.deserialize(in);
    String gpName = ((AString) nodeGroupRecord.getValueByPos(MetadataRecordTypes.NODEGROUP_ARECORD_GROUPNAME_FIELD_INDEX)).getStringValue();
    IACursor cursor = ((AUnorderedList) nodeGroupRecord.getValueByPos(MetadataRecordTypes.NODEGROUP_ARECORD_NODENAMES_FIELD_INDEX)).getCursor();
    List<String> nodeNames = new ArrayList<>();
    while (cursor.next()) {
        nodeNames.add(((AString) cursor.get()).getStringValue());
    }
    return new NodeGroup(gpName, nodeNames);
}