java create array_Java Schema.createArray方法代码示例

import org.apache.avro.Schema; //导入方法依赖的package包/类

private Schema getComplexTypeDesc(DataType type) {

Schema resultTypeInfo = null;

switch (type.getCategory()) {

case Basic:

resultTypeInfo = parquetAvroSchemaMap.get(type.toString());

break;

case Struct:

// create record here

StructType structObjectType = (StructType) type;

List dataTypes = Arrays.asList(structObjectType.getColumnTypes());

Schema[] schemas = dataTypes.stream().map(dataType -> getComplexTypeDesc(dataType))

.toArray(size -> new Schema[size]);

resultTypeInfo = ParquetUtils.createAvroRecordSchema(columnName,

structObjectType.getColumnNames(), schemas);

break;

case List:

resultTypeInfo =

Schema.createArray(getComplexTypeDesc(((ListType) type).getTypeOfElement()));

break;

case Union:

final UnionType unionObjectType = (UnionType) type;

final DataType[] columnTypes1 = unionObjectType.getColumnTypes();

List colTypes = new ArrayList<>();

for (int i = 0; i < columnTypes1.length; i++) {

colTypes.add(getComplexTypeDesc(columnTypes1[i]));

}

resultTypeInfo = Schema.createUnion(colTypes);

break;

case Map:

MapType mapObjectType = (MapType) type;

resultTypeInfo = Schema.createMap(getComplexTypeDesc(mapObjectType.getTypeOfValue()));

break;

default:

break;

}

return resultTypeInfo;

}