pl.touk.nussknacker.engine.flink.api.serialization.SerializerWithSpecifiedClass.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nussknacker-flink-extensions-api_2.12 Show documentation

nussknacker-flink-extensions-api

There is a newer version: 1.17.0

Show newest version

package pl.touk.nussknacker.engine.flink.api.serialization

import com.esotericsoftware.kryo.Serializer
import org.apache.flink.api.common.ExecutionConfig

abstract class SerializerWithSpecifiedClass[T](acceptsNull: Boolean, immutable: Boolean)
    extends Serializer[T](acceptsNull, immutable)
    with Serializable {

  def clazz: Class[_]

  def registerIn(config: ExecutionConfig) = {
    config.getRegisteredTypesWithKryoSerializers.put(clazz, new ExecutionConfig.SerializableSerializer(this))
    config.getDefaultKryoSerializers.put(clazz, new ExecutionConfig.SerializableSerializer(this))
  }

}