Java Code Examples for org.apache.hadoop.hive.serde2.typeinfo.TypeInfoFactory#getPrimitiveTypeInfo()

@Test
public void test_getOrcField_primitive() throws Exception {
    // Expected ORC types
    TypeInfo[] expectedTypes = {
            TypeInfoFactory.getPrimitiveTypeInfo("int"),
            TypeInfoFactory.getPrimitiveTypeInfo("bigint"),
            TypeInfoFactory.getPrimitiveTypeInfo("boolean"),
            TypeInfoFactory.getPrimitiveTypeInfo("float"),
            TypeInfoFactory.getPrimitiveTypeInfo("double"),
            TypeInfoFactory.getPrimitiveTypeInfo("binary"),
            TypeInfoFactory.getPrimitiveTypeInfo("string")
    };

    // Build a fake Avro record with all types
    Schema testSchema = buildPrimitiveAvroSchema();
    List<Schema.Field> fields = testSchema.getFields();
    for (int i = 0; i < fields.size(); i++) {
        assertEquals(expectedTypes[i], NiFiOrcUtils.getOrcField(fields.get(i).schema()));
    }

}
 

public static TypeInfo getPrimitiveOrcTypeFromPrimitiveAvroType(Schema.Type avroType) throws IllegalArgumentException {
    if (avroType == null) {
        throw new IllegalArgumentException("Avro type is null");
    }
    switch (avroType) {
        case INT:
            return TypeInfoFactory.getPrimitiveTypeInfo("int");
        case LONG:
            return TypeInfoFactory.getPrimitiveTypeInfo("bigint");
        case BOOLEAN:
            return TypeInfoFactory.getPrimitiveTypeInfo("boolean");
        case BYTES:
            return TypeInfoFactory.getPrimitiveTypeInfo("binary");
        case DOUBLE:
            return TypeInfoFactory.getPrimitiveTypeInfo("double");
        case FLOAT:
            return TypeInfoFactory.getPrimitiveTypeInfo("float");
        case STRING:
            return TypeInfoFactory.getPrimitiveTypeInfo("string");
        default:
            throw new IllegalArgumentException("Avro type " + avroType.getName() + " is not a primitive type");
    }
}
 

public static TypeInfo getPrimitiveOrcTypeFromPrimitiveAvroType(Schema.Type avroType) throws IllegalArgumentException {
    if (avroType == null) {
        throw new IllegalArgumentException("Avro type is null");
    }
    switch (avroType) {
        case INT:
            return TypeInfoFactory.getPrimitiveTypeInfo("int");
        case LONG:
            return TypeInfoFactory.getPrimitiveTypeInfo("bigint");
        case BOOLEAN:
        case NULL: // ORC has no null type, so just pick the smallest. All values are necessarily null.
            return TypeInfoFactory.getPrimitiveTypeInfo("boolean");
        case BYTES:
            return TypeInfoFactory.getPrimitiveTypeInfo("binary");
        case DOUBLE:
            return TypeInfoFactory.getPrimitiveTypeInfo("double");
        case FLOAT:
            return TypeInfoFactory.getPrimitiveTypeInfo("float");
        case STRING:
            return TypeInfoFactory.getPrimitiveTypeInfo("string");
        default:
            throw new IllegalArgumentException("Avro type " + avroType.getName() + " is not a primitive type");
    }
}
 

public static TypeInfo getPrimitiveOrcTypeFromPrimitiveFieldType(DataType rawDataType) throws IllegalArgumentException {
    if (rawDataType == null) {
        throw new IllegalArgumentException("Avro type is null");
    }
    RecordFieldType fieldType = rawDataType.getFieldType();
    if (RecordFieldType.INT.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("int");
    }
    if (RecordFieldType.LONG.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("bigint");
    }
    if (RecordFieldType.BOOLEAN.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("boolean");
    }
    if (RecordFieldType.DOUBLE.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("double");
    }
    if (RecordFieldType.FLOAT.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("float");
    }
    if (RecordFieldType.STRING.equals(fieldType)) {
        return TypeInfoFactory.getPrimitiveTypeInfo("string");
    }

    throw new IllegalArgumentException("Field type " + fieldType.name() + " is not a primitive type");
}
 

static TypeInfo createLong() {
    return TypeInfoFactory.getPrimitiveTypeInfo("bigint");
}
 

static TypeInfo createString() {
    return TypeInfoFactory.getPrimitiveTypeInfo("string");
}
 

static TypeInfo createBinary() {
    return TypeInfoFactory.getPrimitiveTypeInfo("binary");
}
 

static TypeInfo createString() {
    return TypeInfoFactory.getPrimitiveTypeInfo("string");
}
 

static TypeInfo createFloat() {
    return TypeInfoFactory.getPrimitiveTypeInfo("float");
}
 

static TypeInfo createBoolean() {
    return TypeInfoFactory.getPrimitiveTypeInfo("boolean");
}
 

static TypeInfo createLong() {
    return TypeInfoFactory.getPrimitiveTypeInfo("bigint");
}
 

static TypeInfo createInt() {
    return TypeInfoFactory.getPrimitiveTypeInfo("int");
}
 

static TypeInfo createFloat() {
    return TypeInfoFactory.getPrimitiveTypeInfo("float");
}
 

static TypeInfo createBinary() {
    return TypeInfoFactory.getPrimitiveTypeInfo("binary");
}
 

static TypeInfo createBinary() {
    return TypeInfoFactory.getPrimitiveTypeInfo("binary");
}
 

static TypeInfo createDouble() {
    return TypeInfoFactory.getPrimitiveTypeInfo("double");
}
 

static TypeInfo createInt() {
    return TypeInfoFactory.getPrimitiveTypeInfo("int");
}
 

static TypeInfo createBoolean() {
    return TypeInfoFactory.getPrimitiveTypeInfo("boolean");
}
 

static TypeInfo createDouble() {
    return TypeInfoFactory.getPrimitiveTypeInfo("double");
}
 

static TypeInfo createInt() {
    return TypeInfoFactory.getPrimitiveTypeInfo("int");
}