19
19
package org .apache .flink .api .java .typeutils .runtime .kryo ;
20
20
21
21
import org .apache .flink .annotation .VisibleForTesting ;
22
- import org .apache .flink .api .common .ExecutionConfig ;
23
- import org .apache .flink .api .common .ExecutionConfig .SerializableSerializer ;
22
+ import org .apache .flink .api .common .SerializableSerializer ;
24
23
import org .apache .flink .api .common .serialization .SerializerConfig ;
24
+ import org .apache .flink .api .common .serialization .SerializerConfigImpl ;
25
25
import org .apache .flink .api .common .typeutils .TypeSerializer ;
26
26
import org .apache .flink .api .common .typeutils .TypeSerializerSnapshot ;
27
27
import org .apache .flink .api .java .typeutils .AvroUtils ;
@@ -138,8 +138,7 @@ private static ChillSerializerRegistrar loadFlinkChillPackageRegistrar() {
138
138
139
139
// ------------------------------------------------------------------------
140
140
141
- private final LinkedHashMap <Class <?>, ExecutionConfig .SerializableSerializer <?>>
142
- defaultSerializers ;
141
+ private final LinkedHashMap <Class <?>, SerializableSerializer <?>> defaultSerializers ;
143
142
private final LinkedHashMap <Class <?>, Class <? extends Serializer <?>>> defaultSerializerClasses ;
144
143
145
144
/**
@@ -167,8 +166,7 @@ private static ChillSerializerRegistrar loadFlinkChillPackageRegistrar() {
167
166
// ------------------------------------------------------------------------
168
167
// legacy fields; these fields cannot yet be removed to retain backwards compatibility
169
168
170
- private LinkedHashMap <Class <?>, ExecutionConfig .SerializableSerializer <?>>
171
- registeredTypesWithSerializers ;
169
+ private LinkedHashMap <Class <?>, SerializableSerializer <?>> registeredTypesWithSerializers ;
172
170
private LinkedHashMap <Class <?>, Class <? extends Serializer <?>>>
173
171
registeredTypesWithSerializerClasses ;
174
172
private LinkedHashSet <Class <?>> registeredTypes ;
@@ -181,15 +179,17 @@ private static ChillSerializerRegistrar loadFlinkChillPackageRegistrar() {
181
179
public KryoSerializer (Class <T > type , SerializerConfig serializerConfig ) {
182
180
this .type = checkNotNull (type );
183
181
184
- this .defaultSerializers = serializerConfig .getDefaultKryoSerializers ();
182
+ this .defaultSerializers =
183
+ ((SerializerConfigImpl ) serializerConfig ).getDefaultKryoSerializers ();
185
184
this .defaultSerializerClasses = serializerConfig .getDefaultKryoSerializerClasses ();
186
185
187
186
this .kryoRegistrations =
188
187
buildKryoRegistrations (
189
188
this .type ,
190
189
serializerConfig .getRegisteredKryoTypes (),
191
190
serializerConfig .getRegisteredTypesWithKryoSerializerClasses (),
192
- serializerConfig .getRegisteredTypesWithKryoSerializers (),
191
+ ((SerializerConfigImpl ) serializerConfig )
192
+ .getRegisteredTypesWithKryoSerializers (),
193
193
serializerConfig .isForceKryoAvroEnabled ());
194
194
}
195
195
@@ -206,7 +206,7 @@ protected KryoSerializer(KryoSerializer<T> toCopy) {
206
206
CollectionUtil .newLinkedHashMapWithExpectedSize (toCopy .kryoRegistrations .size ());
207
207
208
208
// deep copy the serializer instances in defaultSerializers
209
- for (Map .Entry <Class <?>, ExecutionConfig . SerializableSerializer <?>> entry :
209
+ for (Map .Entry <Class <?>, SerializableSerializer <?>> entry :
210
210
toCopy .defaultSerializers .entrySet ()) {
211
211
212
212
this .defaultSerializers .put (entry .getKey (), deepCopySerializer (entry .getValue ()));
@@ -220,7 +220,7 @@ protected KryoSerializer(KryoSerializer<T> toCopy) {
220
220
if (kryoRegistration .getSerializerDefinitionType ()
221
221
== KryoRegistration .SerializerDefinitionType .INSTANCE ) {
222
222
223
- ExecutionConfig . SerializableSerializer <? extends Serializer <?>> serializerInstance =
223
+ SerializableSerializer <? extends Serializer <?>> serializerInstance =
224
224
kryoRegistration .getSerializableSerializerInstance ();
225
225
226
226
if (serializerInstance != null ) {
@@ -561,7 +561,7 @@ private void checkKryoInitialized() {
561
561
562
562
// Add default serializers first, so that the type registrations without a serializer
563
563
// are registered with a default serializer
564
- for (Map .Entry <Class <?>, ExecutionConfig . SerializableSerializer <?>> entry :
564
+ for (Map .Entry <Class <?>, SerializableSerializer <?>> entry :
565
565
defaultSerializers .entrySet ()) {
566
566
kryo .addDefaultSerializer (entry .getKey (), entry .getValue ().getSerializer ());
567
567
}
@@ -606,8 +606,7 @@ private static LinkedHashMap<String, KryoRegistration> buildKryoRegistrations(
606
606
LinkedHashSet <Class <?>> registeredTypes ,
607
607
LinkedHashMap <Class <?>, Class <? extends Serializer <?>>>
608
608
registeredTypesWithSerializerClasses ,
609
- LinkedHashMap <Class <?>, ExecutionConfig .SerializableSerializer <?>>
610
- registeredTypesWithSerializers ,
609
+ LinkedHashMap <Class <?>, SerializableSerializer <?>> registeredTypesWithSerializers ,
611
610
TernaryBoolean isForceAvroKryoEnabledOpt ) {
612
611
613
612
final LinkedHashMap <String , KryoRegistration > kryoRegistrations = new LinkedHashMap <>();
@@ -629,9 +628,8 @@ private static LinkedHashMap<String, KryoRegistration> buildKryoRegistrations(
629
628
registeredTypeWithSerializerClassEntry .getValue ()));
630
629
}
631
630
632
- for (Map .Entry <Class <?>, ExecutionConfig .SerializableSerializer <?>>
633
- registeredTypeWithSerializerEntry :
634
- checkNotNull (registeredTypesWithSerializers ).entrySet ()) {
631
+ for (Map .Entry <Class <?>, SerializableSerializer <?>> registeredTypeWithSerializerEntry :
632
+ checkNotNull (registeredTypesWithSerializers ).entrySet ()) {
635
633
636
634
kryoRegistrations .put (
637
635
registeredTypeWithSerializerEntry .getKey ().getName (),
@@ -684,8 +682,8 @@ private void readObject(ObjectInputStream in) throws IOException, ClassNotFoundE
684
682
}
685
683
}
686
684
687
- private ExecutionConfig . SerializableSerializer <? extends Serializer <?>> deepCopySerializer (
688
- ExecutionConfig . SerializableSerializer <? extends Serializer <?>> original ) {
685
+ private SerializableSerializer <? extends Serializer <?>> deepCopySerializer (
686
+ SerializableSerializer <? extends Serializer <?>> original ) {
689
687
try {
690
688
return InstantiationUtil .clone (
691
689
original , Thread .currentThread ().getContextClassLoader ());
0 commit comments