Merge pull request #102 from Pr0methean/patch-1

Fix #95
magro · Dec 9, 2018 · b11df0d · b11df0d
2 parents d5f4901 + ca06b9c
commit b11df0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/de/javakaffee/kryoserializers/EnumMapSerializer.java b/src/main/java/de/javakaffee/kryoserializers/EnumMapSerializer.java
@@ -80,7 +80,7 @@ public class EnumMapSerializer extends Serializer<EnumMap<? extends Enum<?>, ?>>
         final EnumMap rawResult = result;
         final int size = input.readInt(true);
         for ( int i = 0; i < size; i++ ) {
-            final int ordinal = input.readInt(true);
+            final int ordinal = input.readVarInt(true);
             final Enum<?> key = enumConstants[ordinal];
             final Object value = kryo.readClassAndObject( input );
             rawResult.put( key, value );
@@ -93,7 +93,7 @@ public void write(final Kryo kryo, final Output output, final EnumMap<? extends
         kryo.writeClass( output, getKeyType( map ) );
         output.writeInt(map.size(), true);
         for ( final Map.Entry<? extends Enum<?>,?> entry :  map.entrySet() ) {
-            output.writeInt(entry.getKey().ordinal(), true);
+            output.writeVarInt(entry.getKey().ordinal(), true);
             kryo.writeClassAndObject(output, entry.getValue());
         }
         if ( TRACE ) trace( "kryo", "Wrote EnumMap: " + map );