Flink 数据类型 & TypeInformation信息

原生数据类型

Java Tuples类型

Scala Case Class类型

POJOs 类型

Flink Value类型

特殊数据类型

Scala API类型信息

Java API类型信息

自定义TypeInformation


Flink 流应用程序处理的是以数据对象表示的事件流。所以在Flink内部,我么需要能够处理这些对象。它们需要被序列化和反序列化,以便通过网络传送它们;或者从状态后端、检查点保存点读取它们。为了有效地做到这一点,Flink需要明确知道应用程序所处理的数据类型。并为每个数据类型生成特定的序列化器、反序列化器和比较器。Flink支持非常完善的数据类型,数据类型描述信息都是由 TypeInformation定义,比较常用的 TypeInformation有BasicTypeInfoTupleTypeInfoCaseClassTypeInfoPojoTypeInfo类等。TypeInformation主要作用是为了在 Flink系统内有效地对数据结构类型进行管理能够在分布式计算过程中对数据的类型进行管理和推断。同时基于对数据的类型信息管理,Flink内部对数据存储也进行了相应的性能优化。Flink能够支持任意的Java或Scala的数据类型,不用像 Hadoop中的org.apache.hadoop.io.Writable 而实现特定的序列化和反序列化接口,从而让用户能够更加容易使用已有的数据结构类型。另外使用 TypeInformation管理数据类型信息,能够在数据处理之前将数据类型推断出来,而不是真正在触发计算后才识别出,这样能够及时有效地避免用户在使用 Flink编写应用的过程中的数据类型问题。

原生数据类型

Flink通过实现 BasicTypeInfo数据类型,能够支持任意Java 原生基本类型(装箱)或 String 类型,例如Integer、String、Double等,如以下代码所示,通过从给定的元素集中创建 DataStream数据集。

//创建 Int 类型的数据集
DataStreamSource<Integer> integerDataStreamSource = env.fromElements(1, 2, 3, 4, 5);
//创建 String 的类型的数据集
DataStreamSource<String> stringDataStreamSource = env.fromElements("Java", "Scala");

Flink实现另外一种 TypeInfomation是 BasicArrayTypeInfo,对应的是 Java基本类型数组(装箱)或 String对象的数组,如下代码通过使用 Array数组和 List集合创建 DataStream数据集。

List<Integer> integers = Arrays.asList(1, 2, 3, 4, 5);
//通过 List 集合创建数据集
DataStreamSource<Integer> integerDataStreamSource1 = env.fromCollection(integers);

Java Tuples类型

通过定义 TupleTypeInfo来描述 Tuple类型数据,Flink在Java接口中定义了元祖类(Tuple)供用户使用。Flink Tuples是固定长度固定类型的Java Tuple实现,不支持空值存储。目前支持任意的Flink Java Tuple类型字段数量上限为25,如果字段数量超过上限,可以通过继承Tuple类的方式进行拓展。如下代码所示,创建Tuple数据类型数据集。

//通过实例化 Tuple2 创建具有两个元素的数据集
DataStreamSource<Tuple2<String, Integer>> tuple2DataStreamSource = env.fromElements(new Tuple2<>("a", 1), new Tuple2<>("b", 2));
//通过实例化 Tuple3 创建具有三个元素的数据集
DataStreamSource<Tuple3<String, Integer, Long>> tuple3DataStreamSource = env.fromElements(new Tuple3<>("a", 1, 3L), new Tuple3<>("b", 2, 3L));

Scala Case Class类型

Flink通过实现 CaseClassTypeInfo支持任意的 Scala Case Class,包括Scala tuples类型,支持的字段数量上限为22,支持通过字段名称和位置索引获取指标,不支持存储空值。如下代码实例所示,定义WordCount Case Class数据类型,然后通过fromElements方法创建input数据集,调用 keyBy()方法对数据集根据 word字段重新分区。

//定义 WordCount Case Class 数据结构
case class WordCount(word: Sring, count: Int)
//通过 fromElements 方法创建数据集
val input = env.fromElements(WordCount("hello", 1),WordCount("word",2))
val keyStream1 = input.keyBy("word")//根据word字段为分区字段,
val keyStream2 = input.keyBy(0)//也可以通过制定position分区

通过使用Scala Tuple创建 DataStream数据集,其他的使用方式和 Case Class相似。需要注意的是,如果根据名称获取字段,可以使用 Tuple中的默认字段名称。

//通过实例化Scala Tuple2 创建具有两个元素的数据集
val tupleStream: DataStream[Tuple2[String,Int]] = env.fromElements(("a",1),("b",2));
//使用默认名字段获取字段,表示第一个 tuple字段,相当于下标0
tuple2DataStreamSource.keyBy("_1");

POJOs 类型

POJOs类可以完成复杂数据结构的定义,Flink通过实现 PojoTypeInfo来描述任意的 POJOs,包括 Java 和 Scala类。在Flink中使用POJOs类可以通过字段名称获取字段,例如 dataStream.join(otherStream).where("name").equalTo("personName"),对于用户做数据处理则非常透明和简单,如代码所示。如果在 Flink中使用POJOs数据类型,需要遵循以下要求:
【1】POJOs 类必须是 Public修饰且必须独立定义,不能是内部类;
【2】POJOs类中必须含有默认空构造器
【3】POJOs类中所有的 Fields必须是 Public或者具有 Public修饰的 getter和 setter方法;
【4】POJOs类中的字段类型必须是 Flink支持的

//类和属性具有 public 修饰
public class Persion{
    public String name;
    public Integer age;
    //具有默认的空构造器
    public Persion(){}
    public Persion(String name,Integer age){
        this.name = name;
        this.age = age;
    };
}

定义好 POJOs Class后,就可以在 Flink环境中使用了,如下代码所示,使用 fromElements接口构建 Person类的数据集。POJOs类仅支持字段名称指定字段,如代码中通过 Person name来指定 Keyby字段。

DataStreamSource<Persion> persionDataStreamSource = env.fromElements(new Persion("zzx", 18), new Persion("fj", 16));
persionData.keyBy("name").sum("age");

Flink Value类型

Value数据类型实现了org.apache.flink.types.Value,其中包括 read()和 write()两个方法完成序列化和反序列化操作,相对于通用的序列化工具会有着比较高效的性能。目前 Flink提供了內建的 Value类型有 IntValue、DoubleValue以及 StringValue等,用户可以结合原生数据类型和 Value类型使用。

特殊数据类型

在Flink中也支持一些比较特殊的数据数据类型,例如 Scala中的List、Map、Either、Option、Try数据类型,以及 Java中Either数据类型,还有 Hadoop的 Writable数据类型。如下代码所示,创建 Map和 List类型数据集。这种数据类型使用场景不是特别广泛,主要原因是数据中的操作相对不像 POJOs类那样方便和透明,用户无法根据字段位置或者名称获取字段信息,同时要借助Types Hint帮助Flink推断数据类型信息,关于Tyeps Hmt介绍可以参考下一小节。

//创建 map 类型数据集
Map map = new HashMap<>();
map.put("name","zzx");
map.put("age",12);
env.fromElements(map);
//创建 List 类型数据集
env.fromElements(Arrays.asList(1,2,3,4,5),Arrays.asList(3,4,5));

TypeInformation信息获取:通常情况下 Flink都能正常进行数据类型推断,并选择合适的 serializers以及 comparators。但在某些情况下却无法直接做到,例如定义函数时如果使用到了泛型,JVM就会出现类型擦除的问题,使得 Flink并不能很容易地获取到数据集中的数据类型信息。同时在 Scala API和 Java API中,Flink分别使用了不同的方式重构了数据类型信息。

Scala API类型信息

Scala API通过使用 Manifest和类标签,在编译器运行时获取类型信息,即使是在函数定义中使用了泛型,也不会像Java API出现类型擦除的问题,这使得Scala API具有非常精密的类型管理机制。同时在Flink中使用到Scala Macros框架,在编译代码的过程中推断函数输入参数和返回值的类型信息,同时在Flink中注册成TypeInformation以支持上层计算算子使用。
当使用Scala API开发 Flink应用,如果使用到 Flink已经通过 TypeInformation定义的数据类型,TypeInformation类不会自动创建,而是使用隐式参数的方式引入,代码不会直接抛出编码异常,但是当启动 Flink应用程序时就会报”could not find implicit value for evidence parameter of type TypeInformation”的错误。这时需要将 TypeInformation类隐式参数引入到当前程序环境中,代码实例如下:

import org.apache.flink.api.scala._

Java API类型信息

由于 Java的泛型会出现类型擦除问题,Flink通过 Java反射机制尽可能重构类型信息,例如使用函数签名以及子类的信息等。同时类型推断在当输出类型依赖于输入参数类型时相对比较容易做到,但是如果函数的输出类型不依赖于输入参数的类型信息,这个时候就需要借助于类型提示(Ctype Himts)来告诉系统函数中传入的参数类型信息和输出参数信息。如代码清单通过在returns方法中传入TypeHint实例指定输出参数类型,帮助 Flink系统对输出类型进行数据类型参数的推断和收集。

//定义泛型函数,输入参数 T,O 输出参数为 O
class MyMapFucntion<T,O> implements MapFunction<T,O>{
    @Override
    public O map(T t) throws Exception {
        //定义计算逻辑
        return null;
    }
}

//通过 List 集合创建数据集
DataStreamSource<Integer> input = env.fromCollection(integers);
input.flatMap(new MyMapFucntion<String,Integer>()).returns(new TypeHint<Integer>() {//通过returns方法指定返回参数类型
})

在使用Java API定义 POJOs类型数据时,PojoTypeInformation为 POJOs类中的所有字段创建序列化器,对于标准的类型,例如IntegerStringLong等类型是通过 Flink自带的序列化器进行数据序列化,对于其他类型数据都是直接调用 Kryo序列化工具来进行序列化。通常情况下,如果 Kryo序列化工具无法对 POJOs类序列化时,可以使用 Avro对 POJOs类进行序列化,如下代码通过在 ExecutionConfig中调用 enableForceAvro()来开启Avro序列化。

//获取运行环境
final StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
//开启 avro 序列化
env.getConfig().enableForceAvro();

如果用户想使用 Kryo序列化工具来序列化 POJOs所有字段,则在 ExecutionConfig中调用 enableForceKryo()来开启 Kryo序列化。

//获取运行环境
final StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
//开启 Kryo 序列化
env.getConfig().enableForceKryo();

如果默认的 Kryo序列化类不能序列化 POJOs对象,通过调用 ExecutionConfig的 addDefaultKryoSerializer()方法向 Kryo中添加自定义的序列化器。

 public void addDefaultKryoSerializer(Class<?> type, Class<? extends Serializer<?>> serializerClass)

自定义TypeInformation

除了使用已有的 TypeInformation所定义的数据格式类型之外,用户也可以自定义实现 TypeInformation,来满足的不同的数据类型定义需求。Flink提供了可插拔的 Type Information Factory让用户将自定义的 TypeInformation注册到 Flink类型系统中。如下代码所示只需要通过实现 org.apache.flink.api.common.typeinfo.TypeInfoFactory接口,返回相应的类型信息。通过@TypeInfo注解创建数据类型,定义 CustomTuple数据类型。

@TypeInfo(CustomTypeInfoFactory.class)
public class CustomTuple<T0,T1>{
	public T0 field0;
	public T1 field1;
}

然后定义CustomTypeInfoFactory 类继承于TypeInfoFactory,参数类型指定CustomTuple。最后重写createTypeInfo方法,创建的CustomTupleTypeInfo就是CustomTuple数据类型TypeInformation。

public class CustomTypeInfoFactory extends TypeInfoFactory<CustomTuple>{
	@Override
	public TypeInfomation<CustomTuple> createTypeInfo(Type t, Map<String,TypeInfoFactory<?>> genericParameters){
		return new CustomTupleTypeInfo(genericParameters.get("T0"),genericParameters.get("T1");
	}
}


----关注公众号,获取更多内容----

猜你喜欢

转载自blog.csdn.net/zhengzhaoyang122/article/details/107164147