Improve Kryo Codec for registrations

Author: artembilan

spring-integration-core/src/main/java/org/springframework/integration/codec/kryo/AbstractKryoCodec.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2015-2019 the original author or authors.
+ * Copyright 2015-2020 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -44,6 +44,7 @@ public abstract class AbstractKryoCodec implements Codec {
 	protected AbstractKryoCodec() {
 		KryoFactory factory = () -> {
 			Kryo kryo = new Kryo();
+			kryo.setRegistrationRequired(true);
 			// configure Kryo instance, customize settings
 			configureKryoInstance(kryo);
 			return kryo;

spring-integration-core/src/main/java/org/springframework/integration/codec/kryo/KryoClassListRegistrar.java

@@ -17,6 +17,7 @@
 package org.springframework.integration.codec.kryo;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 
 import org.springframework.util.Assert;
@@ -43,6 +44,13 @@ public class KryoClassListRegistrar extends AbstractKryoRegistrar {
 
 	private int initialValue = DEFAULT_INITIAL_ID;
 
+	/**
+	 * @param classes the vararg of classes to validateRegistration
+	 */
+	public KryoClassListRegistrar(Class<?>... classes) {
+		this(Arrays.asList(classes));
+	}
+
 	/**
 	 * @param classes the list of classes to validateRegistration
 	 */
@@ -56,8 +64,7 @@ public KryoClassListRegistrar(List<Class<?>> classes) {
 	 * @param initialValue the initial value
 	 */
 	public void setInitialValue(int initialValue) {
-		Assert.isTrue(initialValue >= MIN_REGISTRATION_VALUE,
-				"'initialValue' must be >= " + MIN_REGISTRATION_VALUE);
+		Assert.isTrue(initialValue >= MIN_REGISTRATION_VALUE, "'initialValue' must be >= " + MIN_REGISTRATION_VALUE);
 		this.initialValue = initialValue;
 	}
 

spring-integration-core/src/test/java/org/springframework/integration/codec/kryo/CompositeCodecTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2015-2019 the original author or authors.
+ * Copyright 2015-2020 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -39,8 +39,9 @@ public class CompositeCodecTests {
 
 	@Before
 	public void setup() {
-		Map<Class<?>, Codec> codecs = new HashMap<Class<?>, Codec>();
-		this.codec = new CompositeCodec(codecs, new PojoCodec());
+		Map<Class<?>, Codec> codecs = new HashMap<>();
+		this.codec = new CompositeCodec(codecs, new PojoCodec(
+				new KryoClassListRegistrar(SomeClassWithNoDefaultConstructors.class)));
 	}
 
 	@Test

spring-integration-core/src/test/java/org/springframework/integration/codec/kryo/KryoCodecTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2015-2019 the original author or authors.
+ * Copyright 2015-2020 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -66,7 +66,7 @@ public void testSerializationWithStreams() throws IOException {
 
 	@Test
 	public void testPojoSerialization() throws IOException {
-		PojoCodec codec = new PojoCodec();
+		PojoCodec codec = new PojoCodec(new KryoClassListRegistrar(SomeClassWithNoDefaultConstructors.class));
 		SomeClassWithNoDefaultConstructors foo = new SomeClassWithNoDefaultConstructors("foo", 123);
 		ByteArrayOutputStream bos = new ByteArrayOutputStream();
 		codec.encode(foo, bos);
@@ -101,12 +101,12 @@ public void testPrimitiveSerialization() throws IOException {
 
 	@Test
 	public void testMapSerialization() throws IOException {
-		PojoCodec codec = new PojoCodec();
+		PojoCodec codec = new PojoCodec(new KryoClassListRegistrar(HashMap.class));
 		Map<String, Integer> map = new HashMap<>();
 		map.put("one", 1);
 		map.put("two", 2);
 		ByteArrayOutputStream bos = new ByteArrayOutputStream();
-		codec.encode(map, bos);
+		codec.encode(map, bos);4
 		Map<?, ?> m2 = (Map<?, ?>) codec.decode(bos.toByteArray(), HashMap.class);
 		assertThat(m2.size()).isEqualTo(2);
 		assertThat(m2.get("one")).isEqualTo(1);
@@ -115,7 +115,7 @@ public void testMapSerialization() throws IOException {
 
 	@Test
 	public void testComplexObjectSerialization() throws IOException {
-		PojoCodec codec = new PojoCodec();
+		PojoCodec codec = new PojoCodec(new KryoClassListRegistrar(Foo.class));
 		Foo foo = new Foo();
 		foo.put("one", 1);
 		foo.put("two", 2);