Spring/spring-integration
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve Kryo Codec for registrations

Browse Source
This commit is contained in:
Artem Bilan
2020-07-22 12:07:05 -04:00
parent 064cb57ac0
commit 6a02b5abe9
4 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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

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

@@ -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 class KryoCodecTests {
@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 class KryoCodecTests {
@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 class KryoCodecTests {
@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);