由于Flink支持的数据类型有限,特别是LocalDate之类的还不支持,所以对于POJO中使用了这些类型,就需要自行指定.
方法是在POJO class上添加org.apache.flink.api.common.typeinfo.TypeInfo里面指定一个类来处理这个POJO的各个字段的类型.下面是一个通用的类,代码如下
@Data
@NoArgsConstructor
@AllArgsConstructor
@TypeInfo(FlinkTypeInfoFactory.class)
public class SomePOJO{
private LocalDate localDate;
private LocalTime localTime;
private LocalDateTime localDateTime;
}
public final class FlinkTypeInfoFactory<T> extends TypeInfoFactory<T> {
@Override
public TypeInformation<T> createTypeInfo(Type t,
Map<String, TypeInformation<?>> genericParameters) {
Class<T> clazz= (Class<T>) t;
Map<String,TypeInformation<?>> typeInfo= Maps.newHashMap();
Utils.fields(clazz).forEach(field -> {
String key=field.getName();
Class<?> type=field.getType();
if(type==String.class){
typeInfo.put(key,Types.STRING);
}else if(type == LocalDate.class){
typeInfo.put(key,Types.LOCAL_DATE);
}else if(type== LocalTime.class){
typeInfo.put(key,Types.LOCAL_TIME);
}else if(type== LocalDateTime.class){
typeInfo.put(key,Types.LOCAL_DATE_TIME);
}else if(type==Integer.class){
typeInfo.put(key,Types.INT);
}else if(type==Long.class){
typeInfo.put(key,Types.LONG);
}else if(type==Boolean.class){
typeInfo.put(key,Types.BOOLEAN);
}
});
return Types.POJO(clazz,typeInfo);
}
}